feat: 代码标准化和文件重命名

## 新增功能 - 创建统一的公共函数库 (common/logging.sh, common/error_handler.sh) - 添加功能演示脚本 (common/demo_usage.sh) - 完善的使用文档 (common/README.md) ## 代码重构 - 重构所有脚本使用统一的公共库 - 为所有函数添加完整的文档注释 - 统一代码格式（4空格缩进、严格模式） - 标准化错误处理和日志输出 ## 文件重命名 - gcp/create_ai_project.sh → gcp/create_ai_projects.sh (单复数统一) - gcp/delete_all_project.sh → gcp/delete_all_projects.sh (单复数统一) - linux/install_ohmyzsh.sh → linux/install_oh_my_zsh.sh (专有名词规范) - linux/create_raid0_with_ext4.sh → linux/create_raid0_array.sh (简化命名) - common/example.sh → common/demo_usage.sh (更具描述性) ## 技术改进 - 使用 readonly 声明常量 - 启用 set -euo pipefail 严格模式 - 统一的 ANSI 颜色日志输出 - 完善的命令重试机制 - 栈追踪支持
2025-12-26 14:47:18 +08:00
commit 7def817482
11 changed files with 2312 additions and 0 deletions
--- a/gcp/create_ai_projects.sh
+++ b/gcp/create_ai_projects.sh
@@ -0,0 +1,514 @@
+#!/bin/bash
+# ============================================================================
+# 文件名: create_ai_projects.sh
+# 描述: 批量创建 Google Cloud Platform AI 项目并配置相关服务
+# 作者: Cloud Tools Project
+# 版本: 2.0.0
+# ============================================================================
+
+set -euo pipefail  # 启用严格模式
+
+# 获取脚本目录
+readonly SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+readonly PROJECT_ROOT="$(dirname "$SCRIPT_DIR")"
+
+# 加载公共库
+source "${PROJECT_ROOT}/common/logging.sh"
+source "${PROJECT_ROOT}/common/error_handler.sh"
+
+# 默认配置
+readonly DEFAULT_PROJECT_ID_PREFIX="project"
+readonly DEFAULT_START_NUM=1
+readonly DEFAULT_REPEAT_NUM=5
+readonly DEFAULT_MAX_RETRIES=3
+readonly DEFAULT_RETRY_DELAY=5
+
+# 创建日志文件
+readonly LOG_FILE="created_projects_$(date +%Y%m%d_%H%M%S).log"
+log_set_file "$LOG_FILE"
+
+#
+# 显示脚本使用方法
+#
+usage() {
+    log_info "用法: $0 [选项]"
+    log_info "选项:"
+    log_info "  -p, --prefix PREFIX      项目ID前缀 (默认: $DEFAULT_PROJECT_ID_PREFIX)"
+    log_info "  -s, --start START        项目编号起始值 (默认: $DEFAULT_START_NUM)"
+    log_info "  -n, --number NUMBER      创建项目的数量 (默认: $DEFAULT_REPEAT_NUM)"
+    log_info "  --max-retries RETRIES    单个命令失败时的最大重试次数 (默认: $DEFAULT_MAX_RETRIES)"
+    log_info "  --retry-delay DELAY      每次重试之间的延迟秒数 (默认: $DEFAULT_RETRY_DELAY)"
+    log_info "  --debug                  启用调试模式"
+    log_info "  -h, --help               显示此帮助信息"
+}
+
+#
+# 检查必要的命令是否存在
+#
+check_command() {
+    if ! command -v "$1" &> /dev/null; then
+        log_error "命令 $1 未找到。请确保它已安装并在PATH中。"
+        exit 1
+    fi
+}
+
+#
+# 重试执行命令
+#
+retry_command() {
+    local max_attempts=$1
+    local delay=$2
+    shift 2
+    local cmd_to_run=("$@")
+    local attempt=1
+    local exit_code
+    local temp_output
+    local temp_error
+
+    while [ $attempt -le "$max_attempts" ]; do
+        log_info "尝试执行 (第 $attempt/$max_attempts 次): ${cmd_to_run[*]}"
+
+        # 创建临时文件捕获输出和错误
+        temp_output=$(mktemp)
+        temp_error=$(mktemp)
+
+        if "${cmd_to_run[@]}" > "$temp_output" 2> "$temp_error"; then
+            log_success "命令成功执行: ${cmd_to_run[*]}"
+            # 如果有输出，显示它
+            if [ -s "$temp_output" ]; then
+                cat "$temp_output"
+            fi
+            rm -f "$temp_output" "$temp_error" 2>/dev/null || true
+            return 0
+        else
+            exit_code=$?
+            log_warning "命令执行失败 (第 $attempt/$max_attempts 次)，退出码: $exit_code"
+
+            # 显示错误信息用于调试
+            if [ -s "$temp_error" ]; then
+                log_error "错误输出: $(cat "$temp_error")"
+            fi
+
+            if [ $attempt -lt "$max_attempts" ]; then
+                log_info "将在 $delay 秒后重试..."
+                sleep "$delay"
+            fi
+        fi
+        rm -f "$temp_output" "$temp_error" 2>/dev/null || true
+        ((attempt++))
+    done
+
+    log_error "命令在 $max_attempts 次尝试后仍然失败: ${cmd_to_run[*]}"
+    return 1
+}
+
+#
+# 重试并获取命令输出
+#
+retry_command_with_output() {
+    local max_attempts=$1
+    local delay=$2
+    local output_var=$3
+    shift 3
+    local cmd_to_run=("$@")
+    local attempt=1
+    local exit_code
+    local temp_output
+    local temp_error
+
+    while [ $attempt -le "$max_attempts" ]; do
+        log_info "尝试执行 (第 $attempt/$max_attempts 次): ${cmd_to_run[*]}"
+
+        temp_output=$(mktemp)
+        temp_error=$(mktemp)
+
+        if "${cmd_to_run[@]}" > "$temp_output" 2> "$temp_error"; then
+            log_success "命令成功执行: ${cmd_to_run[*]}"
+            # 将输出赋值给指定变量
+            eval "$output_var=\$(cat '$temp_output')"
+            rm -f "$temp_output" "$temp_error" 2>/dev/null || true
+            return 0
+        else
+            exit_code=$?
+            log_warning "命令执行失败 (第 $attempt/$max_attempts 次)，退出码: $exit_code"
+
+            if [ -s "$temp_error" ]; then
+                log_error "错误输出: $(cat "$temp_error")"
+            fi
+
+            if [ $attempt -lt "$max_attempts" ]; then
+                log_info "将在 $delay 秒后重试..."
+                sleep "$delay"
+            fi
+        fi
+        rm -f "$temp_output" "$temp_error" 2>/dev/null || true
+        ((attempt++))
+    done
+
+    log_error "命令在 $max_attempts 次尝试后仍然失败: ${cmd_to_run[*]}"
+    return 1
+}
+
+#
+# 创建 GCP 项目
+#
+create_project() {
+    local project_id=$1
+    local max_retries=$2
+    local retry_delay=$3
+
+    log_info "创建项目: $project_id"
+
+    # 检查项目是否已存在
+    if gcloud projects describe "$project_id" &>/dev/null; then
+        log_warning "项目 $project_id 已存在，跳过创建"
+        return 0
+    fi
+
+    if retry_command "$max_retries" "$retry_delay" gcloud projects create "$project_id" --name="$project_id"; then
+        log_success "项目 $project_id 创建成功"
+        return 0
+    else
+        log_error "创建项目 $project_id 失败 (已重试)"
+        return 1
+    fi
+}
+
+#
+# 链接结算账号
+#
+link_billing() {
+    local project_id=$1
+    local billing_id=$2
+    local max_retries=$3
+    local retry_delay=$4
+
+    log_info "链接结算账号到项目: $project_id"
+    if retry_command "$max_retries" "$retry_delay" gcloud beta billing projects link "$project_id" --billing-account="$billing_id"; then
+        log_success "结算账号链接成功"
+        return 0
+    else
+        log_error "链接结算账号到项目 $project_id 失败 (已重试)"
+        return 1
+    fi
+}
+
+#
+# 启用服务
+#
+enable_services() {
+    local project_id=$1
+    local max_retries=$2
+    local retry_delay=$3
+
+    log_info "启用 aiplatform.googleapis.com 服务"
+    if retry_command "$max_retries" "$retry_delay" gcloud services enable aiplatform.googleapis.com --project="$project_id"; then
+        log_success "服务启用成功"
+        return 0
+    else
+        log_error "启用 aiplatform.googleapis.com 服务失败 (已重试)"
+        return 1
+    fi
+}
+
+#
+# 创建服务账号
+#
+create_service_account() {
+    local project_id=$1
+    local max_retries=$2
+    local retry_delay=$3
+
+    log_info "创建服务账号 service-account@$project_id.iam.gserviceaccount.com"
+
+    # 检查服务账号是否已存在
+    if gcloud iam service-accounts describe "service-account@$project_id.iam.gserviceaccount.com" --project="$project_id" &>/dev/null; then
+        log_warning "服务账号已存在，跳过创建"
+        return 0
+    fi
+
+    if retry_command "$max_retries" "$retry_delay" gcloud iam service-accounts create service-account \
+        --display-name="AI Platform Service Account" \
+        --project="$project_id"; then
+        log_success "服务账号创建成功"
+        return 0
+    else
+        log_error "创建服务账号失败 (已重试)"
+        return 1
+    fi
+}
+
+#
+# 添加 IAM 策略
+#
+add_iam_policy() {
+    local project_id=$1
+    local max_retries=$2
+    local retry_delay=$3
+
+    log_info "授予 aiplatform.serviceAgent 角色"
+    if retry_command "$max_retries" "$retry_delay" gcloud projects add-iam-policy-binding "$project_id" \
+        --member="serviceAccount:service-account@$project_id.iam.gserviceaccount.com" \
+        --role="roles/aiplatform.serviceAgent"; then
+        log_success "IAM 策略授予成功"
+        return 0
+    else
+        log_error "授予 IAM 策略失败 (已重试)"
+        return 1
+    fi
+}
+
+#
+# 创建服务账号密钥
+#
+create_service_account_key() {
+    local project_id=$1
+    local max_retries=$2
+    local retry_delay=$3
+
+    log_info "创建服务账号密钥"
+
+    # 检查密钥文件是否已存在
+    if [ -f "pass-$project_id.json" ]; then
+        log_warning "密钥文件 pass-$project_id.json 已存在，跳过创建"
+        return 0
+    fi
+
+    if retry_command "$max_retries" "$retry_delay" gcloud iam service-accounts keys create "pass-$project_id.json" \
+        --iam-account="service-account@$project_id.iam.gserviceaccount.com" \
+        --project="$project_id"; then
+        log_success "服务账号密钥创建成功: pass-$project_id.json"
+        return 0
+    else
+        log_error "创建服务账号密钥失败 (已重试)"
+        return 1
+    fi
+}
+
+# 参数解析和验证
+PROJECT_ID_PREFIX="$DEFAULT_PROJECT_ID_PREFIX"
+START_NUM="$DEFAULT_START_NUM"
+REPEAT_NUM="$DEFAULT_REPEAT_NUM"
+MAX_RETRIES="$DEFAULT_MAX_RETRIES"
+RETRY_DELAY="$DEFAULT_RETRY_DELAY"
+
+# 解析命令行参数
+while [[ $# -gt 0 ]]; do
+    case $1 in
+        -p|--prefix)
+            PROJECT_ID_PREFIX="$2"
+            shift 2
+            ;;
+        -s|--start)
+            START_NUM="$2"
+            shift 2
+            ;;
+        -n|--number)
+            REPEAT_NUM="$2"
+            shift 2
+            ;;
+        --max-retries)
+            MAX_RETRIES="$2"
+            shift 2
+            ;;
+        --retry-delay)
+            RETRY_DELAY="$2"
+            shift 2
+            ;;
+        --debug)
+            DEBUG_MODE=true
+            shift
+            ;;
+        -h|--help)
+            usage
+            exit 0
+            ;;
+        *)
+            log_error "未知选项: $1"
+            usage
+            exit 1
+            ;;
+    esac
+done
+
+# 检查必要的命令
+check_command gcloud
+
+# 验证参数
+if ! [[ "$START_NUM" =~ ^[0-9]+$ ]]; then
+    log_error "起始编号必须是一个正整数"
+    exit 1
+fi
+
+if ! [[ "$REPEAT_NUM" =~ ^[0-9]+$ ]] || [ "$REPEAT_NUM" -lt 1 ]; then
+    log_error "项目数量必须是一个正整数"
+    exit 1
+fi
+
+if ! [[ "$MAX_RETRIES" =~ ^[0-9]+$ ]] || [ "$MAX_RETRIES" -lt 0 ]; then
+    log_error "最大重试次数必须是一个非负整数"
+    exit 1
+fi
+
+if ! [[ "$RETRY_DELAY" =~ ^[0-9]+$ ]] || [ "$RETRY_DELAY" -lt 0 ]; then
+    log_error "重试延迟必须是一个非负整数"
+    exit 1
+fi
+
+log_info "################################################################################"
+log_info "开始创建 GCP 项目批量任务"
+log_info "################################################################################"
+
+# 调试模式提示
+if [ "$DEBUG_MODE" = true ]; then
+    log_debug "调试模式已启用"
+fi
+
+# 获取结算账号
+log_info "获取结算账号..."
+billing_id=""
+if retry_command_with_output "$MAX_RETRIES" "$RETRY_DELAY" billing_id_output gcloud beta billing accounts list --format="value(name.basename())" --filter="open=true"; then
+    # 提取第一个结算账号ID
+    billing_id=$(echo "$billing_id_output" | head -n 1 | tr -d '[:space:]')
+
+    if [ -z "$billing_id" ]; then
+        log_error "未找到有效的活动结算账号。请检查您的结算账户。"
+        exit 1
+    fi
+
+    log_success "结算账号: $billing_id"
+else
+    log_error "获取结算账号失败 (已重试)。请检查您的结算账户。"
+    exit 1
+fi
+
+# 计算结束编号
+END_NUM=$((START_NUM + REPEAT_NUM - 1))
+
+log_info "配置信息："
+log_info "  项目前缀: $PROJECT_ID_PREFIX"
+log_info "  起始编号: $START_NUM"
+log_info "  结束编号: $END_NUM"
+log_info "  项目总数: $REPEAT_NUM"
+log_info "  最大重试: $MAX_RETRIES 次"
+log_info "  重试延迟: $RETRY_DELAY 秒"
+
+# 显示将要创建的项目列表
+log_info "将创建以下项目："
+for i in $(seq "$START_NUM" "$END_NUM"); do
+    PROJECT_ID="${PROJECT_ID_PREFIX}$(printf "%02d" "$i")"
+    log_info "  - $PROJECT_ID"
+done
+
+# 确认开始创建
+log_warning "准备开始创建项目..."
+sleep 1
+
+# 统计变量
+successful_projects=0
+failed_projects=0
+
+# 创建项目和相关资源 - 主循环
+log_debug "开始主循环，从 $START_NUM 到 $END_NUM"
+
+for i in $(seq "$START_NUM" "$END_NUM"); do
+    PROJECT_ID="${PROJECT_ID_PREFIX}$(printf "%02d" "$i")"
+    current_num=$((i - START_NUM + 1))
+
+    log_info "-----------------------------------------------------"
+    log_info "开始处理项目 ($current_num/$REPEAT_NUM): $PROJECT_ID"
+    log_debug "当前循环变量 i=$i, 项目ID=$PROJECT_ID"
+    log_info "-----------------------------------------------------"
+
+    # 执行所有步骤
+    project_success=true
+
+    log_debug "步骤1: 创建项目"
+    if ! create_project "$PROJECT_ID" "$MAX_RETRIES" "$RETRY_DELAY"; then
+        log_debug "创建项目失败"
+        project_success=false
+    fi
+
+    if [ "$project_success" = true ]; then
+        log_debug "步骤2: 链接结算账号"
+        if ! link_billing "$PROJECT_ID" "$billing_id" "$MAX_RETRIES" "$RETRY_DELAY"; then
+            log_debug "链接结算账号失败"
+            project_success=false
+        fi
+    fi
+
+    if [ "$project_success" = true ]; then
+        log_debug "步骤3: 启用服务"
+        if ! enable_services "$PROJECT_ID" "$MAX_RETRIES" "$RETRY_DELAY"; then
+            log_debug "启用服务失败"
+            project_success=false
+        fi
+    fi
+
+    if [ "$project_success" = true ]; then
+        log_debug "步骤4: 创建服务账号"
+        if ! create_service_account "$PROJECT_ID" "$MAX_RETRIES" "$RETRY_DELAY"; then
+            log_debug "创建服务账号失败"
+            project_success=false
+        fi
+    fi
+
+    if [ "$project_success" = true ]; then
+        log_debug "步骤5: 添加IAM策略"
+        if ! add_iam_policy "$PROJECT_ID" "$MAX_RETRIES" "$RETRY_DELAY"; then
+            log_debug "添加IAM策略失败"
+            project_success=false
+        fi
+    fi
+
+    if [ "$project_success" = true ]; then
+        log_debug "步骤6: 创建服务账号密钥"
+        if ! create_service_account_key "$PROJECT_ID" "$MAX_RETRIES" "$RETRY_DELAY"; then
+            log_debug "创建服务账号密钥失败"
+            project_success=false
+        fi
+    fi
+
+    # 统计结果
+    if [ "$project_success" = true ]; then
+        log_success "项目 $PROJECT_ID 所有操作处理完成"
+        ((successful_projects++))
+    else
+        log_warning "项目 $PROJECT_ID 处理过程中出现错误，部分操作可能未完成。"
+        ((failed_projects++))
+    fi
+
+    # 添加延迟，避免 API 限制
+    log_debug "等待2秒后继续下一个项目..."
+    sleep 2
+
+    log_debug "完成项目 $PROJECT_ID 的处理，准备处理下一个项目"
+done
+
+log_debug "主循环结束"
+
+# 显示结果
+log_info "================== 执行完成 =================="
+log_success "成功创建: $successful_projects 个项目"
+log_error "创建失败: $failed_projects 个项目"
+log_info "详细日志: $LOG_FILE"
+echo "================== 创建完成于: $(date) ==================" >> "$LOG_FILE"
+
+# 显示成功创建的项目列表
+if [ "$successful_projects" -gt 0 ]; then
+    log_success "成功创建的项目："
+    for i in $(seq "$START_NUM" "$END_NUM"); do
+        PROJECT_ID="${PROJECT_ID_PREFIX}$(printf "%02d" "$i")"
+        if [ -f "pass-$PROJECT_ID.json" ]; then
+            log_info "  ✓ $PROJECT_ID (密钥文件: pass-$PROJECT_ID.json)"
+        fi
+    done
+fi
+
+if [ "$failed_projects" -gt 0 ]; then
+    log_error "有 $failed_projects 个项目在处理过程中遇到无法恢复的错误。"
+    log_info "请检查上面的日志以获取详细信息。"
+    exit 1
+fi
+
+log_success "所有项目创建/配置完成。"