Initial commit: 智能项目定价模型

scripts/backup.sh

@@ -0,0 +1,178 @@
+#!/bin/bash
+# 智能项目定价模型 - 数据库备份脚本
+# 遵循瑞小美部署规范
+
+set -e
+
+# 配置
+BACKUP_DIR="/data/backups/pricing_model"
+RETENTION_DAYS=7
+DATE=$(date +%Y%m%d_%H%M%S)
+BACKUP_FILE="pricing_model_${DATE}.sql.gz"
+
+# 颜色输出
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+NC='\033[0m'
+
+log_info() {
+    echo -e "${GREEN}[INFO]${NC} $(date '+%Y-%m-%d %H:%M:%S') $1"
+}
+
+log_warn() {
+    echo -e "${YELLOW}[WARN]${NC} $(date '+%Y-%m-%d %H:%M:%S') $1"
+}
+
+log_error() {
+    echo -e "${RED}[ERROR]${NC} $(date '+%Y-%m-%d %H:%M:%S') $1"
+}
+
+# 加载环境变量
+load_env() {
+    cd "$(dirname "$0")/.."
+    
+    if [ -f ".env" ]; then
+        source .env
+    else
+        log_error ".env 文件不存在"
+        exit 1
+    fi
+}
+
+# 创建备份目录
+create_backup_dir() {
+    if [ ! -d "$BACKUP_DIR" ]; then
+        mkdir -p "$BACKUP_DIR"
+        chmod 700 "$BACKUP_DIR"
+    fi
+}
+
+# 执行备份
+do_backup() {
+    log_info "开始备份数据库..."
+    
+    # 使用 docker exec 执行 mysqldump
+    docker exec pricing-mysql mysqldump \
+        -u root \
+        -p"${MYSQL_ROOT_PASSWORD}" \
+        --single-transaction \
+        --routines \
+        --triggers \
+        --databases pricing_model \
+        2>/dev/null | gzip > "${BACKUP_DIR}/${BACKUP_FILE}"
+    
+    if [ $? -eq 0 ]; then
+        local size=$(du -h "${BACKUP_DIR}/${BACKUP_FILE}" | cut -f1)
+        log_info "备份完成: ${BACKUP_FILE} (${size})"
+    else
+        log_error "备份失败"
+        rm -f "${BACKUP_DIR}/${BACKUP_FILE}"
+        exit 1
+    fi
+}
+
+# 清理旧备份
+cleanup_old_backups() {
+    log_info "清理 ${RETENTION_DAYS} 天前的备份..."
+    
+    local deleted=0
+    while IFS= read -r file; do
+        rm -f "$file"
+        deleted=$((deleted + 1))
+    done < <(find "$BACKUP_DIR" -name "pricing_model_*.sql.gz" -mtime +${RETENTION_DAYS} -type f)
+    
+    if [ $deleted -gt 0 ]; then
+        log_info "已删除 ${deleted} 个旧备份"
+    fi
+}
+
+# 列出备份
+list_backups() {
+    log_info "备份列表:"
+    echo ""
+    ls -lh "${BACKUP_DIR}"/pricing_model_*.sql.gz 2>/dev/null || echo "暂无备份"
+    echo ""
+}
+
+# 恢复备份
+restore_backup() {
+    local backup_file="$1"
+    
+    if [ -z "$backup_file" ]; then
+        log_error "请指定备份文件"
+        list_backups
+        exit 1
+    fi
+    
+    if [ ! -f "$backup_file" ]; then
+        # 尝试在备份目录中查找
+        backup_file="${BACKUP_DIR}/${backup_file}"
+        if [ ! -f "$backup_file" ]; then
+            log_error "备份文件不存在: $backup_file"
+            exit 1
+        fi
+    fi
+    
+    log_warn "即将恢复数据库，当前数据将被覆盖!"
+    read -p "确认恢复? (yes/no): " confirm
+    
+    if [ "$confirm" != "yes" ]; then
+        log_info "取消恢复"
+        exit 0
+    fi
+    
+    log_info "开始恢复数据库..."
+    
+    gunzip -c "$backup_file" | docker exec -i pricing-mysql mysql \
+        -u root \
+        -p"${MYSQL_ROOT_PASSWORD}" \
+        2>/dev/null
+    
+    if [ $? -eq 0 ]; then
+        log_info "数据库恢复完成"
+    else
+        log_error "恢复失败"
+        exit 1
+    fi
+}
+
+# 主函数
+main() {
+    local action="${1:-backup}"
+    
+    load_env
+    create_backup_dir
+    
+    case $action in
+        backup)
+            do_backup
+            cleanup_old_backups
+            ;;
+        
+        restore)
+            restore_backup "$2"
+            ;;
+        
+        list)
+            list_backups
+            ;;
+        
+        cleanup)
+            cleanup_old_backups
+            ;;
+        
+        *)
+            echo "用法: $0 {backup|restore <file>|list|cleanup}"
+            echo ""
+            echo "命令:"
+            echo "  backup           执行备份"
+            echo "  restore <file>   恢复指定备份"
+            echo "  list             列出所有备份"
+            echo "  cleanup          清理旧备份"
+            exit 1
+            ;;
+    esac
+}
+
+main "$@"