This commit is contained in:
vipg
2025-11-15 17:18:00 +08:00
parent c1870aa135
commit 578e1596ef
3 changed files with 218 additions and 2 deletions

View File

@@ -0,0 +1,216 @@
package logic4exchange
import (
"futures_trade_record/db" // 数据库操作相关包
"net/http"
"time" // 时间处理包
"github.com/gin-gonic/gin" // Gin框架用于处理HTTP请求
"github.com/google/uuid" // UUID生成工具
"go.uber.org/zap" // 日志库
)
// CreateExchangeRequest 注册请求参数结构
type CreateExchangeRequest struct {
}
// CreateExchangeResponse 注册响应结构
type CreateExchangeResponse struct {
Success bool `json:"success"` // 操作是否成功
Message string `json:"message"` // 提示信息
Data CreateExchangeData `json:"data"` // 响应数据
}
// CreateExchangeData 响应数据结构
type CreateExchangeData struct {
ExchangeID string `json:"exchange_id"` // 交易所唯一标识ID
}
// CreateExchangeHandler 处理交易所创建逻辑
func CreateExchangeHandler(c *gin.Context) {
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 CreateExchangeRequest
// 绑定并验证请求参数包含新增的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, CreateExchangeResponse{
Success: false,
Message: "请求参数错误name、code、tick和tick_price为必填项且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, CreateExchangeResponse{
Success: false,
Message: "系统错误,请稍后重试",
})
return
}
defer func() {
if r := recover(); r != nil {
if err := tx.Rollback(); err != nil {
zap.L().Error("💥 panic后事务回滚失败",
zap.String("req_id", reqID),
zap.Error(err),
)
}
zap.L().Error("💥 事务处理发生panic",
zap.String("req_id", reqID),
zap.Any("recover", r),
)
c.JSON(http.StatusInternalServerError, CreateExchangeResponse{
Success: false,
Message: "系统错误,请稍后重试",
})
}
}()
// 1. 在exchange表中创建记录并获取自动生成的ID
var exchangeID string
err = tx.QueryRow("INSERT INTO exchange DEFAULT VALUES RETURNING id").Scan(&exchangeID)
if err != nil {
tx.Rollback()
zap.L().Error("❌ exchange表插入失败",
zap.String("req_id", reqID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, CreateExchangeResponse{
Success: false,
Message: "创建交易所记录失败",
})
return
}
zap.L().Debug("📝 exchange表插入成功",
zap.String("req_id", reqID),
zap.String("exchange_id", exchangeID),
)
// 2. 插入交易所名称到exchange_name表修正表名
_, err = tx.Exec("INSERT INTO exchange_name (exchange_id, name) VALUES ($1, $2)", exchangeID, req.Name)
if err != nil {
tx.Rollback()
zap.L().Error("❌ exchange_name表插入失败",
zap.String("req_id", reqID),
zap.String("exchange_id", exchangeID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, CreateExchangeResponse{
Success: false,
Message: "保存名称信息失败",
})
return
}
// 3. 插入交易所代码到exchange_code表修正表名
_, err = tx.Exec("INSERT INTO exchange_code (exchange_id, code) VALUES ($1, $2)", exchangeID, req.Code)
if err != nil {
tx.Rollback()
zap.L().Error("❌ exchange_code表插入失败",
zap.String("req_id", reqID),
zap.String("exchange_id", exchangeID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, CreateExchangeResponse{
Success: false,
Message: "保存代码信息失败",
})
return
}
// 4. 插入跳点值到exchange_tick表新增
_, err = tx.Exec("INSERT INTO exchange_tick (exchange_id, tick) VALUES ($1, $2)", exchangeID, req.Tick)
if err != nil {
tx.Rollback()
zap.L().Error("❌ exchange_tick表插入失败",
zap.String("req_id", reqID),
zap.String("exchange_id", exchangeID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, CreateExchangeResponse{
Success: false,
Message: "保存跳点信息失败",
})
return
}
// 5. 插入跳点价格到exchange_tick_price表新增
_, err = tx.Exec("INSERT INTO exchange_tick_price (exchange_id, price) VALUES ($1, $2)", exchangeID, req.TickPrice)
if err != nil {
tx.Rollback()
zap.L().Error("❌ exchange_tick_price表插入失败",
zap.String("req_id", reqID),
zap.String("exchange_id", exchangeID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, CreateExchangeResponse{
Success: false,
Message: "保存跳点价格信息失败",
})
return
}
// 提交事务
if err := tx.Commit(); err != nil {
tx.Rollback()
zap.L().Error("❌ 事务提交失败",
zap.String("req_id", reqID),
zap.String("exchange_id", exchangeID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, CreateExchangeResponse{
Success: false,
Message: "数据提交失败,请稍后重试",
})
return
}
// 记录请求处理耗时
duration := time.Since(startTime)
zap.L().Info("✅ 交易所创建请求处理完成",
zap.String("req_id", reqID),
zap.String("exchange_id", exchangeID),
zap.Duration("duration", duration),
)
// 返回成功响应
c.JSON(http.StatusOK, CreateExchangeResponse{
Success: true,
Message: "创建成功",
Data: CreateExchangeData{
ExchangeID: exchangeID,
},
})
}

View File

@@ -1,4 +1,4 @@
package logic
package logic4variety
import (
"futures_trade_record/db" // 数据库操作相关包

View File

@@ -3,7 +3,7 @@ package main
import (
"futures_trade_record/db" // 数据库相关操作包
"futures_trade_record/logger" // 日志工具包
"futures_trade_record/logic" // 业务逻辑处理包
"futures_trade_record/logic/variety" // 业务逻辑处理包
"time"
"github.com/gin-contrib/cors"