temp

pkg/crypto/generate.go

@@ -4,6 +4,9 @@ import (
 	"crypto/rand"
 	"encoding/hex"
 	"io"
+	mathrand "math/rand"
+	"strconv"
+	"time"
 )
 
 // 生成AES-128密钥的函数，符合市面规范
@@ -29,3 +32,32 @@ func GenerateSecretId() (string, error) {
 	// 将字节数组转换为16进制字符串
 	return hex.EncodeToString(bytes), nil
 }
+
+// GenerateTransactionID 生成16位数的交易单号
+func GenerateTransactionID() string {
+	length := 16
+	// 获取当前时间戳
+	timestamp := time.Now().UnixNano()
+
+	// 转换为字符串
+	timeStr := strconv.FormatInt(timestamp, 10)
+
+	// 生成随机数
+	mathrand.Seed(time.Now().UnixNano())
+	randomPart := strconv.Itoa(mathrand.Intn(1000000))
+
+	// 组合时间戳和随机数
+	combined := timeStr + randomPart
+
+	// 如果长度超出指定值，则截断；如果不够，则填充随机字符
+	if len(combined) >= length {
+		return combined[:length]
+	}
+
+	// 如果长度不够，填充0
+	for len(combined) < length {
+		combined += strconv.Itoa(mathrand.Intn(10)) // 填充随机数
+	}
+
+	return combined
+}

pkg/response/response.go

@@ -1,6 +1,7 @@
 package response
 
 import (
+	"context"
 	"github.com/zeromicro/go-zero/rest/httpx"
 	"net/http"
 )
@@ -11,59 +12,42 @@ type Response struct {
 	Data    interface{} `json:"data,omitempty"`
 	Message string      `json:"message"`
 }
-
-// 定义分页响应结构
-type PageResult struct {
-	List     interface{} `json:"list"`
-	Total    int64       `json:"total"`
-	Page     int         `json:"page"`
-	PageSize int         `json:"pageSize"`
+type ResponseWithTransactionID struct {
+	Response
+	TransactionID string `json:"transaction_id"`
 }
 
 // 响应成功
-func Success(w http.ResponseWriter, data interface{}) {
-	result := Response{
-		Code:    http.StatusOK,
-		Data:    data,
-		Message: "操作成功",
+func Success(ctx context.Context, w http.ResponseWriter, data interface{}) {
+	// 从上下文中获取 TransactionID
+	transactionID := ctx.Value("TransactionID")
+
+	// 判断是否存在 TransactionID
+	if transactionID != nil {
+		result := ResponseWithTransactionID{
+			Response: Response{
+				Code:    http.StatusOK,
+				Data:    data,
+				Message: "success",
+			},
+			TransactionID: transactionID.(string), // 将 TransactionID 添加到响应
+		}
+		httpx.OkJsonCtx(ctx, w, result) // 返回带有 TransactionID 的响应
+	} else {
+		result := Response{
+			Code:    http.StatusOK,
+			Data:    data,
+			Message: "success",
+		}
+		httpx.OkJsonCtx(ctx, w, result) // 返回没有 TransactionID 的响应
 	}
-	httpx.OkJson(w, result)
 }
 
 // 响应失败
-func Fail(w http.ResponseWriter, code int, message string) {
+func Fail(ctx context.Context, w http.ResponseWriter, err error) {
 	result := Response{
-		Code:    code,
-		Message: message,
+		Code:    -1,
+		Message: err.Error(),
 	}
-	httpx.WriteJson(w, code, result)
-}
-
-// 无权限
-func Unauthorized(w http.ResponseWriter, message string) {
-	result := Response{
-		Code:    http.StatusUnauthorized,
-		Message: message,
-	}
-	httpx.WriteJson(w, http.StatusUnauthorized, result)
-}
-
-// 响应分页数据
-func Page(w http.ResponseWriter, list interface{}, total int64, page int, pageSize int) {
-	result := Response{
-		Code:    http.StatusOK,
-		Data:    PageResult{List: list, Total: total, Page: page, PageSize: pageSize},
-		Message: "查询成功",
-	}
-	httpx.OkJson(w, result)
-}
-
-// 自定义错误响应
-func CustomError(w http.ResponseWriter, code int, message string, data interface{}) {
-	result := Response{
-		Code:    code,
-		Data:    data,
-		Message: message,
-	}
-	httpx.WriteJson(w, code, result)
+	httpx.OkJsonCtx(ctx, w, result)
 }