This commit is contained in:
vipg
2025-11-14 18:02:56 +08:00
parent 54e0a5a0c4
commit 81170f637e

View File

@@ -10,84 +10,78 @@ import (
)
// CreateVarietyRequest 注册请求参数结构
// 用于接收客户端发送的JSON数据绑定并验证必填字段
type CreateVarietyRequest struct {
Name string `json:"name" binding:"required"` // 品种名称,必填
Code string `json:"code" binding:"required"` // 品种代码,必填
Name string `json:"name" binding:"required"` // 品种名称,必填
Code string `json:"code" binding:"required"` // 品种代码,必填
Tick float64 `json:"tick" binding:"omitempty,min=0"` // 跳点值非必填需大于等于0
TickPrice float64 `json:"tick_price" binding:"omitempty,min=0"` // 跳点价格非必填需大于等于0
}
// CreateVarietyResponse 注册响应结构
// 统一的API响应格式包含成功状态、提示信息和数据
type CreateVarietyResponse struct {
Success bool `json:"success"` // 操作是否成功
Message string `json:"message"` // 提示信息
Data CreateData `json:"data"` // 响应数据
Success bool `json:"success"` // 操作是否成功
Message string `json:"message"` // 提示信息
Data CreateVarietyData `json:"data"` // 响应数据
}
// CreateVarietyData 响应数据结构
// 包含创建成功后的品种ID
type CreateVarietyData struct {
VarietyID string `json:"variety_id"` // 品种唯一标识ID
}
// CreateVarietyHandler 处理品种创建逻辑
// 接收HTTP请求完成参数验证、数据库事务处理并返回响应
func CreateVarietyHandler(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),
zap.String("method", c.Request.Method),
)
var req CreateRequest
// 绑定并验证请求参数(检查name和code是否存在
var req CreateVarietyRequest
// 绑定并验证请求参数(包含新增的tick和tick_price
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{
c.JSON(http.StatusBadRequest, CreateVarietyResponse{
Success: false,
Message: "请求参数错误name和code为必填项",
Message: "请求参数错误name和code为必填项tick和tick_price需大于等于0",
})
return
}
// 记录通过验证的请求参数
zap.L().Debug("✅ 请求参数验证通过",
zap.String("req_id", reqID),
zap.String("name", req.Name),
zap.String("code", req.Code),
zap.Float64("tick", req.Tick),
zap.Float64("tick_price", req.TickPrice),
)
// 开启数据库事务,确保多表操作原子性(要么全成功,要么全失败)
// 开启数据库事务
tx, err := db.DB.Begin()
if err != nil {
zap.L().Error("❌ 事务开启失败",
zap.String("req_id", reqID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, CreateResponse{
c.JSON(http.StatusInternalServerError, CreateVarietyResponse{
Success: false,
Message: "系统错误,请稍后重试",
})
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),
@@ -98,8 +92,7 @@ func CreateVarietyHandler(c *gin.Context) {
zap.String("req_id", reqID),
zap.Any("recover", r),
)
// 返回系统错误响应
c.JSON(http.StatusInternalServerError, CreateResponse{
c.JSON(http.StatusInternalServerError, CreateVarietyResponse{
Success: false,
Message: "系统错误,请稍后重试",
})
@@ -110,12 +103,12 @@ func CreateVarietyHandler(c *gin.Context) {
var varietyID string
err = tx.QueryRow("INSERT INTO variety DEFAULT VALUES RETURNING id").Scan(&varietyID)
if err != nil {
tx.Rollback() // 操作失败,回滚事务
tx.Rollback()
zap.L().Error("❌ variety表插入失败",
zap.String("req_id", reqID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, CreateResponse{
c.JSON(http.StatusInternalServerError, CreateVarietyResponse{
Success: false,
Message: "创建品种记录失败",
})
@@ -127,47 +120,79 @@ func CreateVarietyHandler(c *gin.Context) {
zap.String("variety_id", varietyID),
)
// 2. 插入品种名称到name表variety_id关联
_, err = tx.Exec("INSERT INTO name (variety_id, name) VALUES ($1, $2)", varietyID, req.Name)
// 2. 插入品种名称到variety_name表修正表名
_, err = tx.Exec("INSERT INTO variety_name (variety_id, name) VALUES ($1, $2)", varietyID, req.Name)
if err != nil {
tx.Rollback() // 操作失败,回滚事务
zap.L().Error("❌ name表插入失败",
tx.Rollback()
zap.L().Error("❌ variety_name表插入失败",
zap.String("req_id", reqID),
zap.String("variety_id", varietyID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, CreateResponse{
c.JSON(http.StatusInternalServerError, CreateVarietyResponse{
Success: false,
Message: "保存名称信息失败",
})
return
}
// 3. 插入品种代码到code表variety_id关联
_, err = tx.Exec("INSERT INTO code (variety_id, code) VALUES ($1, $2)", varietyID, req.Code)
// 3. 插入品种代码到variety_code表修正表名
_, err = tx.Exec("INSERT INTO variety_code (variety_id, code) VALUES ($1, $2)", varietyID, req.Code)
if err != nil {
tx.Rollback() // 操作失败,回滚事务
zap.L().Error("❌ code表插入失败",
tx.Rollback()
zap.L().Error("❌ variety_code表插入失败",
zap.String("req_id", reqID),
zap.String("variety_id", varietyID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, CreateResponse{
c.JSON(http.StatusInternalServerError, CreateVarietyResponse{
Success: false,
Message: "保存代码信息失败",
})
return
}
// 提交事务(所有操作成功后确认提交
// 4. 插入跳点值到variety_tick表新增
_, err = tx.Exec("INSERT INTO variety_tick (variety_id, tick) VALUES ($1, $2)", varietyID, req.Tick)
if err != nil {
tx.Rollback()
zap.L().Error("❌ variety_tick表插入失败",
zap.String("req_id", reqID),
zap.String("variety_id", varietyID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, CreateVarietyResponse{
Success: false,
Message: "保存跳点信息失败",
})
return
}
// 5. 插入跳点价格到variety_tick_price表新增
_, err = tx.Exec("INSERT INTO variety_tick_price (variety_id, price) VALUES ($1, $2)", varietyID, req.TickPrice)
if err != nil {
tx.Rollback()
zap.L().Error("❌ variety_tick_price表插入失败",
zap.String("req_id", reqID),
zap.String("variety_id", varietyID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, CreateVarietyResponse{
Success: false,
Message: "保存跳点价格信息失败",
})
return
}
// 提交事务
if err := tx.Commit(); err != nil {
tx.Rollback() // 提交失败时尝试回滚
tx.Rollback()
zap.L().Error("❌ 事务提交失败",
zap.String("req_id", reqID),
zap.String("variety_id", varietyID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, CreateResponse{
c.JSON(http.StatusInternalServerError, CreateVarietyResponse{
Success: false,
Message: "数据提交失败,请稍后重试",
})
@@ -182,11 +207,11 @@ func CreateVarietyHandler(c *gin.Context) {
zap.Duration("duration", duration),
)
// 返回成功响应包含创建的品种ID
c.JSON(http.StatusOK, CreateResponse{
// 返回成功响应
c.JSON(http.StatusOK, CreateVarietyResponse{
Success: true,
Message: "创建成功",
Data: CreateData{
Data: CreateVarietyData{
VarietyID: varietyID,
},
})