mysql数据恢复脚本

#!/bin/bash
##################################################################################################################
##   Create_time     : 2023/6/30                                                                                ##
##   Author          : Ops-DaBao                                                                                ##
##   Mail            :                                                                    ##
##   Feature         : restore mysql Data                                                                       ##
##   Usage           : When executing the script, you can pass parameters.                                      ##
##                     If you do not pass parameters, the default is used                                       ##
##################################################################################################################
# shellcheck disable=SC2034
red_color="\033[31m"
green_color="\033[32m"
text_end="\033[0m"
start_time=$(date +%s)
backup_file=""
username="root"
password="root"
database_name=""
table_name=""
tip=""


restore_single_table_data() {
	# 还原单表数据
	echo -e "${red_color}-----------本次进行单表数据恢复-----------${text_end}"
	sed  -n  "/CREATE TABLE \`$table_name\` /,/\;/p" $backup_file  > $table_name.sql
	sed -n "/INSERT INTO \`$table_name\`/p" $backup_file >> $table_name.sql
	docker cp $table_name.sql MySQL8:/$table_name.sql
	docker exec MySQL8 /bin/bash -c "mysql -u $username -p$password -e 'use $database_name;' -e 'source /$table_name.sql;'"
	echo -e "${green_color}-----------恢复完毕！-----------${text_end}"
	echo -e "${green_color}-----本次是使用${backup_file}进行数据恢复-----------${text_end}"

}

restore_the_entire_library_data() {
	# 如果数据库不存在则先创建出来，在继续回复数据
	echo -e "${red_color}-----------本次进行单库据恢复-----------${text_end}"
	docker exec MySQL8 /bin/bash -c "mysql -u $username -p$password -e 'CREATE DATABASE IF NOT EXISTS $database_name'"
	# 还原单库数据
	docker cp $backup_file MySQL8:/backup_file.sql
	docker exec MySQL8 /bin/bash -c "mysql -u $username -p$password -e 'use $database_name;' -e 'source /backup_file.sql;'"
	echo -e "${green_color}-----------恢复完毕！-----------${text_end}"
	echo -e "${green_color}-----本次是使用${backup_file}进行数据恢复-----------${text_end}"
}


Restore_a_database_in_full() {
	# 从全量数据中，还原一个数据库
	echo -e "${red_color}----------本次进行全量中单库据恢复-----------${text_end}"
	docker exec MySQL8 /bin/bash -c "mysql -u $username -p$password  --one-database $database_name" < $backup_file
	echo -e "${green_color}----------恢复完毕！-----------${text_end}"
	echo -e "${green_color}-----本次是使用${backup_file}进行数据恢复-----------${text_end}"
}


usage() {
  echo "脚本使用方法:    bash $0 <选项> <参数>"
  echo "选项说明:"
  echo " -f <文件路径>:  数据保存路径,绝对路径 （默认为空）"
  echo " -u <用户名>:   指定用户名 (默认为空)"
  echo " -p <密码>:     指定密码 (默认为空)"
  echo " -d <数据库名>:  指定数据库名 (默认为空)"
  echo " -b <数据表>:  指定数据库表 (默认为空)"
  echo " -t <恢复种类>:  指定恢复 数据 table(集合) 或 singlelibrary(数据库)  或 fulllibrary(全库) (默认为空)"
  exit 1
}


# 检查是否有参数传递，如果有则覆盖默认值
while getopts ":f:u:p:d:b:t:*:" opt; do
  case $opt in
    f) backup_file="$OPTARG";;
    u) username="$OPTARG";;
    p) password="$OPTARG";;
    d) database_name="$OPTARG";;
    b) table_name="$OPTARG";;
    t) tip="$OPTARG";;
    *) echo "参数传入有误: -$OPTARG"; usage;;
  esac


done
# 检查备份文件是否存在
if [ ! -f "$backup_file" ]; then
  echo "Backup file not found: $backup_file"
  exit 1
fi


case $tip in
    table)
      echo -e "${green_color}Mysql database_name: ${database_name}${text_end}"
      echo -e "${green_color}Mysql database_name: ${collection_name}${text_end}"
      restore_single_table_data;
      ;;
    singlelibrary)
      echo -e "${green_color}Mysql database_name: ${database_name}${text_end}"
      restore_the_entire_library_data;
      ;;
    fulllibrary)
      echo -e "${green_color}Mysql database_name: ${database_name}${text_end}"
      Restore_a_database_in_full;
      ;;
    *)
      echo -e "${red_color}暂时无法确认你要恢复那些数据${text_end}"
esac


end_time=$(date +%s)
execution_time=$((end_time - start_time))

echo -e "${green_color}本次执行脚本耗时为：$execution_time 秒${text_end}"