From 74a7670703668fe1d81b3113f8daf36765df5eed Mon Sep 17 00:00:00 2001
From: TianyuZhang1214 <neo@apecloud.com>
Date: Mon, 25 Nov 2024 18:38:00 +0800
Subject: [PATCH] to #119 feat: replica_setup.sh under postgres is available.

---
 .../replica-setup-postgres/replica_setup.sh   | 43 +++++++++----------
 docker/Dockerfile                             |  1 +
 docker/entrypoint.sh                          |  4 +-
 3 files changed, 23 insertions(+), 25 deletions(-)

diff --git a/devtools/replica-setup-postgres/replica_setup.sh b/devtools/replica-setup-postgres/replica_setup.sh
index d8c64029..5d48c495 100644
--- a/devtools/replica-setup-postgres/replica_setup.sh
+++ b/devtools/replica-setup-postgres/replica_setup.sh
@@ -6,12 +6,11 @@ usage() {
 }
 
 MYDUCK_HOST=${MYDUCK_HOST:-127.0.0.1}
-MYDUCK_PORT=${MYDUCK_PORT:-3306}
-MYDUCK_USER=${MYDUCK_USER:-root}
+MYDUCK_PORT=${MYDUCK_PORT:-5432}
+MYDUCK_USER=${MYDUCK_USER:-mysql}
 MYDUCK_PASSWORD=${MYDUCK_PASSWORD:-}
 MYDUCK_SERVER_ID=${MYDUCK_SERVER_ID:-2}
 MYDUCK_IN_DOCKER=${MYDUCK_IN_DOCKER:-false}
-GTID_MODE="ON"
 
 while [[ $# -gt 0 ]]; do
     case $1 in
@@ -62,34 +61,32 @@ while [[ $# -gt 0 ]]; do
     esac
 done
 
-source checker.sh
-
 # Check if all parameters are set
 if [[ -z "$SOURCE_HOST" || -z "$SOURCE_PORT" || -z "$SOURCE_USER" ]]; then
     echo "Error: Missing required Postgres connection variables: SOURCE_HOST, SOURCE_PORT, SOURCE_USER."
     usage
 fi
 
-# Step 3: Check Postgres configuration
+# Step 1: Check Postgres configuration
 echo "Checking Postgres configuration..."
 # TODO(neo.zty): add check for Postgres configuration
-check_command "Postgres configuration check"
 
-# Step 3: Prepare MyDuck Server for replication
-echo "Preparing MyDuck Server for replication..."
-# TODO(neo.zty): add prepare for MyDuck Server for replication
-check_command "preparing MyDuck Server for replication"
+# Step 2: Establish replication
+echo "Starting replication..."
+export PUBLICATION_NAME="myduck_publication"
+export SUBSCRIPTION_NAME="myduck_subscription"
 
-# Step 5: Copy the existing data if the Postgres instance is not empty
-if [[ $SOURCE_IS_EMPTY -ne 0 ]]; then
-    echo "Copying a snapshot of the Postgres instance to MyDuck Server..."
-    source snapshot.sh
-    check_command "copying a snapshot of the Postgres instance"
-else
-    echo "This Postgres instance is empty. Skipping snapshot."
-fi
+CREATE_SUBSCRIPTION_SQL="CREATE SUBSCRIPTION ${SUBSCRIPTION_NAME} \
+    CONNECTION 'dbname=${SOURCE_DATABASE} host=${SOURCE_HOST} port=${SOURCE_PORT} user=${SOURCE_USER} password=${SOURCE_PASSWORD}' \
+    PUBLICATION ${PUBLICATION_NAME};"
 
-# Step 6: Establish replication
-echo "Starting replication..."
-source start_replication.sh
-check_command "starting replication"
\ No newline at end of file
+psql -h $MYDUCK_HOST -p $MYDUCK_PORT -U $MYDUCK_USER <<EOF
+${CREATE_SUBSCRIPTION_SQL}
+EOF
+
+if [[ -n "$?" ]]; then
+    echo "SQL executed successfully."
+else
+    echo "SQL execution failed. Check the error message above."
+    exit 1
+fi
\ No newline at end of file
diff --git a/docker/Dockerfile b/docker/Dockerfile
index 974dea91..5fc08647 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -106,6 +106,7 @@ WORKDIR /home/admin
 COPY --from=builder /myduckserver /usr/local/bin/myduckserver
 COPY --chown=admin:admin --chmod=755 docker/*.sh .
 COPY --chown=admin:admin --chmod=755 devtools/replica-setup-mysql ./replica-setup-mysql
+COPY --chown=admin:admin --chmod=755 devtools/replica-setup-postgres ./replica-setup-postgres
 
 # ENV LC_CTYPE="en_US.UTF-8"
 # ENV LANG="en_US.UTF-8"
diff --git a/docker/entrypoint.sh b/docker/entrypoint.sh
index 417d206e..3a236aa5 100644
--- a/docker/entrypoint.sh
+++ b/docker/entrypoint.sh
@@ -18,10 +18,10 @@ parse_dsn() {
     SOURCE_DATABASE=""
 
     # Detect type
-    if [[ "$dsn" =~ ^postgresql:// ]]; then
+    if [[ "$dsn" =~ ^postgres:// ]]; then
         SOURCE_TYPE="POSTGRES"
         # Strip the prefix
-        dsn="${dsn#postgresql://}"
+        dsn="${dsn#postgres://}"
     elif [[ "$dsn" =~ ^mysql:// ]]; then
         SOURCE_TYPE="MYSQL"
         # Strip the prefix