#!/bin/sh
#
# 使用方法：
# ./restore.sh /你备份文件的全路径
#ocpyang@126.com

INNOBACKUPEX=innobackupex
INNOBACKUPEX_PATH=/usr/bin/$INNOBACKUPEX
TMP_LOG="/var/log/restore.$$.log"
MY_CNF=/usr/local/mysql/my.cnf
BACKUP_DIR=/backup # 你的备份主目录
FULLBACKUP_DIR=$BACKUP_DIR/full # 全库备份的目录
INCRBACKUP_DIR=$BACKUP_DIR/incre # 增量备份的目录
MEMORY=4096M # 还原的时候使用的内存限制数

#############################################################################

#显示错误

#############################################################################

error()
{
echo "$1" 1>&2
exit 1
}

#############################################################################

# 检查innobackupex错误输出

#############################################################################

check_innobackupex_fail()
{
if [ -z "`tail -2 $TMP_LOG | grep 'completed OK!'`" ] ; then
echo "$INNOBACKUPEX命令执行失败失败:"; echo
echo "---------- $INNOBACKUPEX的错误输出 ----------"
cat $TMP_LOG
#保留一份备份的详细日志
logfiledate=restore.`date +%Y%m%d%H%M`.txt
cat $TMP_LOG>/backup/$logfiledate
rm -f $TMP_LOG
exit 1
fi
}

# 选项检测
if [ ! -x $INNOBACKUPEX_PATH ]; then
error "$INNOBACKUPEX_PATH在指定路径不存在,请确认是否安装或核实链接是否正确."
fi

if [ ! -d $BACKUP_DIR ]; then
error "备份目录$BACKUP_DIR不存在."
fi

if [ $# != 1 ] ; then
error "使用方法: $0 使用还原目录的绝对路径"
fi

if [ ! -d $1 ]; then
error "还原到:$1不存在."
fi

# Some info output
echo "----------------------------"
echo
echo "$0: MySQL还原脚本"
START_RESTORE_TIME=`date +%F' '%T' '%w`
echo "数据库还原开始于: $START_RESTORE_TIME"
echo

PARENT_DIR=`dirname $1`
if [ $PARENT_DIR = $FULLBACKUP_DIR ]; then
FULLBACKUP=$1
echo "还原`basename $FULLBACKUP`"
echo

else
if [ `dirname $PARENT_DIR` = $INCRBACKUP_DIR ]; then
INCR=`basename $1`
FULL=`basename $PARENT_DIR`
FULLBACKUP=$FULLBACKUP_DIR/$FULL

if [ ! -d $FULLBACKUP ]; then
error "全备:$FULLBACKUP不存在."
fi

echo "还原$FULL到增量$INCR"
echo

echo "Prepare完整备份集..........."
echo "*****************************"
$INNOBACKUPEX_PATH --defaults-file=$MY_CNF --apply-log --redo-only --use-memory=$MEMORY $FULLBACKUP > $TMP_LOG 2>&1
check_innobackupex_fail

# Prepare增量备份集
for i in `find $PARENT_DIR -mindepth 1 -maxdepth 1 -type d -printf "%P\n" | sort -n`; do

echo "Prepare增量备份集$i........"
echo "*****************************"
$INNOBACKUPEX_PATH --defaults-file=$MY_CNF --apply-log --redo-only --use-memory=$MEMORY $FULLBACKUP --incremental-dir=$PARENT_DIR/$i > $TMP_LOG 2>&1
check_innobackupex_fail

if [ $INCR = $i ]; then
break
fi
done
else
error "未知的备份类型"
fi
fi

echo "prepare全备集，回滚那些未提交的事务..........."
$INNOBACKUPEX_PATH --defaults-file=$MY_CNF --apply-log --use-memory=$MEMORY $FULLBACKUP > $TMP_LOG 2>&1
check_innobackupex_fail
echo "*****************************"
echo "1.数据库还原中 ...请稍等"
echo "*****************************"

$INNOBACKUPEX_PATH --defaults-file=$MY_CNF --copy-back $FULLBACKUP > $TMP_LOG 2>&1
check_innobackupex_fail

rm -f $TMP_LOG
echo "2.恭喜,还原成功!."
echo "*****************************"

#修改目录权限
echo "修改mysql目录的权限."
mysqlcnf="/usr/local/mysql/my.cnf"
mysqldatadir=`grep -i "^basedir=" $mysqlcnf`
`echo 'chown -R mysql:mysql' ${mysqldatadir:8}`
echo "3.权限修改成功!"
echo "*****************************"

#自动启动mysql
echo -n "请输入确认启动mysql[1]:"
read var
echo "你输入的是$var"
errorlog=`grep -i "^log-error" $MY_CNF`
if [ $var = 1 ] ; then
echo "mysql启动中............"
`/usr/local/mysql/bin/mysqld_safe > /dev/null & `
sleep 10
PortNum=`netstat -lnt|grep 3306|wc -l`
if [ $PortNum = 1 ];
then
echo "4.mysql启动成功"
echo "*****************************"
else
echo "4.mysql启动失败,请检查错误日志$errorlog"
echo "*****************************************"
fi
exit 0
else
echo "输入不正确!"
fi

END_RESTORE_TIME=`date +%F' '%T' '%w`
echo "数据库还原完成于: $END_RESTORE_TIME"
exit 0