first commit

app/main/api/internal/logic/pay/wechatpaycallbacklogic.go

@@ -0,0 +1,90 @@
+package pay
+
+import (
+	"aedata-server/app/main/api/internal/service"
+	"aedata-server/pkg/lzkit/lzUtils"
+	"context"
+	"net/http"
+	"time"
+
+	"aedata-server/app/main/api/internal/svc"
+
+	"github.com/wechatpay-apiv3/wechatpay-go/services/payments"
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type WechatPayCallbackLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewWechatPayCallbackLogic(ctx context.Context, svcCtx *svc.ServiceContext) *WechatPayCallbackLogic {
+	return &WechatPayCallbackLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *WechatPayCallbackLogic) WechatPayCallback(w http.ResponseWriter, r *http.Request) error {
+	notification, err := l.svcCtx.WechatPayService.HandleWechatPayNotification(l.ctx, r)
+	if err != nil {
+		logx.Errorf("微信支付回调，%v", err)
+		return nil
+	}
+
+	// 兼容旧订单，假设没有前缀的是查询订单
+	return l.handleQueryOrderPayment(w, notification)
+}
+
+// 处理查询订单支付
+func (l *WechatPayCallbackLogic) handleQueryOrderPayment(w http.ResponseWriter, notification *payments.Transaction) error {
+	order, findOrderErr := l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, *notification.OutTradeNo)
+	if findOrderErr != nil {
+		logx.Errorf("微信支付回调，查找订单信息失败: %+v", findOrderErr)
+		return nil
+	}
+
+	amount := lzUtils.ToWechatAmount(order.Amount)
+	if amount != *notification.Amount.Total {
+		logx.Errorf("微信支付回调，金额不一致")
+		return nil
+	}
+
+	if order.Status != "pending" {
+		w.WriteHeader(http.StatusOK)
+		_, _ = w.Write([]byte("success"))
+		return nil
+	}
+
+	switch *notification.TradeState {
+	case service.TradeStateSuccess:
+		order.Status = "paid"
+		order.PayTime = lzUtils.TimeToNullTime(time.Now())
+	case service.TradeStateClosed:
+		order.Status = "closed"
+		order.CloseTime = lzUtils.TimeToNullTime(time.Now())
+	case service.TradeStateRevoked:
+		order.Status = "failed"
+	default:
+		return nil
+	}
+
+	order.PlatformOrderId = lzUtils.StringToNullString(*notification.TransactionId)
+	if updateErr := l.svcCtx.OrderModel.UpdateWithVersion(l.ctx, nil, order); updateErr != nil {
+		logx.Errorf("微信支付回调，更新订单失败%+v", updateErr)
+		return nil
+	}
+
+	if order.Status == "paid" {
+		if asyncErr := l.svcCtx.AsynqService.SendQueryTask(order.Id); asyncErr != nil {
+			logx.Errorf("异步任务调度失败: %v", asyncErr)
+			return asyncErr
+		}
+	}
+
+	w.WriteHeader(http.StatusOK)
+	_, _ = w.Write([]byte("success"))
+	return nil
+}