diff --git a/create/create.sh b/create/create.sh index 6abbf8c..26a8a95 100644 --- a/create/create.sh +++ b/create/create.sh @@ -1,5 +1,5 @@ #!/bin/bash -# create.sh - 启动Python容器执行create_table.py、create_src.py和create_src_dockerfile.py脚本 +# create.sh - 启动Python容器执行create_table.py、create_src.py、create_src_dockerfile.py和create_src_mod.py脚本 set -e # 遇到错误立即退出 @@ -100,9 +100,35 @@ docker run --rm \ echo '✅ create_src_dockerfile.py 执行完成!' " +# ========== 第四部分:更新go.mod ========== +echo "" +echo "📋 第四步:更新go.mod文件..." +echo "🚀 执行create_src_mod.py..." + +# 运行Python容器执行create_src_mod.py +docker run --rm \ + --name "${CONTAINER_NAME}-mod" \ + -v "${SCRIPT_DIR}:/app/create" \ + -v "${PROJECT_ROOT}/services:/app/services" \ + -e TABLE_NAME="${TABLE_NAME}" \ + -w /app \ + python:3.13.7-alpine3.22 \ + sh -c " + echo '📋 容器内环境信息:' + python --version + echo '' + + echo '⚙️ 执行 create_src_mod.py...' + python create/create_src_mod.py + + echo '' + echo '✅ create_src_mod.py 执行完成!' + " + echo "" echo "==========================================" echo "🎉 所有任务执行完成!" echo "📋 表名: ${TABLE_NAME}" echo "📁 服务目录: ${PROJECT_ROOT}/services/${TABLE_NAME}" -echo "🐳 Dockerfile: ${PROJECT_ROOT}/services/${TABLE_NAME}/src/Dockerfile" \ No newline at end of file +echo "🐳 Dockerfile: ${PROJECT_ROOT}/services/${TABLE_NAME}/src/Dockerfile" +echo "📦 go.mod: ${PROJECT_ROOT}/services/${TABLE_NAME}/src/go.mod" \ No newline at end of file diff --git a/create/create_src_mod.py b/create/create_src_mod.py index e69de29..3ea366f 100644 --- a/create/create_src_mod.py +++ b/create/create_src_mod.py @@ -0,0 +1,69 @@ +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- +""" +create_src_mod.py - 更新go.mod文件内容 +""" + +import os +import sys + +def update_go_mod(table_name): + """更新指定服务目录下的go.mod文件内容""" + + # 获取项目根目录 + project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) + services_dir = os.path.join(project_root, 'services') + + # 服务目录路径 + service_dir = os.path.join(services_dir, table_name) + go_mod_path = os.path.join(service_dir, 'src', 'go.mod') + + print(f"🚀 开始更新go.mod: {go_mod_path}") + + # go.mod内容模板,将xxxxx替换为table_name + go_mod_content = f"""module {table_name} + +go 1.25.0 + +require ( + +) + +require ( + +) +""" + + try: + # 检查文件是否存在 + if not os.path.exists(go_mod_path): + print(f"❌ go.mod文件不存在: {go_mod_path}") + return False + + # 清空文件内容并写入新内容 + with open(go_mod_path, 'w', encoding='utf-8') as f: + f.write(go_mod_content) + + print(f"✅ 成功更新go.mod: {go_mod_path}") + print(f"📋 模块名: {table_name}") + return True + + except Exception as e: + print(f"❌ 更新go.mod时出错: {e}") + return False + +if __name__ == "__main__": + # 从环境变量获取表名 + table_name = os.environ.get('TABLE_NAME') + + if not table_name: + print("❌ 错误: 未设置 TABLE_NAME 环境变量") + sys.exit(1) + + print(f"📋 接收到的表名: {table_name}") + success = update_go_mod(table_name) + + if success: + sys.exit(0) + else: + sys.exit(1) \ No newline at end of file