diff --git a/backend/src/logic4country/create.go b/backend/src/logic4country/create.go index 3af378c..a9872ca 100644 --- a/backend/src/logic4country/create.go +++ b/backend/src/logic4country/create.go @@ -1,48 +1,43 @@ package logic4country import ( - "asset_assistant/db" // 数据库操作相关包 + "asset_assistant/db" "net/http" - "time" // 时间处理包 + "time" - "github.com/gin-gonic/gin" // Gin框架,用于处理HTTP请求 - "github.com/google/uuid" // UUID生成工具 - "go.uber.org/zap" // 日志库 + "github.com/gin-gonic/gin" + "github.com/google/uuid" + "go.uber.org/zap" ) // CreateRequest 注册请求参数结构 -// 用于接收客户端发送的JSON数据,绑定并验证必填字段 type CreateRequest struct { Name string `json:"name" binding:"required"` // 国家名称,必填 Code string `json:"code" binding:"required"` // 国家代码,必填 + Flag string `json:"flag"` // 国旗信息,可选 } // CreateResponse 注册响应结构 -// 统一的API响应格式,包含成功状态、提示信息和数据 type CreateResponse struct { - Success bool `json:"success"` // 操作是否成功 - Message string `json:"message"` // 提示信息 - Data CreateData `json:"data"` // 响应数据 + Success bool `json:"success"` + Message string `json:"message"` + Data CreateData `json:"data"` } // CreateData 响应数据结构 -// 包含创建成功后的国家ID type CreateData struct { - CountryID string `json:"country_id"` // 国家唯一标识ID + CountryID string `json:"country_id"` } // CreateHandler 处理国家创建逻辑 -// 接收HTTP请求,完成参数验证、数据库事务处理并返回响应 func CreateHandler(c *gin.Context) { - startTime := time.Now() // 记录请求开始时间,用于统计耗时 - // 获取或生成请求ID,用于追踪整个请求链路 + startTime := time.Now() reqID := c.Request.Header.Get("X-RegisterRequest-ID") if reqID == "" { reqID = uuid.New().String() zap.L().Debug("✨ 生成新的请求ID", zap.String("req_id", reqID)) } - // 记录请求接收日志,包含关键追踪信息 zap.L().Info("📥 收到国家创建请求", zap.String("req_id", reqID), zap.String("path", c.Request.URL.Path), @@ -50,14 +45,12 @@ func CreateHandler(c *gin.Context) { ) var req CreateRequest - // 绑定并验证请求参数(检查name和code是否存在) if err := c.ShouldBindJSON(&req); err != nil { zap.L().Warn("⚠️ 请求参数验证失败", zap.String("req_id", reqID), zap.Error(err), zap.Any("request_body", c.Request.Body), ) - // 返回参数错误响应 c.JSON(http.StatusBadRequest, CreateResponse{ Success: false, Message: "请求参数错误:name和code为必填项", @@ -65,14 +58,14 @@ func CreateHandler(c *gin.Context) { return } - // 记录通过验证的请求参数 zap.L().Debug("✅ 请求参数验证通过", zap.String("req_id", reqID), zap.String("name", req.Name), zap.String("code", req.Code), + zap.String("flag", req.Flag), // 新增国旗参数日志 ) - // 开启数据库事务,确保多表操作原子性(要么全成功,要么全失败) + // 开启数据库事务 tx, err := db.DB.Begin() if err != nil { zap.L().Error("❌ 事务开启失败", @@ -85,10 +78,9 @@ func CreateHandler(c *gin.Context) { }) return } - // 延迟执行的恢复函数,处理panic情况 + defer func() { - if r := recover(); r != nil { // 捕获panic - // 回滚事务 + if r := recover(); r != nil { if err := tx.Rollback(); err != nil { zap.L().Error("💥 panic后事务回滚失败", zap.String("req_id", reqID), @@ -99,7 +91,6 @@ func CreateHandler(c *gin.Context) { zap.String("req_id", reqID), zap.Any("recover", r), ) - // 返回系统错误响应 c.JSON(http.StatusInternalServerError, CreateResponse{ Success: false, Message: "系统错误,请稍后重试", @@ -107,11 +98,11 @@ func CreateHandler(c *gin.Context) { } }() - // 1. 在country表中创建记录并获取自动生成的ID + // 1. 创建country主表记录 var countryID string err = tx.QueryRow("INSERT INTO country DEFAULT VALUES RETURNING id").Scan(&countryID) if err != nil { - tx.Rollback() // 操作失败,回滚事务 + tx.Rollback() zap.L().Error("❌ country表插入失败", zap.String("req_id", reqID), zap.Error(err), @@ -128,10 +119,10 @@ func CreateHandler(c *gin.Context) { zap.String("country_id", countryID), ) - // 2. 插入国家名称到name表(与country_id关联) + // 2. 插入国家名称 _, err = tx.Exec("INSERT INTO country_name (country_id, name) VALUES ($1, $2)", countryID, req.Name) if err != nil { - tx.Rollback() // 操作失败,回滚事务 + tx.Rollback() zap.L().Error("❌ country_name表插入失败", zap.String("req_id", reqID), zap.String("country_id", countryID), @@ -144,10 +135,10 @@ func CreateHandler(c *gin.Context) { return } - // 3. 插入国家代码到code表(与country_id关联) + // 3. 插入国家代码 _, err = tx.Exec("INSERT INTO country_code (country_id, code) VALUES ($1, $2)", countryID, req.Code) if err != nil { - tx.Rollback() // 操作失败,回滚事务 + tx.Rollback() zap.L().Error("❌ country_code表插入失败", zap.String("req_id", reqID), zap.String("country_id", countryID), @@ -160,9 +151,31 @@ func CreateHandler(c *gin.Context) { return } - // 提交事务(所有操作成功后确认提交) + // 4. 新增:插入国旗信息(如果提供) + if req.Flag != "" { + _, err = tx.Exec("INSERT INTO country_flag (country_id, flag) VALUES ($1, $2)", countryID, req.Flag) + if err != nil { + tx.Rollback() + zap.L().Error("❌ country_flag表插入失败", + zap.String("req_id", reqID), + zap.String("country_id", countryID), + zap.Error(err), + ) + c.JSON(http.StatusInternalServerError, CreateResponse{ + Success: false, + Message: "保存国旗信息失败", + }) + return + } + zap.L().Debug("📝 country_flag表插入成功", + zap.String("req_id", reqID), + zap.String("country_id", countryID), + ) + } + + // 提交事务 if err := tx.Commit(); err != nil { - tx.Rollback() // 提交失败时尝试回滚 + tx.Rollback() zap.L().Error("❌ 事务提交失败", zap.String("req_id", reqID), zap.String("country_id", countryID), @@ -175,7 +188,6 @@ func CreateHandler(c *gin.Context) { return } - // 记录请求处理耗时 duration := time.Since(startTime) zap.L().Info("✅ 国家创建请求处理完成", zap.String("req_id", reqID), @@ -183,7 +195,6 @@ func CreateHandler(c *gin.Context) { zap.Duration("duration", duration), ) - // 返回成功响应,包含创建的国家ID c.JSON(http.StatusOK, CreateResponse{ Success: true, Message: "创建成功", @@ -191,4 +202,4 @@ func CreateHandler(c *gin.Context) { CountryID: countryID, }, }) -} +} \ No newline at end of file