Chore: Update walg and adminapi pitr scripts

Signed-off-by: Lakshmipathi <[email protected]>

Author: Lakshmipathi

ansible/files/admin_api_scripts/commence_walg_backup.sh

@@ -2,6 +2,10 @@
 
 set -euo pipefail
 
-WALG_SENTINEL_USER_DATA="{ \"project_id\": $1, \"backup_id\": $2 }" nohup wal-g backup-push /var/lib/postgresql/data --config /etc/wal-g/config.json --verify >> /var/log/wal-g/backup-push.log 2>&1 &
-
-echo "WAL-G backup job commenced"
+if [ ! -f /usr/bin/admin-mgr ]; then
+  WALG_SENTINEL_USER_DATA="{ \"project_id\": $1, \"backup_id\": $2 }" nohup wal-g backup-push /var/lib/postgresql/data --config /etc/wal-g/config.json --verify >> /var/log/wal-g/backup-push.log 2>&1 &
+  echo "WAL-G backup job commenced"
+else
+  /usr/bin/admin-mgr commence-backup "$1" "$2"
+  echo "WAL-G backup job commenced" | systemd-cat -t commence_walg_backup.sh -p emerg
+fi

ansible/files/admin_api_scripts/commence_walg_restore.sh

@@ -45,5 +45,10 @@ function commence_walg_restore {
 
 set -euo pipefail
 
+if [ ! -f /usr/bin/admin-mgr ]; then
 commence_walg_restore "$1" "$2" >> /var/log/wal-g/backup-fetch.log 2>&1 &
 echo "WAL-G restore job commenced"
+else
+/usr/bin/admin-mgr commence-restore "$1" "$2"
+echo "WAL-G restore job commenced" | systemd-cat -t commence_walg_restore.sh -p emerg
+fi

ansible/files/admin_api_scripts/complete_walg_restore.sh

@@ -2,6 +2,7 @@
 
 set -euo pipefail
 
+if [ ! -f /usr/bin/admin-mgr ]; then
 # Disable recovery commands in the event of a restart
 sed -i "s/.*restore_command/#restore_command/" /etc/postgresql-custom/wal-g.conf
 sed -i "s/.*recovery_target_time/#recovery_target_time/" /etc/postgresql-custom/wal-g.conf
@@ -13,3 +14,7 @@ sed -i "s/.*recovery_target_inclusive/#recovery_target_inclusive/" /etc/postgres
 rm -rf /tmp/walg_data/ /tmp/.wal-g/ /tmp/wal_fetch_dir/
 
 echo "Cleanup post WAL-G restoration complete"
+else
+/usr/bin/admin-mgr complete-restore
+echo "Cleanup post WAL-G restoration complete" | systemd-cat -t complete_walg_restore.sh -p emerg
+fi

ansible/files/admin_api_scripts/disable_walg.sh

@@ -2,10 +2,16 @@
 
 set -euo pipefail
 
+if [ ! -f /usr/bin/admin-mgr ]; then
 sed -i "s/.*archive_mode/#archive_mode/" /etc/postgresql-custom/wal-g.conf
 sed -i "s/.*archive_command/#archive_command/" /etc/postgresql-custom/wal-g.conf
 sed -i "s/.*archive_timeout/#archive_timeout/" /etc/postgresql-custom/wal-g.conf
 
 systemctl restart postgresql
 
 echo "WAL-G successfully disabled"
+else
+/usr/bin/admin-mgr disable-walg
+chmod -x /usr/bin/admin-mgr 
+echo "WAL-G successfully disabled" | systemd-cat -t disable_walg.sh -p emerg
+fi

ansible/files/admin_api_scripts/enable_walg.sh

@@ -2,10 +2,19 @@
 
 set -euo pipefail
 
+if [ ! -f /usr/bin/admin-mgr ]; then
 sed -i "s/.*archive_mode/archive_mode/" /etc/postgresql-custom/wal-g.conf
 sed -i "s/.*archive_command/archive_command/" /etc/postgresql-custom/wal-g.conf
 sed -i "s/.*archive_timeout/archive_timeout/" /etc/postgresql-custom/wal-g.conf
 
 systemctl restart postgresql
 
 echo "WAL-G successfully enabled"
+else
+/usr/bin/admin-mgr enable-walg
+#replace restore_command
+sed -i '/restore_command/d' /etc/postgresql-custom/wal-g.conf
+echo "#restore_command = '/usr/bin/admin-mgr wal-fetch %f %p >> /var/log/wal-g/wal-fetch.log 2>&1'" >> /etc/postgersql-custom/wal-g.conf
+chmod +x /usr/bin/admin-mgr && chmod ug+s /usr/bin/admin-mgr
+echo "WAL-G successfully enabled" | systemd-cat -t enable_walg.sh -p emerg
+fi

ansible/files/walg_helper_scripts/wal_change_ownership.sh

@@ -2,6 +2,7 @@
 
 set -euo pipefail
 
+if [ ! -f /usr/bin/admin-mgr ]; then
 filename=$1
 
 if [[ -z "$filename" ]]; then
@@ -40,3 +41,6 @@ fi
 
 # once valid, proceed to change ownership
 chown postgres:postgres "$full_path"
+else
+echo "Do nothing, handled via admin-mgr"
+fi

ansible/files/walg_helper_scripts/wal_fetch.sh

@@ -2,6 +2,7 @@
 
 set -euo pipefail
 
+if [ ! -f /usr/bin/admin-mgr ]; then
 # Fetch the WAL file and temporarily store them in /tmp
 sudo -u wal-g wal-g wal-fetch "$1" /tmp/wal_fetch_dir/"$1" --config /etc/wal-g/config.json 
 
@@ -10,3 +11,6 @@ sudo -u root /root/wal_change_ownership.sh "$1"
 
 # Move file to its final destination
 mv /tmp/wal_fetch_dir/"$1" /var/lib/postgresql/data/"$2"
+else
+echo "Do nothing, handled via wal-g.conf file"
+fi