Updates to handle vservers
[matthijs/upstream/backupninja.git] / handlers / mysql
index 4439bcc533173f84b9ca962ada281781d11f756e..b552ddebc346c7c5647e50d830c451e9d5a6f5dd 100644 (file)
@@ -11,29 +11,57 @@ getconf dbhost localhost
 getconf hotcopy no
 getconf sqldump no
 getconf user root
+getconf vsname
 
-# create backup dirs
+# 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
 
-[ -d $backupdir ] || mkdir -p $backupdir
-[ -d $backupdir ] || fatal "Backup directory '$backupdir'"
+# 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 dirs, 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'"
 hotdir="$backupdir/hotcopy"
 dumpdir="$backupdir/sqldump"
-[ "$sqldump" == "no" -o -d $dumpdir ] || mkdir -p $dumpdir
-[ "$hotcopy" == "no" -o -d $hotdir ] || mkdir -p $hotdir
+
+if [ $usevserver ]
+then
+       [ "$sqldump" == "no" -o -d $vroot$dumpdir ] || $VSERVER $vsname exec mkdir -p $dumpdir
+       [ "$hotcopy" == "no" -o -d $vroot$hotdir ] || $VSERVER $vsname exec mkdir -p $hotdir
+else
+       [ "$sqldump" == "no" -o -d $dumpdir ] || mkdir -p $dumpdir
+       [ "$hotcopy" == "no" -o -d $hotdir ] || mkdir -p $hotdir
+fi
 
 # create .my.cnf
- # (we do this because we don't want to have to specify the password on the command line
- # because then anyone would be able to see it with a 'ps aux'. instead, we create a 
- # temporary ~/.my.cnf in root's home directory).
+# (we do this because we don't want to have to specify the password on the command line
+# because then anyone would be able to see it with a 'ps aux'. instead, we create a 
+# temporary ~/.my.cnf in root's home directory).
 
 if [ "$dbusername" != "" ]; then
-       home=`grep '^root/etc/passwd | awk -F: '{print $6}'`
+       home=`grep '^root:' $vroot/etc/passwd | awk -F: '{print $6}'`
        [ -d $home ] || fatal "Can't find root's home directory ($home)."
-       mycnf="$home/.my.cnf"
+       mycnf="$vroot$home/.my.cnf"
        if [ -f $mycnf ]; then
                # rename temporarily
                tmpcnf="$home/my.cnf.disable"
-               debug "mv $mycnf $tmpcnf"
+               debug "mv $mycnf $tmpcnf"
                mv $mycnf $tmpcnf
        fi
        oldmask=`umask`
@@ -59,32 +87,42 @@ fi
 
 if [ "$hotcopy" == "yes" ]; then 
        if [ "$databases" == "all" ]; then
-               execstr="$MYSQLHOTCOPY --quiet --allowold --regexp /.\*/./.\*/ $hotdir"
-               debug 0 "su $user -c '$execstr'"
+               if [ $usevserver ]
+               then
+                       execstr="$VSERVER $vsname exec $MYSQLHOTCOPY --quiet --allowold --regexp /.\*/./.\*/ $hotdir"
+               else
+                       execstr="$MYSQLHOTCOPY --quiet --allowold --regexp /.\*/./.\*/ $hotdir"
+               fi
+               debug "su $user -c '$execstr'"
                if [ ! $test ]; then
                        output=`su $user -c "$execstr" 2>&1`
                        code=$?
                        if [ "$code" == "0" ]; then
-                               debug $output
-                               debug 1 "Successfully finished hotcopy of all mysql databases"
+                               debug $output
+                               info "Successfully finished hotcopy of all mysql databases"
                        else
-                               debug 2 $output
-                               debug 2 "Failed to hotcopy all mysql databases"
+                               warning $output
+                               warning "Failed to hotcopy all mysql databases"
                        fi
                fi
        else    
                for db in $databases; do
-                       execstr="$MYSQLHOTCOPY --allowold $db $hotdir"
-                       debug 0 "su $user -c '$execstr'"
+                       if [ $usevserver ]
+                       then
+                               execstr="$VSERVER $vsname exec $MYSQLHOTCOPY --allowold $db $hotdir"
+                       else
+                               execstr="$MYSQLHOTCOPY --allowold $db $hotdir"
+                       fi
+                       debug "su $user -c '$execstr'"
                        if [ ! $test ]; then
                                output=`su $user -c "$execstr" 2>&1`
                                code=$?
                                if [ "$code" == "0" ]; then
-                                       debug $output
-                                       debug 1 "Successfully finished hotcopy of mysql database $db"
+                                       debug $output
+                                       info "Successfully finished hotcopy of mysql database $db"
                                else
-                                       debug 2 $output
-                                       debug 2 "Failed to hotcopy mysql database $db"
+                                       warning $output
+                                       warning "Failed to hotcopy mysql database $db"
                                fi
                        fi
                done
@@ -95,37 +133,47 @@ fi
 
 if [ "$sqldump" == "yes" ]; then
        if [ "$databases" == "all" ]; then
-               databases=`echo 'show databases' | su $user -c "$MYSQL" | grep -v Database`
+               if [ $usevserver ]
+               then
+                       databases=`echo 'show databases' | $VSERVER $vsname exec su $user -c "$MYSQL" | grep -v Database`
+               else
+                       databases=`echo 'show databases' | su $user -c "$MYSQL" | grep -v Database`
+               fi
        fi
 
        for db in $databases; do
-               execstr="$MYSQLDUMP --lock-tables --complete-insert --add-drop-table --quick --quote-names $db > $dumpdir/${db}.sql"
-               debug 0 "su $user -c '$execstr'"
+               if [ $usevserver ]
+               then
+                       execstr="$VSERVER $vsname exec $MYSQLDUMP --lock-tables --complete-insert --add-drop-table --quick --quote-names $db > $vroot$dumpdir/${db}.sql"
+               else
+                       execstr="$MYSQLDUMP --lock-tables --complete-insert --add-drop-table --quick --quote-names $db > $dumpdir/${db}.sql"
+               fi
+               debug "su $user -c '$execstr'"
                if [ ! $test ]; then
                        output=`su $user -c "$execstr" 2>&1`
                        code=$?
                        if [ "$code" == "0" ]; then
-                               debug $output
-                               debug 1 "Successfully finished dump of mysql database $db"
+                               debug $output
+                               info "Successfully finished dump of mysql database $db"
                        else
-                               debug 2 $output
-                               debug 2 "Failed to dump mysql databases $db"
+                               warning $output
+                               warning "Failed to dump mysql databases $db"
                        fi
                fi
        done
        
        if [ "$compress" == "yes" ]; then
-               output=`$GZIP -f $dumpdir/*.sql 2>&1`
-               debug $output
+               output=`$GZIP -f $vroot$dumpdir/*.sql 2>&1`
+               debug $output
        fi
 fi
 
 if [ "$dbusername" != "" ]; then
        ## clean up tmp config file
-       debug "rm $mycnf"
+       debug "rm $mycnf"
        rm $mycnf
        if [ -f "$tmpcnf" ]; then
-               debug "mv $tmpcnf $mycnf"
+               debug "mv $tmpcnf $mycnf"
                mv $tmpcnf $mycnf
        fi
 fi