asset_assistant/backend/country/src/logic/update.go

package logic

import (
	"country/db"
	"net/http"
	"time"

	"github.com/gin-gonic/gin"
	"github.com/google/uuid"
	"go.uber.org/zap"
)

// UpdateRequest 更新请求参数结构
type UpdateRequest struct {
	CountryID string `json:"country_id" binding:"required"` // 国家ID，必填
	Name      string `json:"name"`                          // 国家名称，可选
	Code      string `json:"code"`                          // 国家代码，可选
}

// 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
	// 绑定并验证请求参数（主要验证country_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: "请求参数错误：country_id为必填项",
		})
		return
	}

	// 验证name和code不能同时为空
	if req.Name == "" && req.Code == "" {
		zap.L().Warn("⚠️ 请求参数验证失败",
			zap.String("req_id", reqID),
			zap.String("country_id", req.CountryID),
			zap.String("reason", "name和code不能同时为空"),
		)
		c.JSON(http.StatusBadRequest, UpdateResponse{
			Success: false,
			Message: "请求参数错误：name和code不能同时为空",
		})
		return
	}

	zap.L().Debug("✅ 请求参数验证通过",
		zap.String("req_id", reqID),
		zap.String("country_id", req.CountryID),
		zap.String("name", req.Name),
		zap.String("code", req.Code),
	)

	// 开启数据库事务
	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 name SET name = $1 WHERE country_id = $2", req.Name, req.CountryID)
		if err != nil {
			tx.Rollback()
			zap.L().Error("❌ name表更新失败",
				zap.String("req_id", reqID),
				zap.String("country_id", req.CountryID),
				zap.Error(err),
			)
			c.JSON(http.StatusInternalServerError, UpdateResponse{
				Success: false,
				Message: "更新名称信息失败",
			})
			return
		}
		zap.L().Debug("📝 name表更新成功",
			zap.String("req_id", reqID),
			zap.String("country_id", req.CountryID),
		)
	}

	// 如果code不为空，更新code表
	if req.Code != "" {
		_, err = tx.Exec("UPDATE code SET code = $1 WHERE country_id = $2", req.Code, req.CountryID)
		if err != nil {
			tx.Rollback()
			zap.L().Error("❌ code表更新失败",
				zap.String("req_id", reqID),
				zap.String("country_id", req.CountryID),
				zap.Error(err),
			)
			c.JSON(http.StatusInternalServerError, UpdateResponse{
				Success: false,
				Message: "更新代码信息失败",
			})
			return
		}
		zap.L().Debug("📝 code表更新成功",
			zap.String("req_id", reqID),
			zap.String("country_id", req.CountryID),
		)
	}

	// 提交事务
	if err := tx.Commit(); err != nil {
		tx.Rollback()
		zap.L().Error("❌ 事务提交失败",
			zap.String("req_id", reqID),
			zap.String("country_id", req.CountryID),
			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("country_id", req.CountryID),
		zap.Duration("duration", duration),
	)

	// 返回成功响应
	c.JSON(http.StatusOK, UpdateResponse{
		Success: true,
		Message: "更新成功",
	})
}