forked from ISTI-ansible-roles/ansible-roles
Change the postgresql backup script so that it can backup local databases as postgres user, without authentication.
This commit is contained in:
Andrea Dell'Amico
parent
5a4d729503
commit
3bc6406f44
|
|
@ -1,12 +1,14 @@
|
|||
#!/bin/bash
|
||||
|
||||
|
||||
# Set up the environment
|
||||
if [ -f /etc/default/pg_backup ] ; then
|
||||
. /etc/default/pg_backup
|
||||
else
|
||||
N_DAYS_TO_SPARE=7
|
||||
USE_NAGIOS=no
|
||||
BUILD_DBLIST=yes
|
||||
PG_USE_AUTH=yes
|
||||
PG_USE_AUTH=no
|
||||
PG_PASS_FILE=/root/.pgpass
|
||||
BACKUPDIR=/var/lib/pgsql/backups
|
||||
DB_LIST=
|
||||
|
|
@ -22,39 +24,35 @@ TIMESTAMP_LOG=$BACKUPDIR/.timestamp
|
|||
# If nagios is active, save the report status for each backup
|
||||
# Nagios return values: 0 = OK, 1 = WARNING, 2 = CRITICAL, 3 = UNKNOWN
|
||||
NAGIOS_LOG=$BACKUPDIR/.nagios-status
|
||||
|
||||
if [ ! -d ${BACKUPDIR} ] ; then
|
||||
mkdir -p ${BACKUPDIR}
|
||||
fi
|
||||
if [ ! -d ${HISTDIR} ] ; then
|
||||
mkdir -p ${HISTDIR}
|
||||
fi
|
||||
DUMP_RESULT=0
|
||||
LOCKFILE=${BACKUPDIR}/.dumplock
|
||||
|
||||
umask 0077
|
||||
if [ "$BUILD_DBLIST" == "yes" ] ; then
|
||||
# The psql -l command prints too much stuff
|
||||
create_backup_dirs() {
|
||||
if [ ! -d ${BACKUPDIR} ] ; then
|
||||
mkdir -p ${BACKUPDIR}
|
||||
fi
|
||||
if [ ! -d ${HISTDIR} ] ; then
|
||||
mkdir -p ${HISTDIR}
|
||||
fi
|
||||
if [ "${PG_USE_AUTH}" == "no" ] ; then
|
||||
chown -R postgres:postgres $BACKUPDIR
|
||||
fi
|
||||
}
|
||||
|
||||
cleanup_old_backups() {
|
||||
# Remove the old backups
|
||||
find ${HISTDIR} -ctime +$N_DAYS_TO_SPARE -exec rm -f {} \;
|
||||
}
|
||||
|
||||
create_db_list() {
|
||||
# The psql -l command prints too much stuff
|
||||
#DB_LIST=$( psql -q -t -l -U postgres | grep -v template0 | grep -v template1 | grep -v : | grep -v ^\( | grep -v ^\- | awk '{print $1}' )
|
||||
DB_LIST=$( su - postgres -c "/usr/lib/postgresql/${PG_VERSION}/bin/oid2name " | awk '{print $2}' | tail -n +4 | grep -v template0 | grep -v template1 )
|
||||
fi
|
||||
}
|
||||
|
||||
if [ ! -f $LOCKFILE ] ; then
|
||||
touch $LOCKFILE
|
||||
if [ "$USE_NAGIOS" == "yes" ] ; then
|
||||
> $NAGIOS_LOG
|
||||
fi
|
||||
if [ "${PG_USE_AUTH}" == "yes" -a ! -f $PG_PASS_FILE ] ; then
|
||||
if [ "$USE_NAGIOS" == "yes" ] ; then
|
||||
echo ".pgpass file not found, but authentication needed. All dbs: FAILED" >> $NAGIOS_LOG
|
||||
fi
|
||||
RETVAL=2
|
||||
else
|
||||
# Backup of the db system data
|
||||
# pg_dumpall -U postgres -g > ${HISTDIR}/pgsql-global.data.$SAVE_TIME
|
||||
# Dump all the databases
|
||||
run_backup_as_db_user() {
|
||||
for db in $DB_LIST ; do
|
||||
if [ "${PG_USE_AUTH}" == "yes" ] ; then
|
||||
DB_IN_AUTFILE=$( grep :${db}: $PG_PASS_FILE )
|
||||
grep ":${db}:" "$PG_PASS_FILE" 2>/dev/null
|
||||
DB_IN_AUTFILE_RETVAL=$?
|
||||
if [ $DB_IN_AUTFILE_RETVAL -eq 0 ] ; then
|
||||
PG_HOST=$( grep :${db}: $PG_PASS_FILE | cut -d : -f 1 )
|
||||
|
|
@ -65,11 +63,32 @@ if [ ! -f $LOCKFILE ] ; then
|
|||
else
|
||||
DUMP_RESULT=2
|
||||
fi
|
||||
else
|
||||
${PG_DUMP_BIN} -Fc -U postgres $db > ${HISTDIR}/$db.data.$SAVE_TIME
|
||||
DUMP_RESULT=$?
|
||||
fi
|
||||
pushd ${BACKUPDIR}/ >/dev/null 2>&1
|
||||
rm -f $db.data
|
||||
ln -s ${HISTDIR}/$db.data.$SAVE_TIME ./$db.data
|
||||
popd >/dev/null 2>&1
|
||||
done
|
||||
}
|
||||
|
||||
run_backup_as_postgres() {
|
||||
su - postgres -c "pg_dumpall -g > ${HISTDIR}/pgsql-global.data.$SAVE_TIME"
|
||||
for db in $DB_LIST ; do
|
||||
su - postgres -c "${PG_DUMP_BIN} -Fc $db > ${HISTDIR}/$db.data.$SAVE_TIME"
|
||||
DUMP_RESULT=$?
|
||||
pushd ${BACKUPDIR}/ >/dev/null 2>&1
|
||||
rm -f $db.data
|
||||
ln -s ${HISTDIR}/$db.data.$SAVE_TIME ./$db.data
|
||||
popd >/dev/null 2>&1
|
||||
done
|
||||
}
|
||||
|
||||
clear_nagios_data() {
|
||||
if [ "$USE_NAGIOS" == "yes" ] ; then
|
||||
> $NAGIOS_LOG
|
||||
fi
|
||||
}
|
||||
|
||||
write_nagios_data() {
|
||||
if [ "$USE_NAGIOS" == "yes" ] ; then
|
||||
if [ $DUMP_RESULT -ne 0 ] ; then
|
||||
echo "$db:FAILED" >> $NAGIOS_LOG
|
||||
|
|
@ -78,12 +97,36 @@ if [ ! -f $LOCKFILE ] ; then
|
|||
echo "$db:OK" >> $NAGIOS_LOG
|
||||
fi
|
||||
fi
|
||||
pushd ${BACKUPDIR}/ >/dev/null 2>&1
|
||||
rm -f $db.data
|
||||
ln -s ${HISTDIR}/$db.data.$SAVE_TIME ./$db.data
|
||||
popd >/dev/null 2>&1
|
||||
done
|
||||
}
|
||||
|
||||
|
||||
########
|
||||
#
|
||||
# Main
|
||||
#
|
||||
umask 0077
|
||||
|
||||
create_backup_dirs
|
||||
if [ "$BUILD_DBLIST" == "yes" ] ; then
|
||||
create_db_list
|
||||
fi
|
||||
|
||||
if [ ! -f $LOCKFILE ] ; then
|
||||
touch $LOCKFILE
|
||||
clear_nagios_data
|
||||
if [ "${PG_USE_AUTH}" == "yes" ] ; then
|
||||
if [ ! -f $PG_PASS_FILE -o -z $PG_PASS_FILE ] ; then
|
||||
if [ "$USE_NAGIOS" == "yes" ] ; then
|
||||
echo ".pgpass file not found or empty but authentication needed. All db backups FAILED" >> $NAGIOS_LOG
|
||||
fi
|
||||
RETVAL=2
|
||||
exit 2
|
||||
fi
|
||||
run_backup_as_db_user
|
||||
else
|
||||
run_backup_as_postgres
|
||||
fi
|
||||
write_nagios_data
|
||||
TIMESTAMP=$( date +%s )
|
||||
echo "$TIMESTAMP" > $TIMESTAMP_LOG
|
||||
rm -f $LOCKFILE
|
||||
|
|
@ -94,8 +137,6 @@ else
|
|||
fi
|
||||
fi
|
||||
|
||||
|
||||
# Remove the old backups
|
||||
find ${HISTDIR} -ctime +$N_DAYS_TO_SPARE -exec rm -f {} \;
|
||||
cleanup_old_backups
|
||||
|
||||
exit $RETVAL
|
||||
|
|
|
|||
Loading…
Reference in New Issue