jkaninda/pg-bkup
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Merge pull request #17 from jkaninda/develop

Browse Source 
Refactoring of code
This commit is contained in:
Jonas Kaninda
2023-12-27 21:47:16 +01:00
committed by GitHub
parent 4c6dd089a1 3d885bfb44
commit 908f9081e2
1 changed files with 32 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
src/pg_bkup.sh
View File

@@ -45,7 +45,20 @@ error()
echo "$arg0: $*" >&2 echo "$arg0: $*" >&2
exit 0 exit 0
} }
info() {
{ set +x; } 2> /dev/null
echo 'pg-bkup:' '[INFO] ' "$@"
#set -x
}
warning() {
{ set +x; } 2> /dev/null
echo 'pg-bkup:' '[WARNING] ' "$@"
}
fatal() {
{ set +x; } 2> /dev/null
echo 'pg-bkup:' '[ERROR] ' "$@" >&2
exit 1
}
help() help()
{ {
echo echo
@@ -137,16 +150,16 @@ create_pgpass(){
backup() backup()
{ {
if [[ -z $DB_HOST ]] || [[ -z $DB_NAME ]] || [[ -z $DB_USERNAME ]] || [[ -z $DB_PASSWORD ]]; then if [[ -z $DB_HOST ]] || [[ -z $DB_NAME ]] || [[ -z $DB_USERNAME ]] || [[ -z $DB_PASSWORD ]]; then
echo "Please make sure all required options are set " fatal "Please make sure all required options are set "
else else
export PGPASSWORD=${DB_PASSWORD} export PGPASSWORD=${DB_PASSWORD}
## Test database connection ## Test database connection
## Backup database ## Backup database
pg_dump -h ${DB_HOST} -p ${DB_PORT} -U ${DB_USERNAME} -d ${DB_NAME} -v | gzip > ${STORAGE_PATH}/${DB_NAME}_${TIME}.sql.gz pg_dump -h ${DB_HOST} -p ${DB_PORT} -U ${DB_USERNAME} -d ${DB_NAME} | gzip > ${STORAGE_PATH}/${DB_NAME}_${TIME}.sql.gz
echo "$TIME: ${DB_NAME}_${TIME}.sql.gz" | tee -a "${STORAGE_PATH}/history.txt" echo "$TIME: ${DB_NAME}_${TIME}.sql.gz" | tee -a "${STORAGE_PATH}/history.txt"
echo "Database has been saved" info "Database has been saved"
fi fi
exit 0 exit 0
} }
@@ -154,7 +167,7 @@ exit 0
restore() restore()
{ {
if [[ -z $DB_HOST ]] || [[ -z $DB_NAME ]] || [[ -z $DB_USERNAME ]] || [[ -z $DB_PASSWORD ]]; then if [[ -z $DB_HOST ]] || [[ -z $DB_NAME ]] || [[ -z $DB_USERNAME ]] || [[ -z $DB_PASSWORD ]]; then
echo "Please make sure all required options are set " fatal "Please make sure all required options are set "
else else
## Restore database ## Restore database
export PGPASSWORD=$DB_PASSWORD export PGPASSWORD=$DB_PASSWORD
@@ -164,9 +177,9 @@ else
else else
cat ${STORAGE_PATH}/${FILE_NAME} | psql -h ${DB_HOST} -p ${DB_PORT} -U ${DB_USERNAME} -v -d ${DB_NAME} cat ${STORAGE_PATH}/${FILE_NAME} | psql -h ${DB_HOST} -p ${DB_PORT} -U ${DB_USERNAME} -v -d ${DB_NAME}
fi fi
echo "Database has been restored" info "Database has been restored"
else else
echo "Error, file not found in ${STORAGE_PATH}/${FILE_NAME}" fatal "File not found in ${STORAGE_PATH}/${FILE_NAME}"
fi fi
fi fi
exit exit
@@ -188,19 +201,18 @@ s3_restore()
mount_s3() mount_s3()
{ {
if [[ -z $ACCESS_KEY ]] || [[ -z $SECRET_KEY ]]; then if [[ -z $ACCESS_KEY ]] || [[ -z $SECRET_KEY ]]; then
echo "Please make sure all environment variables are set " fatal "Please make sure all environment variables are set "
echo "BUCKETNAME=$BUCKETNAME \nACCESS_KEY=$nACCESS_KEY \nSECRET_KEY=$SECRET_KEY"
else else
echo "$ACCESS_KEY:$SECRET_KEY" | tee /etc/passwd-s3fs echo "$ACCESS_KEY:$SECRET_KEY" | tee /etc/passwd-s3fs
chmod 600 /etc/passwd-s3fs chmod 600 /etc/passwd-s3fs
echo "Mounting Object storage in /s3mnt .... " info "Mounting Object storage in /s3mnt .... "
if [ -z "$(ls -A /s3mnt)" ]; then if [ -z "$(ls -A /s3mnt)" ]; then
s3fs $BUCKETNAME /s3mnt -o passwd_file=/etc/passwd-s3fs -o use_cache=/tmp/s3cache -o allow_other -o url=$S3_ENDPOINT -o use_path_request_style s3fs $BUCKETNAME /s3mnt -o passwd_file=/etc/passwd-s3fs -o use_cache=/tmp/s3cache -o allow_other -o url=$S3_ENDPOINT -o use_path_request_style
if [ ! -d "/s3mnt$S3_PATH" ]; then if [ ! -d "/s3mnt$S3_PATH" ]; then
mkdir -p /s3mnt$S3_PATH mkdir -p /s3mnt$S3_PATH
fi fi
else else
echo "Object storage already mounted in /s3mnt" info "Object storage already mounted in /s3mnt"
fi fi
export STORAGE_PATH=/s3mnt$S3_PATH export STORAGE_PATH=/s3mnt$S3_PATH
fi fi
@@ -244,13 +256,12 @@ scheduled_mode()
echo "**********************************" echo "**********************************"
echo " Starting PostgreSQL Bkup... " echo " Starting PostgreSQL Bkup... "
echo "***********************************" echo "***********************************"
echo "Running in Scheduled mode " info "Running in Scheduled mode "
echo "Execution period $SCHEDULE_PERIOD" info "Execution period $SCHEDULE_PERIOD"
echo "Log file in /var/log/pg-bkup.log " info "Log file in /var/log/pg-bkup.log "
supervisord -c /etc/supervisor/supervisord.conf supervisord -c /etc/supervisor/supervisord.conf
else else
echo "Scheduled mode supports only backup operation" fatal "Scheduled mode supports only backup operation"
exit 1
fi fi
} }
@@ -262,19 +273,19 @@ then
then then
if [ $STORAGE != 's3' ] if [ $STORAGE != 's3' ]
then then
echo "Restore from local" info "Restore from local"
restore restore
else else
echo "Restore from s3" info "Restore from s3"
s3_restore s3_restore
fi fi
else else
if [ $STORAGE != 's3' ] if [ $STORAGE != 's3' ]
then then
echo "Backup to local destination" info "Backup to local storage"
backup backup
else else
echo "Backup to s3 storage" info "Backup to s3 storage"
s3_backup s3_backup
fi fi
fi fi
@@ -282,6 +293,5 @@ elif [ $EXECUTION_MODE == 'scheduled' ]
then then
scheduled_mode scheduled_mode
else else
echo "Error, unknow execution mode!" fatal "Unknow execution mode"
exit 1
fi fi