aboutsummaryrefslogtreecommitdiff
path: root/handlers/pgsql
blob: 8d8f0c78eecf035d21ec324d182708241a31489a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#
# PostgreSQL handler script for backupninja
#

getconf backupdir /var/backups/postgres
getconf databases all
getconf compress yes
getconf vsname

localhost=`hostname`

# If vservers are configured, decide if the handler should
# use them or if it should just operate on the host

if [ "$vservers" == "yes" ]
    then
    if [ ! -z $vsname ]
	then		
	info "Using vserver '$vsname'"
	usevserver=1
    else
	info "No vserver name specified, actions will be performed on the host"
    fi
fi

# Check to make sure that the specified vserver exists
if [ $usevserver ]
    then
    vroot="$VROOTDIR/$vsname"
    [ -d $vroot ] || fatal "vserver '$vsname' does not exist at '$vroot'"
fi

# create backup dir, the vroot variable will be empty if no vsname was specified
# and will proceed to operate on the host
[ -d $vroot$backupdir ] || mkdir -p $vroot$backupdir
[ -d $vroot$backupdir ] || fatal "Backup directory '$vroot$backupdir'"

# give backup dir the good uid and permissions
# (in respect to the vserver, if $usevserver)
pguid=`grep '^postgres:' $vroot/etc/passwd | awk -F: '{print $3}'`
debug "chown $pguid $vroot$backupdir"
chown $pguid $vroot$backupdir
debug "chmod 700 $vroot$backupdir"
chmod 700 $vroot$backupdir

# if $databases = all, use pg_dumpall
if [ "$databases" == "all" ]; then
    if [ $usevserver ]; then
	execstr="$VSERVER $vsname exec su - postgres -c $PGSQLDUMPALL > $vroot$backupdir/${vsname}.sql"
    else
	execstr="su - postgres -c $PGSQLDUMPALL > $backupdir/${localhost}-all.sql"
    fi
    debug "$execstr"
    if [ ! $test ]; then
	output=`$execstr 2>&1`
	code=$?
	if [ "$code" == "0" ]; then
	    debug $output
	    info "Successfully finished dump of pgsql cluster"
	else
	    warning $output
	    warning "Failed to dump pgsql cluster"
	fi
    fi
    
# else use pg_dump on each specified database
else
    for db in $databases; do
	if [ $usevserver ]
	    then
	    execstr="$VSERVER $vsname exec su - postgres -c $PGSQLDUMP $db > $vroot$backupdir/${db}.sql"
	else
	    execstr="su - postgres -c $PGSQLDUMP $db > $backupdir/${db}.sql"
	fi
	debug "$execstr"
	if [ ! $test ]; then
	    output=`$execstr 2>&1`
	    code=$?
	    if [ "$code" == "0" ]; then
		debug $output
		info "Successfully finished dump of pgsql database ${db}"
	    else
		warning $output
		warning "Failed to dump pgsql database ${db}"
	    fi
	fi
    done
fi

if [ "$compress" == "yes" ]; then
    output=`$GZIP -f $vroot$backupdir/*.sql 2>&1`
    debug $output
fi

return 0