Refactoring of code

This commit is contained in:
2023-12-23 22:02:26 +01:00
parent 1545e5ed17
commit 5c22d09a99
3 changed files with 10 additions and 9 deletions

3
.gitignore vendored
View File

@@ -2,4 +2,5 @@
backup backup
data data
compose.yaml compose.yaml
.env .env
test.md

View File

@@ -109,7 +109,7 @@ bkup --operation restore --storage s3 --file database_20231217_115621.sql --dbna
## Docker run: ## Docker run:
```sh ```sh
docker run --rm --network your_network_name --name pg-bkup -v $PWD/backup:/backup/ -e "DB_HOST=database_host_name" -e "DB_USERNAME=username" -e "DB_PASSWORD=password" jkaninda/pg-bkup:latest bkup -o backup -d database_name -f napata_20231219_022941.sql.gz docker run --rm --network your_network_name --name pg-bkup -v $PWD/backup:/backup/ -e "DB_HOST=database_host_name" -e "DB_USERNAME=username" -e "DB_PASSWORD=password" jkaninda/pg-bkup bkup -o restore -d database_name -f napata_20231219_022941.sql.gz
``` ```
## Docker compose file: ## Docker compose file:

View File

@@ -116,7 +116,7 @@ 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 " echo "Please make sure all required options are set "
else else
export PGPASSWORD=${DB_PASSWORD} export PGPASSWORD=${DB_PASSWORD}
@@ -124,6 +124,8 @@ else
## 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} -v | gzip > ${STORAGE_PATH}/${DB_NAME}_${TIME}.sql.gz
echo "$TIME: ${DB_NAME}_${TIME}.sql.gz" | tee -a "${STORAGE_PATH}/history.txt"
echo "Database has been saved" echo "Database has been saved"
fi fi
exit 0 exit 0
@@ -131,15 +133,13 @@ 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 " echo "Please make sure all required options are set "
else else
## Restore database ## Restore database
export PGPASSWORD=${DB_PASSWORD} export PGPASSWORD=$DB_PASSWORD
if [ -f "${STORAGE_PATH}/$FILE_NAME" ]; then if [ -f "${STORAGE_PATH}/$FILE_NAME" ]; then
#pg_restore -h ${DB_HOST} -P ${DB_PORT} -U ${DB_USERNAME} -v -d ${DB_NAME} ${STORAGE_PATH}/$FILE_NAME if gzip -t $STORAGE_PATH/$FILE_NAME; then
#cat ${STORAGE_PATH}/${FILE_NAME} | psql -h ${DB_HOST} -p ${DB_PORT} -U ${DB_USERNAME} -v -d ${DB_NAME} < ${STORAGE_PATH}/$FILE_NAME
if gzip -t ${STORAGE_PATH}/$FILE_NAME; then
zcat ${STORAGE_PATH}/${FILE_NAME} | psql -h ${DB_HOST} -p ${DB_PORT} -U ${DB_USERNAME} -v -d ${DB_NAME} zcat ${STORAGE_PATH}/${FILE_NAME} | psql -h ${DB_HOST} -p ${DB_PORT} -U ${DB_USERNAME} -v -d ${DB_NAME}
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}
@@ -167,7 +167,7 @@ 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 " echo "Please make sure all environment variables are set "
echo "BUCKETNAME=$BUCKETNAME \nACCESS_KEY=$nACCESS_KEY \nSECRET_KEY=$SECRET_KEY" echo "BUCKETNAME=$BUCKETNAME \nACCESS_KEY=$nACCESS_KEY \nSECRET_KEY=$SECRET_KEY"
else else