匯出MySQL資料庫模式及資料的Bash指令碼

文章目錄

• Bash指令檔：export_db.sh

Bash指令檔：export_db.sh

#!/bin/sh

# export_db.sh
# 匯出資料庫模式及資料
# codingstandards@gmail.com
# v1: 2010-05-23
# v2: 2011-10-08/09

# 命令使用說明
# 從標準輸入讀取資料庫名稱和表名稱
# usage: export_db.sh
# usage: export_db.sh -
# 從指定檔案讀取資料庫名稱和表名稱
# usage: export_db.sh db_and_tables.txt
# 通過here string指定資料庫名稱和表名稱
# usage: export_db.sh <<<"dbname"
# usage: export_db.sh <<<"dbname tablename"

# 輸入格式說明
# 由一行或多行組成，可以是下列之一：
# (1) 空行：將被忽略
# (2) 注釋：以#開頭的行，將被忽略
# (3) db_name：只指定資料庫名字，該資料庫中的所有表都將被匯出
# (4) db_name table_name：指定資料庫名稱和表名稱，只匯出該表

# 資料庫連接參數
# TODO: 修改下面的參數以適應資料庫的設定
DBOPTS="-pxxx --default-character-set=gbk"怪蜀黍

# 儲存到目錄
# TODO: 可設定為本指令碼所在目錄，也可以設定在別的目錄
SAVETO=$(dirname "$0")

# 如果在crontab中執行，需加上下面這行，以保證環境變數正確設定
. /etc/profile

# 進入到儲存目錄
cd "$SAVETO" || exit 1

# 匯出目錄，格式為：dbYYYYmmddHHMMSS，比如 db20111008100923
DIR=db$(date +%Y%m%d%H%M%S)
mkdir -p "$DIR"

# 列印目前時間
now()
{
        date "+%F %T.%N"
}

# 執行SQL語句，輸出執行結果
# usage: db_exec <db> <sql>
db_exec()
{
        mysql $DBOPTS -s "$1" -e "$2"
}

# 匯出資料庫到檔案，分目錄儲存資料庫，分檔案儲存資料表
# usage: db_dump <db> <table>
db_dump()
{
        #echo -n "$1 $2   "
        #db_exec "$1" "select count(*) from $2"
        local count=$(db_exec "$1" "select count(*) from $2")
        printf "$(now)  %-32s%16s\n" "$1.$2" "$count"
        mkdir -p "$DIR/$1"
        mysqldump $DBOPTS --quick --add-drop-table "$1" "$2" >"$DIR/$1/$2.sql"
        echo "$(now)  ok."
}

# 從命令列的第一個參數指定的檔案讀取需要匯出的資料庫和表的名稱，然後匯出
#cat "${1:-export_db.txt}" |
#cat "$1" |
cat $1 |
while read db_name table_name;
do
        #echo "$db_name $table_name"
        if [ "$db_name" -a "${db_name:0:1}" != "#" ]; then      # 忽略空行和注釋
                if [ ! "$table_name" ]; then    # 如果沒有指定表的名稱，就認為是匯出整個資料庫
                        db_exec "$db_name" "show tables" |
                                while read table_name;
                                do
                                        db_dump "$db_name" "$table_name";
                                done
                else    # 匯出指定的資料庫中的表
                        db_dump "$db_name" "$table_name"
                fi
        fi
#done &> $DIR/export_db.log
done 2>&1 | tee $DIR/export_db.log

# 如果沒有任何錶匯出，就把$DIR目錄刪除
if [ ! "$(ls $DIR/*/*.sql 2>/dev/null)" ]; then
        echo "$(now)  no tables dumped!"
        rm -rf $DIR
        exit 2
fi

# 將匯出的檔案壓縮
tar zcf $DIR.tar.gz $DIR
echo "$(now)  $DIR.tar.gz generated." | tee -a $DIR/export_db.log

# END.