mysql: No longer prepend vroot to the backupdir.
[matthijs/upstream/backupninja-vserver.git] / handlers / pgsql.in
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       vservers_running $vsname || fatal "The vserver $vsname is not running."
25       # everything ok
26       info "Using vserver '$vsname'."
27       usevserver=yes
28       vroot="$VROOTDIR/$vsname"
29    else
30       info "No vserver name specified, actions will be performed on the host."
31    fi
32 fi
33
34 # Make sure that the system to backup has the needed executables
35 if [ $usevserver = yes ]; then
36    debug "Examining vserver '$vsname'."
37    if [ "$databases" == "all" ]; then
38       [ -x "$vroot`$VSERVER $vsname exec which $PGSQLDUMPALL`" ] || \
39          fatal "Can't find $PGSQLDUMPALL in vserver $vsname."
40    else
41       [ -x "$vroot`$VSERVER $vsname exec which $PGSQLDUMP`" ] || \
42          fatal "Can't find $PGSQLDUMP in vserver $vsname."
43    fi
44 else
45    if [ "$databases" == "all" ]; then
46       [ -x "`which $PGSQLDUMPALL`" ] || \
47          fatal "Can't find $PGSQLDUMPALL."
48    else
49       [ -x "`which $PGSQLDUMP`" ] || \
50          fatal "Can't find $PGSQLDUMP."
51    fi
52 fi
53
54 # create backup dir, the vroot variable will be empty if no vsname was specified
55 # and will proceed to operate on the host
56 [ -d $vroot$backupdir ] || (debug "mkdir -p $vroot$backupdir"; mkdir -p $vroot$backupdir)
57 [ -d $vroot$backupdir ] || fatal "Backup directory '$vroot$backupdir' does not exist, and could not be created."
58
59 # give backup dir the good uid and permissions
60 # (in respect to the vserver, if $usevserver = yes)
61 if [ $usevserver = yes ]; then
62    pguid=`$VSERVER $vsname exec getent passwd $PGSQLUSER | @AWK@ -F: '{print $3}'`
63 else
64    pguid=`getent passwd $PGSQLUSER | @AWK@ -F: '{print $3}'`
65 fi
66 [ -n "$pguid" ] || \
67     fatal "No user called $PGSQLUSER`[ $usevserver = no ] || echo \" on vserver $vsname\"`."
68 debug "chown $pguid $vroot$backupdir"
69 chown $pguid $vroot$backupdir
70 debug "chmod 700 $vroot$backupdir"
71 chmod 700 $vroot$backupdir
72
73 # if $databases = all, use pg_dumpall
74 if [ "$databases" == "all" ]; then
75     if [ $usevserver = yes ]; then
76        if [ "$compress" == "yes" ]; then          
77           execstr="$VSERVER $vsname exec su - $PGSQLUSER -c \"$PGSQLDUMPALL | $GZIP > $backupdir/${vsname}.sql.gz\""
78        else
79           execstr="$VSERVER $vsname exec su - $PGSQLUSER -c \"$PGSQLDUMPALL > $backupdir/${vsname}.sql\""
80        fi
81     else
82        if [ "$compress" == "yes" ]; then
83           execstr="su - $PGSQLUSER -c \"$PGSQLDUMPALL | $GZIP > $backupdir/${localhost}-all.sql.gz\""
84        else
85         execstr="su - $PGSQLUSER -c \"$PGSQLDUMPALL > $backupdir/${localhost}-all.sql\""
86        fi
87     fi
88     debug "$execstr"
89     if [ ! $test ]; then
90        output=`eval $execstr 2>&1`
91        code=$?
92         if [ "$code" == "0" ]; then
93             debug $output
94             info "Successfully finished dump of pgsql cluster"
95         else
96             warning $output
97             warning "Failed to dump pgsql cluster"
98         fi
99     fi
100     
101 # else use pg_dump on each specified database
102 else
103     for db in $databases; do
104         if [ $usevserver = yes ]; then
105            if [ "$compress" == "yes" ]; then
106               execstr="$VSERVER $vsname exec su - $PGSQLUSER -c \"$PGSQLDUMP $db | $GZIP > $backupdir/${db}.sql.gz\""
107            else
108               execstr="$VSERVER $vsname exec su - $PGSQLUSER -c \"$PGSQLDUMP $db | > $backupdir/${db}.sql\""
109            fi
110         else
111            if [ "$compress" == "yes" ]; then
112               execstr="su - $PGSQLUSER -c \"$PGSQLDUMP $db | $GZIP > $backupdir/${db}.sql.gz\""
113            else
114               execstr="su - $PGSQLUSER -c \"$PGSQLDUMP $db > $backupdir/${db}.sql\""
115            fi
116         fi
117         debug "$execstr"
118         if [ ! $test ]; then
119             output=`eval $execstr 2>&1`
120             code=$?
121             if [ "$code" == "0" ]; then
122                 debug $output
123                 info "Successfully finished dump of pgsql database ${db}"
124             else
125                 warning $output
126                 warning "Failed to dump pgsql database ${db}"
127             fi
128         fi
129     done
130 fi
131
132 return 0
133