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-18 12:17:29 +08:00
parent d1bee92563
commit 5870edb938
4 changed files with 109 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

80
backend/src/logic4exchange/create.go
View File

@@ -1,48 +1,44 @@
package logic4exchange
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数据，绑定并验证必填字段
// 新增short_name字段，保持必填性与其他核心字段一致
type CreateRequest 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"` // 交易所代码，必填
ShortName string `json:"short_name" binding:"required"` // 交易所短名称，必填
}
// 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 {
ExchangeID string `json:"exchange_id"` // 交易所唯一标识ID
ExchangeID string `json:"exchange_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,29 +46,27 @@ func CreateHandler(c *gin.Context) {
)
var req CreateRequest
// 绑定并验证请求参数（检查name和code是否存在）
// 绑定参数时会自动验证name、code、short_name三个必填字段
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为必填项",
Message: "请求参数错误：name、code和short_name为必填项",
})
return
}
// 记录通过验证的请求参数
zap.L().Debug("✅ 请求参数验证通过",
zap.String("req_id", reqID),
zap.String("name", req.Name),
zap.String("code", req.Code),
zap.String("short_name", req.ShortName), // 新增短名称日志
)
// 开启数据库事务，确保多表操作原子性（要么全成功，要么全失败）
tx, err := db.DB.Begin()
if err != nil {
zap.L().Error("❌ 事务开启失败",
@@ -85,10 +79,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 +92,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 +99,11 @@ func CreateHandler(c *gin.Context) {
}
}()
// 1. 在exchange表中创建记录并获取自动生成的ID
// 1. 创建exchange主记录
var exchangeID string
err = tx.QueryRow("INSERT INTO exchange DEFAULT VALUES RETURNING id").Scan(&exchangeID)
if err != nil {
tx.Rollback() // 操作失败，回滚事务
tx.Rollback()
zap.L().Error("❌ exchange表插入失败",
zap.String("req_id", reqID),
zap.Error(err),
@@ -128,10 +120,10 @@ func CreateHandler(c *gin.Context) {
zap.String("exchange_id", exchangeID),
)
// 2. 插入交易所名称到name表（与exchange_id关联）
// 2. 插入名称信息
_, err = tx.Exec("INSERT INTO exchange_name (exchange_id, name) VALUES ($1, $2)", exchangeID, req.Name)
if err != nil {
tx.Rollback() // 操作失败，回滚事务
tx.Rollback()
zap.L().Error("❌ exchange_name表插入失败",
zap.String("req_id", reqID),
zap.String("exchange_id", exchangeID),
@@ -144,10 +136,26 @@ func CreateHandler(c *gin.Context) {
return
}
// 3. 插入交易所代码到code表（与exchange_id关联）
// 3. 新增：插入短名称信息（对应exchange_short_name表）
_, err = tx.Exec("INSERT INTO exchange_short_name (exchange_id, short_name) VALUES ($1, $2)", exchangeID, req.ShortName)
if err != nil {
tx.Rollback()
zap.L().Error("❌ exchange_short_name表插入失败",
zap.String("req_id", reqID),
zap.String("exchange_id", exchangeID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, CreateResponse{
Success: false,
Message: "保存短名称信息失败",
})
return
}
// 4. 插入代码信息
_, err = tx.Exec("INSERT INTO exchange_code (exchange_id, code) VALUES ($1, $2)", exchangeID, req.Code)
if err != nil {
tx.Rollback() // 操作失败，回滚事务
tx.Rollback()
zap.L().Error("❌ exchange_code表插入失败",
zap.String("req_id", reqID),
zap.String("exchange_id", exchangeID),
@@ -160,9 +168,9 @@ func CreateHandler(c *gin.Context) {
return
}
// 提交事务（所有操作成功后确认提交）
// 提交事务
if err := tx.Commit(); err != nil {
tx.Rollback() // 提交失败时尝试回滚
tx.Rollback()
zap.L().Error("❌ 事务提交失败",
zap.String("req_id", reqID),
zap.String("exchange_id", exchangeID),
@@ -175,7 +183,6 @@ func CreateHandler(c *gin.Context) {
return
}
// 记录请求处理耗时
duration := time.Since(startTime)
zap.L().Info("✅ 交易所创建请求处理完成",
zap.String("req_id", reqID),
@@ -183,7 +190,6 @@ func CreateHandler(c *gin.Context) {
zap.Duration("duration", duration),
)
// 返回成功响应，包含创建的交易所ID
c.JSON(http.StatusOK, CreateResponse{
Success: true,
Message: "创建成功",

22
backend/src/logic4exchange/delete.go
View File

@@ -89,7 +89,7 @@ func DeleteHandler(c *gin.Context) {
}
}()
// 3.1 更新exchange表
// 1. 更新exchange表
_, err = tx.Exec("UPDATE exchange SET deleted = TRUE WHERE id = $1", req.ExchangeID)
if err != nil {
tx.Rollback()
@@ -105,7 +105,7 @@ func DeleteHandler(c *gin.Context) {
return
}
// 3.2 更新name表
// 2. 更新exchange_name表
_, err = tx.Exec("UPDATE exchange_name SET deleted = TRUE WHERE exchange_id = $1", req.ExchangeID)
if err != nil {
tx.Rollback()
@@ -121,7 +121,23 @@ func DeleteHandler(c *gin.Context) {
return
}
// 3.3 更新code表
// 3. 新增：更新exchange_short_name表（软删除短名称记录）
_, err = tx.Exec("UPDATE exchange_short_name SET deleted = TRUE WHERE exchange_id = $1", req.ExchangeID)
if err != nil {
tx.Rollback()
zap.L().Error("❌ exchange_short_name表更新失败",
zap.String("req_id", reqID),
zap.String("exchange_id", req.ExchangeID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, DeleteResponse{
Success: false,
Message: "删除短名称信息失败",
})
return
}
// 4. 更新exchange_code表
_, err = tx.Exec("UPDATE exchange_code SET deleted = TRUE WHERE exchange_id = $1", req.ExchangeID)
if err != nil {
tx.Rollback()

29
backend/src/logic4exchange/read.go
View File

@@ -19,6 +19,7 @@ type ReadRequest struct {
ExchangeID string `form:"exchange_id"` // 交易所ID，可选
Name string `form:"name"` // 交易所名称，可选
Code string `form:"code"` // 交易所代码，可选
ShortName string `form:"short_name"` // 交易所短名称，新增查询条件
Page string `form:"page"` // 页码，可选
PageSize string `form:"page_size"` // 每页条数，可选
}
@@ -36,6 +37,7 @@ type ExchangeInfoViewItem struct {
ExchangeID string `json:"exchange_id"` // 交易所ID
Name string `json:"name"` // 交易所名称
Code string `json:"code"` // 交易所代码
ShortName string `json:"short_name"` // 新增：交易所短名称
}
// ReadResponse 读取响应结构
@@ -76,15 +78,15 @@ func ReadHandler(c *gin.Context) {
return
}
// 验证查询条件至少有一个不为空
if req.ExchangeID == "" && req.Name == "" && req.Code == "" {
// 验证查询条件至少有一个不为空（新增short_name作为可选条件）
if req.ExchangeID == "" && req.Name == "" && req.Code == "" && req.ShortName == "" {
zap.L().Warn("⚠️ 请求参数验证失败",
zap.String("req_id", reqID),
zap.String("reason", "exchange_id、name、code不能同时为空"),
zap.String("reason", "exchange_id、name、code、short_name不能同时为空"),
)
c.JSON(http.StatusBadRequest, ReadResponse{
Success: false,
Message: "请求参数错误：exchange_id、name、code不能同时为空",
Message: "请求参数错误：exchange_id、name、code、short_name不能同时为空",
})
return
}
@@ -104,6 +106,7 @@ func ReadHandler(c *gin.Context) {
zap.String("exchange_id", req.ExchangeID),
zap.String("name", req.Name),
zap.String("code", req.Code),
zap.String("short_name", req.ShortName), // 新增短名称日志
zap.Int("page", page),
zap.Int("page_size", pageSize),
)
@@ -123,14 +126,20 @@ func ReadHandler(c *gin.Context) {
args = append(args, "%"+req.Name+"%")
paramIndex++
}
// 新增：短名称查询条件
if req.ShortName != "" {
whereClauses = append(whereClauses, "short_name LIKE $"+strconv.Itoa(paramIndex))
args = append(args, "%"+req.ShortName+"%")
paramIndex++
}
if req.Code != "" {
whereClauses = append(whereClauses, "code LIKE $"+strconv.Itoa(paramIndex))
args = append(args, "%"+req.Code+"%")
paramIndex++
}
// 构建基础SQL
baseSQL := "SELECT exchange_id, name, code FROM exchange_info_view"
// 构建基础SQL（新增查询short_name字段）
baseSQL := "SELECT exchange_id, name, short_name, code FROM exchange_info_view"
countSQL := "SELECT COUNT(*) FROM exchange_info_view"
if len(whereClauses) > 0 {
whereStr := " WHERE " + strings.Join(whereClauses, " AND ")
@@ -141,11 +150,11 @@ func ReadHandler(c *gin.Context) {
// 计算分页偏移量
offset := (page - 1) * pageSize
// 拼接分页SQL（使用fmt.Sprintf更清晰）
// 拼接分页SQL
querySQL := fmt.Sprintf("%s ORDER BY exchange_id LIMIT $%d OFFSET $%d", baseSQL, paramIndex, paramIndex+1)
args = append(args, pageSize, offset)
// 查询总条数（修正参数传递方式）
// 查询总条数
var total int64
countArgs := args[:len(args)-2] // 排除分页参数
err = db.DB.QueryRow(countSQL, countArgs...).Scan(&total)
@@ -176,11 +185,11 @@ func ReadHandler(c *gin.Context) {
}
defer rows.Close()
// 处理查询结果
// 处理查询结果（新增扫描short_name字段）
var items []ExchangeInfoViewItem
for rows.Next() {
var item ExchangeInfoViewItem
if err := rows.Scan(&item.ExchangeID, &item.Name, &item.Code); err != nil {
if err := rows.Scan(&item.ExchangeID, &item.Name, &item.ShortName, &item.Code); err != nil {
zap.L().Error("❌ 解析查询结果失败",
zap.String("req_id", reqID),
zap.Error(err),

32
backend/src/logic4exchange/update.go
View File

@@ -15,6 +15,7 @@ type UpdateRequest struct {
ExchangeID string `json:"exchange_id" binding:"required"` // 交易所ID，必填
Name string `json:"name"` // 交易所名称，可选
Code string `json:"code"` // 交易所代码，可选
ShortName string `json:"short_name"` // 新增：交易所短名称，可选
}
// UpdateResponse 更新响应结构
@@ -54,16 +55,16 @@ func UpdateHandler(c *gin.Context) {
return
}
// 验证name和code不能同时为空
if req.Name == "" && req.Code == "" {
// 验证name、code和short_name不能同时为空
if req.Name == "" && req.Code == "" && req.ShortName == "" {
zap.L().Warn("⚠️ 请求参数验证失败",
zap.String("req_id", reqID),
zap.String("exchange_id", req.ExchangeID),
zap.String("reason", "name和code不能同时为空"),
zap.String("reason", "name、code和short_name不能同时为空"),
)
c.JSON(http.StatusBadRequest, UpdateResponse{
Success: false,
Message: "请求参数错误：name和code不能同时为空",
Message: "请求参数错误：name、code和short_name不能同时为空",
})
return
}
@@ -73,6 +74,7 @@ func UpdateHandler(c *gin.Context) {
zap.String("exchange_id", req.ExchangeID),
zap.String("name", req.Name),
zap.String("code", req.Code),
zap.String("short_name", req.ShortName), // 新增短名称日志
)
// 开启数据库事务
@@ -153,6 +155,28 @@ func UpdateHandler(c *gin.Context) {
)
}
// 新增：如果short_name不为空，更新short_name表
if req.ShortName != "" {
_, err = tx.Exec("UPDATE exchange_short_name SET short_name = $1 WHERE exchange_id = $2", req.ShortName, req.ExchangeID)
if err != nil {
tx.Rollback()
zap.L().Error("❌ exchange_short_name表更新失败",
zap.String("req_id", reqID),
zap.String("exchange_id", req.ExchangeID),
zap.Error(err),
)
c.JSON(http.StatusInternalServerError, UpdateResponse{
Success: false,
Message: "更新短名称信息失败",
})
return
}
zap.L().Debug("📝 short_name表更新成功",
zap.String("req_id", reqID),
zap.String("exchange_id", req.ExchangeID),
)
}
// 提交事务
if err := tx.Commit(); err != nil {
tx.Rollback()