add

backend/src/logic4variety/update.go

@@ -1 +1,208 @@
 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: "更新成功",
+	})
+}