add

backend/src/logic4exchange/create.go

@@ -0,0 +1,194 @@
+package logic4exchange
+
+import (
+	"asset_assistant/db" // 数据库操作相关包
+	"net/http"
+	"time" // 时间处理包
+
+	"github.com/gin-gonic/gin" // Gin框架，用于处理HTTP请求
+	"github.com/google/uuid"   // UUID生成工具
+	"go.uber.org/zap"          // 日志库
+)
+
+// CreateRequest 注册请求参数结构
+// 用于接收客户端发送的JSON数据，绑定并验证必填字段
+type CreateRequest struct {
+	Name string `json:"name" binding:"required"` // 国家名称，必填
+	Code string `json:"code" binding:"required"` // 国家代码，必填
+}
+
+// CreateResponse 注册响应结构
+// 统一的API响应格式，包含成功状态、提示信息和数据
+type CreateResponse struct {
+	Success bool       `json:"success"` // 操作是否成功
+	Message string     `json:"message"` // 提示信息
+	Data    CreateData `json:"data"`    // 响应数据
+}
+
+// CreateData 响应数据结构
+// 包含创建成功后的国家ID
+type CreateData struct {
+	CountryID string `json:"exchange_id"` // 国家唯一标识ID
+}
+
+// CreateHandler 处理国家创建逻辑
+// 接收HTTP请求，完成参数验证、数据库事务处理并返回响应
+func CreateHandler(c *gin.Context) {
+	startTime := time.Now() // 记录请求开始时间，用于统计耗时
+	// 获取或生成请求ID，用于追踪整个请求链路
+	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 CreateRequest
+	// 绑定并验证请求参数（检查name和code是否存在）
+	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为必填项",
+		})
+		return
+	}
+
+	// 记录通过验证的请求参数
+	zap.L().Debug("✅ 请求参数验证通过",
+		zap.String("req_id", reqID),
+		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, CreateResponse{
+			Success: false,
+			Message: "系统错误，请稍后重试",
+		})
+		return
+	}
+	// 延迟执行的恢复函数，处理panic情况
+	defer func() {
+		if r := recover(); r != nil { // 捕获panic
+			// 回滚事务
+			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, CreateResponse{
+				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, CreateResponse{
+			Success: false,
+			Message: "创建国家记录失败",
+		})
+		return
+	}
+
+	zap.L().Debug("📝 exchange表插入成功",
+		zap.String("req_id", reqID),
+		zap.String("exchange_id", exchangeID),
+	)
+
+	// 2. 插入国家名称到name表（与exchange_id关联）
+	_, err = tx.Exec("INSERT INTO name (exchange_id, name) VALUES ($1, $2)", exchangeID, req.Name)
+	if err != nil {
+		tx.Rollback() // 操作失败，回滚事务
+		zap.L().Error("❌ name表插入失败",
+			zap.String("req_id", reqID),
+			zap.String("exchange_id", exchangeID),
+			zap.Error(err),
+		)
+		c.JSON(http.StatusInternalServerError, CreateResponse{
+			Success: false,
+			Message: "保存名称信息失败",
+		})
+		return
+	}
+
+	// 3. 插入国家代码到code表（与exchange_id关联）
+	_, err = tx.Exec("INSERT INTO code (exchange_id, code) VALUES ($1, $2)", exchangeID, req.Code)
+	if err != nil {
+		tx.Rollback() // 操作失败，回滚事务
+		zap.L().Error("❌ code表插入失败",
+			zap.String("req_id", reqID),
+			zap.String("exchange_id", exchangeID),
+			zap.Error(err),
+		)
+		c.JSON(http.StatusInternalServerError, CreateResponse{
+			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, CreateResponse{
+			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),
+	)
+
+	// 返回成功响应，包含创建的国家ID
+	c.JSON(http.StatusOK, CreateResponse{
+		Success: true,
+		Message: "创建成功",
+		Data: CreateData{
+			CountryID: exchangeID,
+		},
+	})
+}