refactor: 分离数据库导出和导入脚本，优化文件结构
改进内容: 1. 创建独立的 import_database.sh 导入脚本 - 从 doc/database/backup/ 读取 SQL 文件 - 支持导入到 dev/test/prod 环境 - 自动验证导入结果 2. 简化 export_database.sh 导出脚本 - 只负责导出数据库到 backup 文件夹 - 移除导入功能，职责单一 - 添加后续操作提示 3. 优化文件结构 - backup 文件夹只保留 SQL 备份文件 - 配置文件和脚本统一放在根目录 - 移动操作指南到 doc/database/ 目录 4. 更新操作指南 - 详细说明两个脚本的用法 - 完整的迁移流程示例 - 常见问题解答 文件变更: - 新增: import_database.sh (独立导入脚本) - 修改: export_database.sh (简化为导出专用) - 移动: export_guide.md -> doc/database/数据库迁移操作指南.md - 删除: doc/database/backup/ 中的非 SQL 文件 使用方法: - 导出: ./export_database.sh - 导入: ./import_database.sh <dev|test|prod>
This commit is contained in:
329
import_database.sh
Normal file
329
import_database.sh
Normal file
@@ -0,0 +1,329 @@
|
||||
#!/bin/bash
|
||||
|
||||
# CCDI 数据库导入脚本
|
||||
# 功能：从 backup 文件夹导入数据库到目标环境
|
||||
|
||||
set -e # 遇到错误立即退出
|
||||
|
||||
# 颜色定义
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
# 日志函数
|
||||
log_info() {
|
||||
echo -e "${GREEN}[INFO]${NC} $1"
|
||||
}
|
||||
|
||||
log_warn() {
|
||||
echo -e "${YELLOW}[WARN]${NC} $1"
|
||||
}
|
||||
|
||||
log_error() {
|
||||
echo -e "${RED}[ERROR]${NC} $1"
|
||||
}
|
||||
|
||||
# 脚本目录
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
CONFIG_FILE="${SCRIPT_DIR}/db_config.conf"
|
||||
BACKUP_DIR="${SCRIPT_DIR}/doc/database/backup"
|
||||
STRUCTURE_FILE="ccdi_structure.sql"
|
||||
DATA_FILE="ccdi_data.sql"
|
||||
|
||||
# 检查配置文件
|
||||
check_config() {
|
||||
if [ ! -f "$CONFIG_FILE" ]; then
|
||||
log_error "配置文件不存在: $CONFIG_FILE"
|
||||
log_info "请先复制配置模板: cp db_config.conf.template db_config.conf"
|
||||
log_info "然后编辑 db_config.conf 填写目标数据库连接信息"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# 加载配置文件
|
||||
source "$CONFIG_FILE"
|
||||
log_info "配置文件加载成功"
|
||||
}
|
||||
|
||||
# 检查 mysql 命令
|
||||
check_mysql() {
|
||||
if ! command -v mysql &> /dev/null; then
|
||||
log_error "mysql 命令未找到"
|
||||
log_info "请安装 MySQL 客户端工具"
|
||||
exit 1
|
||||
fi
|
||||
log_info "mysql 命令检查通过"
|
||||
}
|
||||
|
||||
# 检查备份文件
|
||||
check_backup_files() {
|
||||
log_info "检查备份文件..."
|
||||
|
||||
local structure_file="${BACKUP_DIR}/${STRUCTURE_FILE}"
|
||||
local data_file="${BACKUP_DIR}/${DATA_FILE}"
|
||||
|
||||
if [ ! -f "$structure_file" ]; then
|
||||
log_error "表结构文件不存在: $structure_file"
|
||||
log_info "请先执行导出: ./export_database.sh export"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [ ! -f "$data_file" ]; then
|
||||
log_error "数据文件不存在: $data_file"
|
||||
log_info "请先执行导出: ./export_database.sh export"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log_info "备份文件检查通过"
|
||||
log_info "表结构文件: $structure_file ($(du -h "$structure_file" | cut -f1))"
|
||||
log_info "数据文件: $data_file ($(du -h "$data_file" | cut -f1))"
|
||||
}
|
||||
|
||||
# 导入表结构
|
||||
import_structure() {
|
||||
local env_type=$1
|
||||
local db_host db_port db_user db_pass db_name
|
||||
|
||||
case "$env_type" in
|
||||
production|prod)
|
||||
db_host="$PROD_DB_HOST"
|
||||
db_port="$PROD_DB_PORT"
|
||||
db_user="$PROD_DB_USER"
|
||||
db_pass="$PROD_DB_PASS"
|
||||
db_name="$PROD_DB_NAME"
|
||||
;;
|
||||
test)
|
||||
db_host="$TEST_DB_HOST"
|
||||
db_port="$TEST_DB_PORT"
|
||||
db_user="$TEST_DB_USER"
|
||||
db_pass="$TEST_DB_PASS"
|
||||
db_name="$TEST_DB_NAME"
|
||||
;;
|
||||
dev|development)
|
||||
db_host="$SOURCE_DB_HOST"
|
||||
db_port="$SOURCE_DB_PORT"
|
||||
db_user="$SOURCE_DB_USER"
|
||||
db_pass="$SOURCE_DB_PASS"
|
||||
db_name="$SOURCE_DB_NAME"
|
||||
;;
|
||||
*)
|
||||
log_error "未知的环境类型: $env_type"
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
|
||||
log_info "导入表结构到 ${env_type} 环境: ${db_host}:${db_port}/${db_name}"
|
||||
|
||||
local structure_file="${BACKUP_DIR}/${STRUCTURE_FILE}"
|
||||
|
||||
# 导入表结构
|
||||
mysql -h "$db_host" \
|
||||
-P "$db_port" \
|
||||
-u "$db_user" \
|
||||
-p"$db_pass" \
|
||||
--default-character-set=utf8mb4 \
|
||||
"$db_name" < "$structure_file" 2>/dev/null
|
||||
|
||||
if [ $? -eq 0 ]; then
|
||||
log_info "表结构导入成功"
|
||||
else
|
||||
log_error "表结构导入失败"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# 导入数据
|
||||
import_data() {
|
||||
local env_type=$1
|
||||
local db_host db_port db_user db_pass db_name
|
||||
|
||||
case "$env_type" in
|
||||
production|prod)
|
||||
db_host="$PROD_DB_HOST"
|
||||
db_port="$PROD_DB_PORT"
|
||||
db_user="$PROD_DB_USER"
|
||||
db_pass="$PROD_DB_PASS"
|
||||
db_name="$PROD_DB_NAME"
|
||||
;;
|
||||
test)
|
||||
db_host="$TEST_DB_HOST"
|
||||
db_port="$TEST_DB_PORT"
|
||||
db_user="$TEST_DB_USER"
|
||||
db_pass="$TEST_DB_PASS"
|
||||
db_name="$TEST_DB_NAME"
|
||||
;;
|
||||
dev|development)
|
||||
db_host="$SOURCE_DB_HOST"
|
||||
db_port="$SOURCE_DB_PORT"
|
||||
db_user="$SOURCE_DB_USER"
|
||||
db_pass="$SOURCE_DB_PASS"
|
||||
db_name="$SOURCE_DB_NAME"
|
||||
;;
|
||||
*)
|
||||
log_error "未知的环境类型: $env_type"
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
|
||||
log_info "导入数据到 ${env_type} 环境: ${db_host}:${db_port}/${db_name}"
|
||||
|
||||
local data_file="${BACKUP_DIR}/${DATA_FILE}"
|
||||
|
||||
# 导入数据
|
||||
mysql -h "$db_host" \
|
||||
-P "$db_port" \
|
||||
-u "$db_user" \
|
||||
-p"$db_pass" \
|
||||
--default-character-set=utf8mb4 \
|
||||
"$db_name" < "$data_file" 2>/dev/null
|
||||
|
||||
if [ $? -eq 0 ]; then
|
||||
log_info "数据导入成功"
|
||||
else
|
||||
log_error "数据导入失败"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# 验证导入结果
|
||||
verify_import() {
|
||||
local env_type=$1
|
||||
local db_host db_port db_user db_pass db_name
|
||||
|
||||
case "$env_type" in
|
||||
production|prod)
|
||||
db_host="$PROD_DB_HOST"
|
||||
db_port="$PROD_DB_PORT"
|
||||
db_user="$PROD_DB_USER"
|
||||
db_pass="$PROD_DB_PASS"
|
||||
db_name="$PROD_DB_NAME"
|
||||
;;
|
||||
test)
|
||||
db_host="$TEST_DB_HOST"
|
||||
db_port="$TEST_DB_PORT"
|
||||
db_user="$TEST_DB_USER"
|
||||
db_pass="$TEST_DB_PASS"
|
||||
db_name="$TEST_DB_NAME"
|
||||
;;
|
||||
dev|development)
|
||||
db_host="$SOURCE_DB_HOST"
|
||||
db_port="$SOURCE_DB_PORT"
|
||||
db_user="$SOURCE_DB_USER"
|
||||
db_pass="$SOURCE_DB_PASS"
|
||||
db_name="$SOURCE_DB_NAME"
|
||||
;;
|
||||
*)
|
||||
log_error "未知的环境类型: $env_type"
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
|
||||
log_info "验证导入结果..."
|
||||
|
||||
# 查询表数量
|
||||
local table_count=$(mysql -h "$db_host" \
|
||||
-P "$db_port" \
|
||||
-u "$db_user" \
|
||||
-p"$db_pass" \
|
||||
--default-character-set=utf8mb4 \
|
||||
-N -e "SELECT COUNT(*) FROM information_schema.tables WHERE table_schema='$db_name';" "$db_name" 2>/dev/null)
|
||||
|
||||
log_info "目标数据库表数量: $table_count"
|
||||
|
||||
# 查询关键表行数（示例：sys_user 表）
|
||||
local user_count=$(mysql -h "$db_host" \
|
||||
-P "$db_port" \
|
||||
-u "$db_user" \
|
||||
-p"$db_pass" \
|
||||
--default-character-set=utf8mb4 \
|
||||
-N -e "SELECT COUNT(*) FROM sys_user;" "$db_name" 2>/dev/null)
|
||||
|
||||
log_info "sys_user 表数据行数: $user_count"
|
||||
|
||||
# 检查数据库字符集
|
||||
local db_charset=$(mysql -h "$db_host" \
|
||||
-P "$db_port" \
|
||||
-u "$db_user" \
|
||||
-p"$db_pass" \
|
||||
--default-character-set=utf8mb4 \
|
||||
-N -e "SELECT DEFAULT_CHARACTER_SET_NAME FROM information_schema.schemata WHERE schema_name='$db_name';" 2>/dev/null)
|
||||
|
||||
log_info "数据库字符集: $db_charset"
|
||||
}
|
||||
|
||||
# 导入数据库
|
||||
import_database() {
|
||||
local env_type=$1
|
||||
|
||||
if [ -z "$env_type" ]; then
|
||||
log_error "请指定目标环境: dev, test 或 production"
|
||||
log_info "用法: $0 <env>"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log_info "========== 开始导入数据库到 ${env_type} 环境 =========="
|
||||
|
||||
check_config
|
||||
check_mysql
|
||||
check_backup_files
|
||||
import_structure "$env_type"
|
||||
import_data "$env_type"
|
||||
verify_import "$env_type"
|
||||
|
||||
log_info "========== 数据库导入完成 =========="
|
||||
}
|
||||
|
||||
# 使用帮助
|
||||
show_usage() {
|
||||
echo "用法: $0 <environment>"
|
||||
echo ""
|
||||
echo "环境:"
|
||||
echo " dev, development 开发环境（源数据库）"
|
||||
echo " test 测试环境"
|
||||
echo " production, prod 生产环境"
|
||||
echo ""
|
||||
echo "说明:"
|
||||
echo " 此脚本从 doc/database/backup/ 文件夹读取备份文件并导入到指定环境"
|
||||
echo " 备份文件:"
|
||||
echo " - ccdi_structure.sql (表结构)"
|
||||
echo " - ccdi_data.sql (数据)"
|
||||
echo ""
|
||||
echo "前置条件:"
|
||||
echo " 1. 已执行 ./export_database.sh export 导出数据库"
|
||||
echo " 2. 已配置 db_config.conf 文件"
|
||||
echo ""
|
||||
echo "示例:"
|
||||
echo " $0 test # 导入到测试环境"
|
||||
echo " $0 prod # 导入到生产环境"
|
||||
echo " $0 dev # 导入到开发环境"
|
||||
}
|
||||
|
||||
# 主函数
|
||||
main() {
|
||||
case "$1" in
|
||||
dev|development)
|
||||
import_database "dev"
|
||||
;;
|
||||
test)
|
||||
import_database "test"
|
||||
;;
|
||||
production|prod)
|
||||
import_database "production"
|
||||
;;
|
||||
help|--help|-h)
|
||||
show_usage
|
||||
;;
|
||||
*)
|
||||
if [ -z "$1" ]; then
|
||||
log_error "缺少环境参数"
|
||||
else
|
||||
log_error "未知的环境: $1"
|
||||
fi
|
||||
show_usage
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
# 执行主函数
|
||||
main "$@"
|
||||
Reference in New Issue
Block a user