fish/asset_assistant
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add

Browse Source
This commit is contained in:
vipg
2025-11-19 16:41:20 +08:00
parent 55df63985a
commit 01c6d0fda2
1 changed files with 46 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

81
backend/src/logic4country/create.go
View File

@@ -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,
},
})
}
}