added compression, added local storage dump path

Author: Fraccaman

Dockerfile

@@ -21,6 +21,8 @@ ENV S3_S3V4 'no'
 ENV SCHEDULE ''
 ENV PASSPHRASE ''
 ENV BACKUP_KEEP_DAYS ''
+ENV LOCAL_STORE_PREFIX ''
+ENV COMPRESS ''
 
 ADD src/run.sh run.sh
 ADD src/env.sh env.sh

docker-compose.yaml

@@ -26,3 +26,5 @@ services:
       POSTGRES_DATABASE: postgres
       POSTGRES_USER: user
       POSTGRES_PASSWORD: password
+      COMPRESS: 'true'  # optional, set to 'true' to enable compression
+      LOCAL_STORE_PREFIX: /data/backups  # optional, local storage prefix for backups

src/backup.sh

@@ -5,27 +5,36 @@ set -o pipefail
 
 source ./env.sh
 
+file_path="${LOCAL_STORE_PREFIX}/db.dump"
+
 echo "Creating backup of $POSTGRES_DATABASE database..."
 pg_dump --format=custom \
         -h $POSTGRES_HOST \
         -p $POSTGRES_PORT \
         -U $POSTGRES_USER \
         -d $POSTGRES_DATABASE \
         $PGDUMP_EXTRA_OPTS \
-        > db.dump
+        > "$file_path"
 
 timestamp=$(date +"%Y-%m-%dT%H:%M:%S")
 s3_uri_base="s3://${S3_BUCKET}/${S3_PREFIX}/${POSTGRES_DATABASE}_${timestamp}.dump"
 
+if [ -n "$COMPRESS" ]; then
+  echo "Compressing backup..."
+  zstd -2 "$file_path"
+  file_path="${file_path}.zst"
+  s3_uri_base="${s3_uri_base}.zst"
+fi
+
 if [ -n "$PASSPHRASE" ]; then
   echo "Encrypting backup..."
-  rm -f db.dump.gpg
-  gpg --symmetric --batch --passphrase "$PASSPHRASE" db.dump
-  rm db.dump
-  local_file="db.dump.gpg"
+  rm -f "$file_path.gpg"
+  gpg --symmetric --batch --passphrase "$PASSPHRASE" "$file_path"
+  rm "$file_path"
+  local_file="$file_path.gpg"
   s3_uri="${s3_uri_base}.gpg"
 else
-  local_file="db.dump"
+  local_file="$file_path"
   s3_uri="$s3_uri_base"
 fi
 

src/install.sh

@@ -11,6 +11,10 @@ apk add postgresql-client
 # install gpg
 apk add gnupg
 
+# install zstd
+apk add zstd
+
+# install aws-cli
 apk add aws-cli
 
 # install go-cron

src/restore.sh

@@ -26,19 +26,27 @@ else
   )
 fi
 
+file_path="${LOCAL_STORE_PREFIX}/db${file_type}"
+
 echo "Fetching backup from S3..."
-aws $aws_args s3 cp "${s3_uri_base}/${key_suffix}" "db${file_type}"
+aws $aws_args s3 cp "${s3_uri_base}/${key_suffix}" "$file_path"
 
 if [ -n "$PASSPHRASE" ]; then
   echo "Decrypting backup..."
-  gpg --decrypt --batch --passphrase "$PASSPHRASE" db.dump.gpg > db.dump
-  rm db.dump.gpg
+  gpg --decrypt --batch --passphrase "$PASSPHRASE" "$file_path.gpg" > "${LOCAL_STORE_PREFIX}/db.dump"
+  rm "$file_path.gpg"
+fi
+
+if [ -n "$COMPRESS" ]; then
+  echo "Decompressing backup..."
+  zstd -d "${LOCAL_STORE_PREFIX}/db.dump.zst" -o "${LOCAL_STORE_PREFIX}/db.dump"
+  rm "${LOCAL_STORE_PREFIX}/db.dump.zst"
 fi
 
 conn_opts="-h $POSTGRES_HOST -p $POSTGRES_PORT -U $POSTGRES_USER -d $POSTGRES_DATABASE"
 
 echo "Restoring from backup..."
-pg_restore $conn_opts --clean --if-exists db.dump
+pg_restore $conn_opts --clean --if-exists "${LOCAL_STORE_PREFIX}/db.dump"
 rm db.dump
 
 echo "Restore complete."