first commit
This commit is contained in:
202
backend/golang.py
Normal file
202
backend/golang.py
Normal file
@@ -0,0 +1,202 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
Go 后端项目初始化模块
|
||||
"""
|
||||
|
||||
import subprocess
|
||||
import sys
|
||||
|
||||
|
||||
def run_command_with_progress(cmd, cwd, description):
|
||||
"""运行命令并显示实时输出"""
|
||||
try:
|
||||
print(f"\n{'=' * 60}")
|
||||
print(f"执行: {description}")
|
||||
print(f"{'=' * 60}")
|
||||
|
||||
process = subprocess.Popen(
|
||||
cmd,
|
||||
cwd=cwd,
|
||||
shell=True,
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.STDOUT,
|
||||
bufsize=0
|
||||
)
|
||||
|
||||
for line in process.stdout:
|
||||
try:
|
||||
decoded_line = line.decode('utf-8', errors='replace')
|
||||
except:
|
||||
try:
|
||||
decoded_line = line.decode('gbk', errors='replace')
|
||||
except:
|
||||
decoded_line = line.decode('latin-1', errors='replace')
|
||||
print(decoded_line, end='')
|
||||
|
||||
process.wait()
|
||||
|
||||
if process.returncode == 0:
|
||||
print(f"\n{'=' * 60}")
|
||||
print(f"✅ {description} - 完成")
|
||||
print(f"{'=' * 60}\n")
|
||||
return True
|
||||
else:
|
||||
print(f"\n{'=' * 60}")
|
||||
print(f"❌ {description} - 失败 (错误码: {process.returncode})")
|
||||
print(f"{'=' * 60}\n")
|
||||
return False
|
||||
|
||||
except Exception as e:
|
||||
print(f"\n❌ 执行失败: {str(e)}")
|
||||
return False
|
||||
|
||||
|
||||
def init_golang_project(backend_dir, project_name):
|
||||
"""初始化 Go 后端项目"""
|
||||
print("\n🚀 初始化 Go 后端项目...")
|
||||
|
||||
# 初始化 Go 模块
|
||||
success = run_command_with_progress(
|
||||
f"go mod init {project_name}-backend",
|
||||
backend_dir,
|
||||
"初始化 Go 模块"
|
||||
)
|
||||
|
||||
if not success:
|
||||
print("\n💡 提示: 请确保已安装 Go")
|
||||
print(" 下载地址: https://golang.org/dl/")
|
||||
sys.exit(1)
|
||||
|
||||
# 创建 main.go - 使用标准库
|
||||
main_go = backend_dir / "main.go"
|
||||
main_go.write_text(f'''package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"log"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
func main() {{
|
||||
mux := http.NewServeMux()
|
||||
|
||||
// 健康检查接口
|
||||
mux.HandleFunc("GET /api/health", healthHandler)
|
||||
|
||||
// 添加 CORS 中间件
|
||||
handler := corsMiddleware(mux)
|
||||
|
||||
log.Println("🚀 服务器启动: http://localhost:8080")
|
||||
log.Println("📍 健康检查: http://localhost:8080/api/health")
|
||||
log.Fatal(http.ListenAndServe(":8080", handler))
|
||||
}}
|
||||
|
||||
func healthHandler(w http.ResponseWriter, r *http.Request) {{
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(map[string]string{{
|
||||
"status": "ok",
|
||||
"message": "Welcome to {project_name} API",
|
||||
}})
|
||||
}}
|
||||
|
||||
func corsMiddleware(next http.Handler) http.Handler {{
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {{
|
||||
w.Header().Set("Access-Control-Allow-Origin", "*")
|
||||
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
|
||||
w.Header().Set("Access-Control-Allow-Headers", "Content-Type, Authorization")
|
||||
|
||||
if r.Method == "OPTIONS" {{
|
||||
w.WriteHeader(http.StatusNoContent)
|
||||
return
|
||||
}}
|
||||
|
||||
next.ServeHTTP(w, r)
|
||||
}})
|
||||
}}
|
||||
''', encoding='utf-8')
|
||||
|
||||
# 整理依赖
|
||||
run_command_with_progress("go mod tidy", backend_dir, "整理 Go 依赖")
|
||||
|
||||
print("\n✅ Go 后端项目初始化成功")
|
||||
print("💡 启动命令: go run main.go")
|
||||
print("📝 注意: 使用 Go 标准库，无需第三方依赖")
|
||||
|
||||
|
||||
def init_gin_project(backend_dir, project_name):
|
||||
"""初始化 Go Gin 后端项目"""
|
||||
print("\n🚀 初始化 Go Gin 后端项目...")
|
||||
|
||||
# 初始化 Go 模块
|
||||
success = run_command_with_progress(
|
||||
f"go mod init {project_name}-backend",
|
||||
backend_dir,
|
||||
"初始化 Go 模块"
|
||||
)
|
||||
|
||||
if not success:
|
||||
print("\n💡 提示: 请确保已安装 Go")
|
||||
print(" 下载地址: https://golang.org/dl/")
|
||||
sys.exit(1)
|
||||
|
||||
# 拉取 Gin 依赖
|
||||
deps_ok = run_command_with_progress(
|
||||
"go get github.com/gin-gonic/gin github.com/gin-contrib/cors",
|
||||
backend_dir,
|
||||
"安装 Gin 依赖"
|
||||
)
|
||||
|
||||
if not deps_ok:
|
||||
print("\n❌ 无法安装 Gin 依赖，请检查 Go 环境和网络")
|
||||
sys.exit(1)
|
||||
|
||||
# 创建 main.go - Gin 框架
|
||||
main_go = backend_dir / "main.go"
|
||||
main_go.write_text(f'''package main
|
||||
|
||||
import (
|
||||
"log"
|
||||
"time"
|
||||
|
||||
"github.com/gin-contrib/cors"
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func main() {{
|
||||
r := gin.New()
|
||||
r.Use(gin.Logger())
|
||||
r.Use(gin.Recovery())
|
||||
|
||||
// CORS 配置
|
||||
r.Use(cors.New(cors.Config{{
|
||||
AllowOrigins: []string{{"*"}},
|
||||
AllowMethods: []string{{"GET", "POST", "PUT", "DELETE", "OPTIONS"}},
|
||||
AllowHeaders: []string{{"Origin", "Content-Type", "Authorization"}},
|
||||
ExposeHeaders: []string{{"Content-Length"}},
|
||||
AllowCredentials: false,
|
||||
MaxAge: 12 * time.Hour,
|
||||
}}))
|
||||
|
||||
// 健康检查
|
||||
r.GET("/api/health", func(c *gin.Context) {{
|
||||
c.JSON(200, gin.H{{
|
||||
"status": "ok",
|
||||
"message": "Welcome to {project_name} API",
|
||||
}})
|
||||
}})
|
||||
|
||||
addr := ":8080"
|
||||
log.Printf("🚀 Gin 服务启动: http://localhost%s", addr)
|
||||
log.Printf("📍 健康检查: http://localhost%s/api/health", addr)
|
||||
|
||||
if err := r.Run(addr); err != nil {{
|
||||
log.Fatalf("启动失败: %v", err)
|
||||
}}
|
||||
}}
|
||||
''', encoding='utf-8')
|
||||
|
||||
# 整理依赖
|
||||
run_command_with_progress("go mod tidy", backend_dir, "整理 Go 依赖")
|
||||
|
||||
print("\n✅ Go Gin 后端项目初始化成功")
|
||||
print("💡 启动命令: go run main.go")
|
||||
Reference in New Issue
Block a user