Skip to content

Instantly share code, notes, and snippets.

@jessiel-hacke
Created November 26, 2018 15:59
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save jessiel-hacke/2d439a59f1cf273a2d556252bd74933c to your computer and use it in GitHub Desktop.
Save jessiel-hacke/2d439a59f1cf273a2d556252bd74933c to your computer and use it in GitHub Desktop.
##############################
## POSTGRESQL BACKUP CONFIG ##
##############################
# Optional system user to run backups as. If the user the script is running as doesn't match this
# the script terminates. Leave blank to skip check.
BACKUP_USER=postgres
# Optional hostname to adhere to pg_hba policies. Will default to "localhost" if none specified.
HOSTNAME=localhost
# Optional username to connect to database as. Will default to "postgres" if none specified.
USERNAME=postgres
# This dir will be created if it doesn't exist. This must be writable by the user the script is
# running as.
BACKUP_DIR=/data/opt/pgbackup/backups/
# List of strings to match against in database name, separated by space or comma, for which we only
# wish to keep a backup of the schema, not the data. Any database names which contain any of these
# values will be considered candidates. (e.g. "system_log" will match "dev_system_log_2010-01")
SCHEMA_ONLY_LIST=""
# Will produce a custom-format backup if set to "yes"
ENABLE_CUSTOM_BACKUPS=no
# Will produce a gzipped plain-format backup if set to "yes"
ENABLE_PLAIN_BACKUPS=yes
# Will produce gzipped sql file containing the cluster globals, like users and passwords, if set to "yes"
ENABLE_GLOBALS_BACKUPS=yes
#### SETTINGS FOR ROTATED BACKUPS ####
## Daily Backups ##
ENABLE_DAILY_BACKUP=yes # Enable daily backups (default: no)
DAYS_TO_KEEP=2 # Number of days to keep daily backups (default: 7)
## Weekly Backups ##
ENABLE_WEEKLY_BACKUP=no # Enable weekly backups (default: no)
WEEKS_TO_KEEP=4 # How many weeks to keep weekly backups (default: 4)
DAY_OF_WEEK_TO_KEEP=5 # Which day to take the weekly backup from (default: 5, options: 1-7 = Monday-Sunday)
## Monthly Backups ##
ENABLE_MONTHLY_BACKUP=no # Enable monthly backups (default: no)
DAY_OF_MONTH_TO_KEEP=01 # Which day to take the monthly backup from (default: 01 = first of the month)
#!/bin/bash
BACKUP_FOLDER="/data/opt/pgbackup/backups/"
###########################
########### LOG ###########
###########################
log()
{
MESSAGE=$1
if [ -z $LOG_FILE ]; then
LOG_FILE="${BACKUP_FOLDER}backup.log"
fi
echo "`date '+%Y-%m-%d %H:%M:%S'` $MESSAGE" >> $LOG_FILE
}
###########################
####### LOAD CONFIG #######
###########################
while [ $# -gt 0 ]; do
case $1 in
-c)
CONFIG_FILE_PATH="$2"
shift 2
;;
*)
log "Unknown Option \"$1\""
exit 2
;;
esac
done
if [ -z $CONFIG_FILE_PATH ] ; then
SCRIPTPATH=''
CONFIG_FILE_PATH="/data/opt/pgbackup/scripts/backup.config"
log "Loaded config file from ${CONFIG_FILE_PATH}"
fi
if [ ! -r ${CONFIG_FILE_PATH} ] ; then
log "Could not load config file from ${CONFIG_FILE_PATH}"
exit 1
fi
source "${CONFIG_FILE_PATH}"
###########################
#### PRE-BACKUP CHECKS ####
###########################
# Make sure we're running as the required backup user
if [ "$BACKUP_USER" != "" -a "$(id -un)" != "$BACKUP_USER" ] ; then
log "This script must be run as $BACKUP_USER. Exiting."
exit 1
fi
###########################
### INITIALISE DEFAULTS ###
###########################
if [ ! $HOSTNAME ]; then
HOSTNAME="localhost"
fi;
if [ ! $USERNAME ]; then
USERNAME="postgres"
fi;
###########################
#### START THE BACKUPS ####
###########################
function perform_backups()
{
SUFFIX=$1
FINAL_BACKUP_DIR=$BACKUP_DIR"`date +\%Y-\%m-\%d`$SUFFIX/"
log "Making backup directory in $FINAL_BACKUP_DIR"
if ! mkdir -p $FINAL_BACKUP_DIR; then
log "Cannot create backup directory in $FINAL_BACKUP_DIR. Go and fix it!"
exit 1;
fi;
#######################
### GLOBALS BACKUPS ###
#######################
log "Performing globals backup"
log "--------------------------------------------"
if [ $ENABLE_GLOBALS_BACKUPS = "yes" ]
then
log "Globals backup"
if ! pg_dumpall -g -U "$USERNAME" | gzip > $FINAL_BACKUP_DIR"globals".sql.gz.in_progress; then
log "[!!ERROR!!] Failed to produce globals backup"
else
mv $FINAL_BACKUP_DIR"globals".sql.gz.in_progress $FINAL_BACKUP_DIR"globals".sql.gz
fi
else
log "None"
fi
###########################
### SCHEMA-ONLY BACKUPS ###
###########################
for SCHEMA_ONLY_DB in ${SCHEMA_ONLY_LIST//,/ }
do
SCHEMA_ONLY_CLAUSE="$SCHEMA_ONLY_CLAUSE or datname ~ '$SCHEMA_ONLY_DB'"
done
SCHEMA_ONLY_QUERY="select datname from pg_database where false $SCHEMA_ONLY_CLAUSE order by datname;"
log "Performing schema-only backups"
log "--------------------------------------------"
SCHEMA_ONLY_DB_LIST=`psql -U "$USERNAME" -At -c "$SCHEMA_ONLY_QUERY" postgres`
log "The following databases were matched for schema-only backup:${SCHEMA_ONLY_DB_LIST}"
for DATABASE in $SCHEMA_ONLY_DB_LIST
do
log "Schema-only backup of $DATABASE"
if ! pg_dump -Fp -s -U "$USERNAME" "$DATABASE" | gzip > $FINAL_BACKUP_DIR"$DATABASE"_SCHEMA.sql.gz.in_progress; then
log "[!!ERROR!!] Failed to backup database schema of $DATABASE"
else
mv $FINAL_BACKUP_DIR"$DATABASE"_SCHEMA.sql.gz.in_progress $FINAL_BACKUP_DIR"$DATABASE"_SCHEMA.sql.gz
fi
done
###########################
###### FULL BACKUPS #######
###########################
for SCHEMA_ONLY_DB in ${SCHEMA_ONLY_LIST//,/ }
do
EXCLUDE_SCHEMA_ONLY_CLAUSE="$EXCLUDE_SCHEMA_ONLY_CLAUSE and datname !~ '$SCHEMA_ONLY_DB'"
done
FULL_BACKUP_QUERY="select datname from pg_database where not datistemplate and datallowconn $EXCLUDE_SCHEMA_ONLY_CLAUSE order by datname;"
log "Performing full backups"
log "--------------------------------------------"
for DATABASE in `psql -U "$USERNAME" -At -c "$FULL_BACKUP_QUERY" postgres`
do
if [ $ENABLE_PLAIN_BACKUPS = "yes" ]
then
log "Plain backup of $DATABASE"
if ! pg_dump -Fp -U "$USERNAME" "$DATABASE" | gzip > $FINAL_BACKUP_DIR"$DATABASE".sql.gz.in_progress; then
log "[!!ERROR!!] Failed to produce plain backup database $DATABASE"
else
mv $FINAL_BACKUP_DIR"$DATABASE".sql.gz.in_progress $FINAL_BACKUP_DIR"$DATABASE".sql.gz
fi
fi
if [ $ENABLE_CUSTOM_BACKUPS = "yes" ]
then
log "Custom backup of $DATABASE"
if ! pg_dump -Fc -U "$USERNAME" "$DATABASE" -f $FINAL_BACKUP_DIR"$DATABASE".custom.in_progress; then
log "[!!ERROR!!] Failed to produce custom backup database $DATABASE"
else
mv $FINAL_BACKUP_DIR"$DATABASE".custom.in_progress $FINAL_BACKUP_DIR"$DATABASE".custom
fi
fi
done
log "All database backups complete!"
}
# MONTHLY BACKUPS
DAY_OF_MONTH=`date +%d`
if [ $ENABLE_MONTHLY_BACKUP = yes -a $DAY_OF_MONTH = $DAY_OF_MONTH_TO_KEEP ];
then
# Delete all expired monthly directories
find $BACKUP_DIR -maxdepth 1 -name "*-monthly" -exec rm -rf '{}' ';'
perform_backups "-monthly"
exit 0;
else
log "Skipped monthly backup!"
fi
# WEEKLY BACKUPS
DAY_OF_WEEK=`date +%u` #1-7 (Monday-Sunday)
EXPIRED_DAYS=`expr $((($WEEKS_TO_KEEP * 7) + 1))`
if [ $ENABLE_WEEKLY_BACKUP = yes -a $DAY_OF_WEEK = $DAY_OF_WEEK_TO_KEEP ];
then
# Delete all expired weekly directories
find $BACKUP_DIR -maxdepth 1 -mtime $EXPIRED_DAYS -name "*-weekly" -exec rm -rf '{}' ';'
perform_backups "-weekly"
exit 0;
else
log "Skipped weekly backup!"
fi
# DAILY BACKUPS
# Delete daily backups 7 days old or more
if [ $ENABLE_DAILY_BACKUP = yes ];
then
find $BACKUP_DIR -maxdepth 1 -mtime +$(($DAYS_TO_KEEP-1)) -name "*-daily" -exec rm -rf '{}' ';'
perform_backups "-daily"
exit 0;
else
log "Skipped daily backup!"
fi
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment