0c72c82ec04294ee6a0f610a1e196aa9f9be2c0a
[matthijs/upstream/backupninja.git] / handlers / pgsql
1 # -*- mode: sh; sh-basic-offset: 3; indent-tabs-mode: nil; -*-
2 #
3 # PostgreSQL handler script for backupninja
4 #
5
6 getconf backupdir /var/backups/postgres
7 getconf databases all
8 getconf compress yes
9 getconf vsname
10
11 localhost=`hostname`
12
13 # Decide if the handler should operate on a vserver or on the host.
14 # In the former case, check that $vsname exists and is running.
15 local usevserver=no
16 local vroot
17 if [ $vservers_are_available = yes ]; then
18    if [ -n "$vsname" ]; then
19       # does it exist ?
20       if ! vservers_exist "$vsname" ; then
21          fatal "The vserver given in vsname ($vsname) does not exist."
22       fi
23       # is it running ?
24       $VSERVERINFO -q $vsname RUNNING
25       if [ $? -ne 0 ]; then
26          fatal "The vserver $vsname is not running."
27       fi
28       # everything ok
29       info "Using vserver '$vsname'."
30       usevserver=yes
31       vroot="$VROOTDIR/$vsname"
32    else
33       info "No vserver name specified, actions will be performed on the host."
34    fi
35 fi
36
37 # Make sure that the system to backup has the needed executables
38 if [ $usevserver = yes ]; then
39    debug "Examining vserver '$vsname'."
40    if [ "$databases" == "all" ]; then
41       [ -x "$vroot`$VSERVER $vsname exec which $PGSQLDUMPALL`" ] || \
42          fatal "Can't find $PGSQLDUMPALL in vserver $vsname."
43    else
44       [ -x "$vroot`$VSERVER $vsname exec which $PGSQLDUMP`" ] || \
45          fatal "Can't find $PGSQLDUMP in vserver $vsname."
46    fi
47 else
48    if [ "$databases" == "all" ]; then
49       [ -x "`which $PGSQLDUMPALL`" ] || \
50          fatal "Can't find $PGSQLDUMPALL."
51    else
52       [ -x "`which $PGSQLDUMP`" ] || \
53          fatal "Can't find $PGSQLDUMP."
54    fi
55 fi
56
57 # create backup dir, the vroot variable will be empty if no vsname was specified
58 # and will proceed to operate on the host
59 [ -d $vroot$backupdir ] || (debug "mkdir -p $vroot$backupdir"; mkdir -p $vroot$backupdir)
60 [ -d $vroot$backupdir ] || fatal "Backup directory '$vroot$backupdir'"
61
62 # give backup dir the good uid and permissions
63 # (in respect to the vserver, if $usevserver = yes)
64 if [ $usevserver = yes ]; then
65    pguid=`$VSERVER $vsname exec getent passwd postgres | awk -F: '{print $3}'`
66 else
67    pguid=`getent passwd postgres | awk -F: '{print $3}'`
68 fi
69 [ -n "$pguid" ] || \
70     fatal "No user called postgres`[ $usevserver = no ] || echo \" on vserver $vsname\"`."
71 debug "chown $pguid $vroot$backupdir"
72 chown $pguid $vroot$backupdir
73 debug "chmod 700 $vroot$backupdir"
74 chmod 700 $vroot$backupdir
75
76 # if $databases = all, use pg_dumpall
77 if [ "$databases" == "all" ]; then
78     if [ $usevserver = yes ]; then
79         execstr="$VSERVER $vsname exec su - postgres -c \"$PGSQLDUMPALL > $backupdir/${vsname}.sql\""
80     else
81         execstr="su - postgres -c \"$PGSQLDUMPALL > $backupdir/${localhost}-all.sql\""
82     fi
83     debug "$execstr"
84     if [ ! $test ]; then
85        output=`eval $execstr 2>&1`
86        code=$?
87         if [ "$code" == "0" ]; then
88             debug $output
89             info "Successfully finished dump of pgsql cluster"
90         else
91             warning $output
92             warning "Failed to dump pgsql cluster"
93         fi
94     fi
95     
96 # else use pg_dump on each specified database
97 else
98     for db in $databases; do
99         if [ $usevserver = yes ]
100             then
101             execstr="$VSERVER $vsname exec su - postgres -c \"$PGSQLDUMP $db > $backupdir/${db}.sql\""
102         else
103             execstr="su - postgres -c \"$PGSQLDUMP $db > $backupdir/${db}.sql\""
104         fi
105         debug "$execstr"
106         if [ ! $test ]; then
107             output=`eval $execstr 2>&1`
108             code=$?
109             if [ "$code" == "0" ]; then
110                 debug $output
111                 info "Successfully finished dump of pgsql database ${db}"
112             else
113                 warning $output
114                 warning "Failed to dump pgsql database ${db}"
115             fi
116         fi
117     done
118 fi
119
120 if [ "$compress" == "yes" ]; then
121     output=`$GZIP -f $vroot$backupdir/*.sql 2>&1`
122     debug $output
123 fi
124
125 return 0
126