From 4430d77c81a280c7ee44e4c37362b26e0d1b3ce7 Mon Sep 17 00:00:00 2001 From: vipg Date: Wed, 26 Nov 2025 16:23:04 +0800 Subject: [PATCH] add --- backend/src/logic4variety/update.go | 209 +++++++++++++++++++++++++++- 1 file changed, 208 insertions(+), 1 deletion(-) diff --git a/backend/src/logic4variety/update.go b/backend/src/logic4variety/update.go index 92360ce..8a0a9da 100644 --- a/backend/src/logic4variety/update.go +++ b/backend/src/logic4variety/update.go @@ -1 +1,208 @@ -package logic4variety \ No newline at end of file +package logic4variety + +import ( + "asset_assistant/db" + "net/http" + "time" + + "github.com/gin-gonic/gin" + "github.com/google/uuid" + "go.uber.org/zap" +) + +// UpdateRequest 更新请求参数结构 +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 更新响应结构 +type UpdateResponse struct { + Success bool `json:"success"` // 操作是否成功 + Message string `json:"message"` // 提示信息 +} + +// UpdateHandler 处理信息更新逻辑 +func UpdateHandler(c *gin.Context) { + startTime := time.Now() + // 获取或生成请求ID + reqID := c.Request.Header.Get("X-UpdateRequest-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 UpdateRequest + // 绑定并验证请求参数（主要验证exchange_id必填） + if err := c.ShouldBindJSON(&req); err != nil { + zap.L().Warn("⚠️ 请求参数验证失败", + zap.String("req_id", reqID), + zap.Error(err), + ) + c.JSON(http.StatusBadRequest, UpdateResponse{ + Success: false, + Message: "请求参数错误：exchange_id为必填项", + }) + return + } + + // 验证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和short_name不能同时为空"), + ) + c.JSON(http.StatusBadRequest, UpdateResponse{ + Success: false, + Message: "请求参数错误：name、code和short_name不能同时为空", + }) + return + } + + zap.L().Debug("✅ 请求参数验证通过", + zap.String("req_id", reqID), + zap.String("exchange_id", req.ExchangeID), + 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("❌ 事务开启失败", + zap.String("req_id", reqID), + zap.Error(err), + ) + c.JSON(http.StatusInternalServerError, UpdateResponse{ + Success: false, + Message: "系统错误，请稍后重试", + }) + return + } + + // 延迟处理panic情况 + 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, UpdateResponse{ + Success: false, + Message: "系统错误，请稍后重试", + }) + } + }() + + // 如果name不为空，更新name表 + if req.Name != "" { + _, err = tx.Exec("UPDATE exchange_name SET name = $1 WHERE exchange_id = $2", req.Name, req.ExchangeID) + if err != nil { + tx.Rollback() + zap.L().Error("❌ exchange_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("📝 name表更新成功", + zap.String("req_id", reqID), + zap.String("exchange_id", req.ExchangeID), + ) + } + + // 如果code不为空，更新code表 + if req.Code != "" { + _, err = tx.Exec("UPDATE exchange_code SET code = $1 WHERE exchange_id = $2", req.Code, req.ExchangeID) + if err != nil { + tx.Rollback() + zap.L().Error("❌ exchange_code表更新失败", + 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("📝 code表更新成功", + zap.String("req_id", reqID), + zap.String("exchange_id", req.ExchangeID), + ) + } + + // 新增：如果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() + zap.L().Error("❌ 事务提交失败", + zap.String("req_id", reqID), + zap.String("exchange_id", req.ExchangeID), + zap.Error(err), + ) + c.JSON(http.StatusInternalServerError, UpdateResponse{ + Success: false, + Message: "数据提交失败，请稍后重试", + }) + return + } + + // 记录请求处理耗时 + duration := time.Since(startTime) + zap.L().Info("✅ 交易所更新请求处理完成", + zap.String("req_id", reqID), + zap.String("exchange_id", req.ExchangeID), + zap.Duration("duration", duration), + ) + + // 返回成功响应 + c.JSON(http.StatusOK, UpdateResponse{ + Success: true, + Message: "更新成功", + }) +}