From acfb04b9f3ec7d384a6ed28e85bde141f214062d Mon Sep 17 00:00:00 2001
From: SK Ali Arman <arman.cse4.bu@gmail.com>
Date: Thu, 5 Dec 2024 16:24:37 +0600
Subject: [PATCH] Add PITR Replication Strategy Support & Fix Same Memeber_ID
 (#40)

Signed-off-by: SK Ali Arman <arman@appscode.com>
---
 init-script/run.sh        |  1 +
 scripts/copy-data.sh      | 32 ++++++++++++++++++++++++++++++++
 scripts/run.sh            | 15 +++++++++++++++
 scripts/standalone-run.sh | 19 +++++++++++++++++++
 4 files changed, 67 insertions(+)
 create mode 100755 scripts/copy-data.sh
 create mode 100755 scripts/standalone-run.sh

diff --git a/init-script/run.sh b/init-script/run.sh
index 43b1dbf..f2a8174 100755
--- a/init-script/run.sh
+++ b/init-script/run.sh
@@ -2,4 +2,5 @@
 
 rm -rf /var/lib/mysql/lost+found
 rm -rf /run-scripts/*
+rm /var/lib/mysql/auto.cnf
 cp /tmp/scripts/* /scripts
diff --git a/scripts/copy-data.sh b/scripts/copy-data.sh
new file mode 100755
index 0000000..c121f98
--- /dev/null
+++ b/scripts/copy-data.sh
@@ -0,0 +1,32 @@
+#!/usr/bin/env bash
+
+ROOT_DIR="$ROOT_DIR"
+TOTAL_DIR_TO_COPY="$TOTAL_DIR_TO_COPY"
+DATA_DIR="$DATA_DIR"
+
+if [ -z "$ROOT_DIR" ] || [ -z "$TOTAL_DIR_TO_COPY" ]; then
+  echo "ROOT_DIR and TOTAL_DIR_TO_COPY must be set."
+  exit 1
+fi
+Size1=$(du -s "$ROOT_DIR" | cut -f1)
+echo "DATA DIRECTORY SIZE: ", $Size1
+for (( i = 1; i <= $TOTAL_DIR_TO_COPY; i++ ));do
+  if [[ -d "$ROOT_DIR$i" ]];then
+    Size2=$(du -s "$ROOT_DIR$i" | cut -f1)
+    echo $Size1, " ", $Size2
+    if [[ "$Size1" == "$Size2" ]];then
+      continue
+    fi
+  fi
+  # not deleting any data
+  # because the sole purpose of this script is to copy the data
+  # rm -rf "$ROOT_DIR$i"/*
+  rm "$ROOT_DIR/mysql.sock"
+  cp -rvL "$ROOT_DIR/"* "$ROOT_DIR$i"/
+  if [[ $? -ne 0 ]]; then
+    echo "Error occurred while copying to $ROOT_DIR$i"
+    exit 1
+  fi
+done
+
+exit 0
\ No newline at end of file
diff --git a/scripts/run.sh b/scripts/run.sh
index b2e4ad4..7fdb918 100755
--- a/scripts/run.sh
+++ b/scripts/run.sh
@@ -15,6 +15,21 @@
 #   POD_IP_TYPE         = Address type of POD_IP (one of IPV4, IPv6)
 
 env | sort | grep "POD\|HOST\|NAME"
+RECOVERY_DONE_FILE="/tmp/recovery.done"
+if [[ "$PITR_RESTORE" == "true" ]]; then
+    while true; do
+      sleep 2
+      echo "Point In Time Recovery In Progress. Waiting for $RECOVERY_DONE_FILE file"
+      if [[ -e "$RECOVERY_DONE_FILE" ]]; then
+        echo "$RECOVERY_DONE_FILE found."
+        break
+      fi
+    done
+fi
+
+if [[ -e "$RECOVERY_DONE_FILE" ]]; then
+  rm $RECOVERY_DONE_FILE
+fi
 
 args=$@
 script_name=${0##*/}
diff --git a/scripts/standalone-run.sh b/scripts/standalone-run.sh
new file mode 100755
index 0000000..3dfbe1b
--- /dev/null
+++ b/scripts/standalone-run.sh
@@ -0,0 +1,19 @@
+#!/bin/sh
+
+if [[ -z "$TOPOLOGY" ]]; then
+  RECOVERY_DONE_FILE="/tmp/recovery.done"
+  if [[ "$PITR_RESTORE" == "true" ]]; then
+    while true; do
+      sleep 2
+      echo "Point In Time Recovery In Progress. Waiting for $RECOVERY_DONE_FILE file"
+      if [[ -e "$RECOVERY_DONE_FILE" ]]; then
+        echo "$RECOVERY_DONE_FILE found."
+        break
+      fi
+    done
+  fi
+
+  if [[ -e "$RECOVERY_DONE_FILE" ]]; then
+    rm $RECOVERY_DONE_FILE
+  fi
+fi