实现 sse 检查订单，减少请求次数

web/handlers/trade.go

@@ -1,15 +1,20 @@
 package handlers
 
 import (
+	"bufio"
+	"fmt"
 	"log/slog"
+	"platform/pkg/env"
 	"platform/web/auth"
 	"platform/web/core"
 	g "platform/web/globals"
 	m "platform/web/models"
 	s "platform/web/services"
+	"reflect"
 	"time"
 
 	"github.com/gofiber/fiber/v2"
+	"github.com/valyala/fasthttp"
 )
 
 type TradeCreateReq struct {
@@ -33,7 +38,7 @@ func TradeCreate(c *fiber.Ctx) error {
 
 	// 解析请求参数
 	req := new(TradeCreateReq)
-	if err := g.Validator.Validate(c, req); err != nil {
+	if err := g.Validator.ParseBody(c, req); err != nil {
 		return err
 	}
 
@@ -76,7 +81,7 @@ func TradeComplete(c *fiber.Ctx) error {
 
 	// 解析请求参数
 	req := new(TradeCompleteReq)
-	if err := g.Validator.Validate(c, req); err != nil {
+	if err := g.Validator.ParseBody(c, req); err != nil {
 		return err
 	}
 
@@ -102,7 +107,7 @@ func TradeCancel(c *fiber.Ctx) error {
 
 	// 解析请求参数
 	req := new(TradeCancelReq)
-	if err := g.Validator.Validate(c, req); err != nil {
+	if err := g.Validator.ParseBody(c, req); err != nil {
 		return err
 	}
 
@@ -115,3 +120,54 @@ func TradeCancel(c *fiber.Ctx) error {
 
 	return c.SendStatus(fiber.StatusNoContent)
 }
+
+type TradeCheckReq struct {
+	s.ModifyTradeData
+}
+
+func TradeCheck(c *fiber.Ctx) error {
+	// 解析请求参数
+	req := new(TradeCheckReq)
+	if err := g.Validator.ParseQuery(c, req); 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) {
+
+		expire := env.TradeExpire
+		interval := 5
+		for range expire / interval {
+			// 检查订单状态
+			result, err := s.Trade.CheckTrade(&req.ModifyTradeData)
+			if err != nil {
+				slog.Error("检查订单状态失败", "trade_no", req.TradeNo, "error", err)
+				return
+			}
+
+			// 写入订单状态
+			_, 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 {
+				slog.Error("刷新缓冲区失败", "trade_no", req.TradeNo, "error", err, "errType", reflect.TypeOf(err))
+				return
+			}
+
+			// 当订单离开支付状态后结束查询
+			if result.Status != m.TradeStatusPending {
+				return
+			}
+
+			time.Sleep(time.Duration(interval) * time.Second)
+		}
+	}))
+
+	return nil
+}