优化支付流程，扩展产品价格返回字段

web/handlers/trade.go

@@ -1,10 +1,7 @@
 package handlers
 
 import (
-	"bufio"
-	"fmt"
 	"log/slog"
-	"platform/pkg/env"
 	"platform/web/auth"
 	"platform/web/core"
 	g "platform/web/globals"
@@ -14,7 +11,6 @@ import (
 	"time"
 
 	"github.com/gofiber/fiber/v2"
-	"github.com/valyala/fasthttp"
 )
 
 // PageTradeByAdmin 分页查询所有订单
@@ -221,6 +217,36 @@ type TradeCreateReq struct {
 
 // ============================================================
 
+// 更新订单备注
+func TradeUpdateRemarkByAdmin(c *fiber.Ctx) error {
+	// 检查权限
+	_, err := auth.GetAuthCtx(c).PermitAdmin()
+	if err != nil {
+		return err
+	}
+
+	// 解析请求参数
+	var req TradeUpdateRemarkReq
+	if err := g.Validator.ParseBody(c, &req); err != nil {
+		return err
+	}
+
+	// 更新订单备注
+	err = s.Trade.UpdateRemark(req.TradeNo, req.Remark)
+	if err != nil {
+		return err
+	}
+
+	return c.SendStatus(fiber.StatusNoContent)
+}
+
+type TradeUpdateRemarkReq struct {
+	TradeNo string `json:"trade_no" validate:"required"`
+	Remark  string `json:"remark"`
+}
+
+// ============================================================
+
 // 完成订单
 func TradeComplete(c *fiber.Ctx) error {
 	// 检查权限
@@ -276,6 +302,29 @@ func TradeCompleteByAdmin(c *fiber.Ctx) error {
 	return c.SendStatus(fiber.StatusNoContent)
 }
 
+// 订单补余额
+func TradeConvertByAdmin(c *fiber.Ctx) error {
+	// 检查权限
+	authCtx, err := auth.GetAuthCtx(c).PermitAdmin(core.ScopeTradeWrite)
+	if err != nil {
+		return err
+	}
+
+	// 解析请求参数
+	var req s.TradeRef
+	if err := g.Validator.ParseBody(c, &req); err != nil {
+		return err
+	}
+
+	// 订单补余额
+	err = s.Trade.ConvertTradeToBalance(authCtx.Admin, &req)
+	if err != nil {
+		return err
+	}
+
+	return c.SendStatus(fiber.StatusNoContent)
+}
+
 // ============================================================
 
 // 取消订单
@@ -287,14 +336,14 @@ func TradeCancel(c *fiber.Ctx) error {
 	}
 
 	// 解析请求参数
-	req := new(TradeCancelReq)
+	req := new(s.TradeRef)
 	if err := g.Validator.ParseBody(c, req); err != nil {
 		return err
 	}
 
 	// 取消交易
-	err = s.Trade.CancelTrade(&req.TradeRef)
-	if err != nil {
+	err = s.Trade.CancelTrade(req)
+	if err != nil && err != s.ErrTradeStatusIgnored {
 		slog.Error("取消交易失败", "trade_no", req.TradeNo, "error", err)
 		return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "取消交易失败"})
 	}
@@ -302,62 +351,63 @@ func TradeCancel(c *fiber.Ctx) error {
 	return c.SendStatus(fiber.StatusNoContent)
 }
 
-type TradeCancelReq struct {
-	s.TradeRef
+// ============================================================
+
+// 结束订单：完成或取消订单
+func TradeFinish(c *fiber.Ctx) error {
+	// 检查权限
+	authCtx, err := auth.GetAuthCtx(c).PermitUser()
+	if err != nil {
+		return err
+	}
+
+	// 解析请求参数
+	var req s.TradeRef
+	if err := g.Validator.ParseBody(c, &req); err != nil {
+		return err
+	}
+
+	// 尝试取消交易
+	err = s.Trade.CancelTrade(&req)
+	if err == s.ErrTradeStatusIgnored {
+		// 尝试完成交易
+		err = s.Trade.CompleteTrade(authCtx.User, &req)
+		if err != nil {
+			return err
+		}
+
+		return c.JSON(map[string]m.TradeStatus{
+			"status": m.TradeStatusSuccess,
+		})
+	} else if err != nil {
+		return err
+	}
+
+	return c.JSON(map[string]m.TradeStatus{
+		"status": m.TradeStatusCanceled,
+	})
 }
 
 // ============================================================
 
 // 检查订单
 func TradeCheck(c *fiber.Ctx) error {
-	// 检查权限：sse 接口暂时不检查权限
-
-	// 解析请求参数
-	req := new(TradeCheckReq)
-	if err := g.Validator.ParseQuery(c, req); err != nil {
+	_, err := auth.GetAuthCtx(c).PermitAdmin(core.ScopeTradeRead)
+	if err != nil {
 		return err
 	}
 
-	c.Set(fiber.HeaderContentType, "text/event-stream")
-	c.Set(fiber.HeaderCacheControl, "no-cache")
-	c.Set(fiber.HeaderConnection, "keep-alive")
-	c.Set(fiber.HeaderTransferEncoding, "chunked")
-	c.Context().SetBodyStreamWriter(fasthttp.StreamWriter(func(w *bufio.Writer) {
+	// 解析请求参数
+	var req s.TradeRef
+	if err := g.Validator.ParseBody(c, &req); err != nil {
+		return err
+	}
 
-		expire := env.TradeExpire
-		interval := 5
-		for range expire / interval {
-			// 检查订单状态
-			result, err := s.Trade.CheckTrade(&req.TradeRef)
-			if err != nil {
-				slog.Error("检查订单状态失败", "trade_no", req.TradeNo, "error", err)
-				return
-			}
+	// 检查订单状态
+	result, err := s.Trade.CheckTrade(&req)
+	if err != nil {
+		return err
+	}
 
-			// 写入订单状态
-			_, err = fmt.Fprintf(w, "data: %d\n\n", result.Status)
-			if err != nil {
-				slog.Error("写入订单状态失败", "trade_no", req.TradeNo, "error", err)
-				return
-			}
-
-			err = w.Flush()
-			if err != nil {
-				return
-			}
-
-			// 当订单离开支付状态后结束查询
-			if result.Status != m.TradeStatusPending {
-				return
-			}
-
-			time.Sleep(time.Duration(interval) * time.Second)
-		}
-	}))
-
-	return nil
-}
-
-type TradeCheckReq struct {
-	s.TradeRef
+	return c.JSON(result)
 }