fix wxpay pending order reconciliation

This commit is contained in:
LiuZhouZhouJieLun
2026-05-19 16:56:31 +08:00
parent 2a242aec0f
commit dbd80a0465
9 changed files with 427 additions and 15 deletions
@@ -59,10 +59,18 @@ func (s *PaymentOrderExpiryService) Stop() {
}
func (s *PaymentOrderExpiryService) runOnce() {
ctx, cancel := context.WithTimeout(context.Background(), expiryCheckTimeout)
defer cancel()
reconcileCtx, cancel := context.WithTimeout(context.Background(), expiryCheckTimeout)
recovered, err := s.paymentSvc.ReconcilePendingWxpayOrders(reconcileCtx)
cancel()
if err != nil {
slog.Warn("[PaymentOrderExpiry] failed to reconcile pending wxpay orders", "error", err)
} else if recovered > 0 {
slog.Info("[PaymentOrderExpiry] reconciled paid wxpay orders", "count", recovered)
}
expired, err := s.paymentSvc.ExpireTimedOutOrders(ctx)
expireCtx, cancel := context.WithTimeout(context.Background(), expiryCheckTimeout)
defer cancel()
expired, err := s.paymentSvc.ExpireTimedOutOrders(expireCtx)
if err != nil {
slog.Error("[PaymentOrderExpiry] failed to expire orders", "error", err)
return
@@ -26,8 +26,14 @@ const (
rateLimitModeFixed = "fixed"
checkPaidResultAlreadyPaid = "already_paid"
checkPaidResultCancelled = "cancelled"
pendingWxpayReconcileLimit = 20
)
type checkPaidOptions struct {
cancelIfUnpaid bool
}
func (s *PaymentService) checkCancelRateLimit(ctx context.Context, userID int64, cfg *PaymentConfig) error {
if !cfg.CancelRateLimitEnabled || cfg.CancelRateLimitMax <= 0 {
return nil
@@ -136,6 +142,14 @@ func (s *PaymentService) cancelCore(ctx context.Context, o *dbent.PaymentOrder,
}
func (s *PaymentService) checkPaid(ctx context.Context, o *dbent.PaymentOrder) string {
return s.checkPaidWithOptions(ctx, o, checkPaidOptions{cancelIfUnpaid: true})
}
func (s *PaymentService) reconcilePaid(ctx context.Context, o *dbent.PaymentOrder) string {
return s.checkPaidWithOptions(ctx, o, checkPaidOptions{})
}
func (s *PaymentService) checkPaidWithOptions(ctx context.Context, o *dbent.PaymentOrder, opts checkPaidOptions) string {
prov, err := s.getOrderProvider(ctx, o)
if err != nil {
return ""
@@ -182,6 +196,9 @@ func (s *PaymentService) checkPaid(ctx context.Context, o *dbent.PaymentOrder) s
}
return checkPaidResultAlreadyPaid
}
if !opts.cancelIfUnpaid {
return ""
}
if cp, ok := prov.(payment.CancelableProvider); ok {
_ = cp.CancelPayment(ctx, queryRef)
}
@@ -268,7 +285,7 @@ func (s *PaymentService) VerifyOrderByOutTradeNo(ctx context.Context, outTradeNo
}
// Only verify orders that are still pending or recently expired
if o.Status == OrderStatusPending || o.Status == OrderStatusExpired {
result := s.checkPaid(ctx, o)
result := s.reconcilePaid(ctx, o)
if result == checkPaidResultAlreadyPaid {
// Reload order to get updated status
o, err = s.entClient.PaymentOrder.Get(ctx, o.ID)
@@ -280,6 +297,37 @@ func (s *PaymentService) VerifyOrderByOutTradeNo(ctx context.Context, outTradeNo
return o, nil
}
// ReconcilePendingWxpayOrders actively checks recent pending WeChat orders so
// missed provider notifications do not wait until order expiry to fulfill.
func (s *PaymentService) ReconcilePendingWxpayOrders(ctx context.Context) (int, error) {
now := time.Now()
orders, err := s.entClient.PaymentOrder.Query().
Where(
paymentorder.StatusEQ(OrderStatusPending),
paymentorder.ExpiresAtGT(now),
paymentorder.Or(
paymentorder.PaymentTypeEQ(payment.TypeWxpay),
paymentorder.PaymentTypeHasPrefix(payment.TypeWxpay+"_"),
paymentorder.ProviderKeyEQ(payment.TypeWxpay),
paymentorder.ProviderKeyHasPrefix(payment.TypeWxpay+"_"),
),
).
Order(dbent.Asc(paymentorder.FieldCreatedAt)).
Limit(pendingWxpayReconcileLimit).
All(ctx)
if err != nil {
return 0, fmt.Errorf("query pending wxpay orders: %w", err)
}
recovered := 0
for _, order := range orders {
if s.reconcilePaid(ctx, order) == checkPaidResultAlreadyPaid {
recovered++
}
}
return recovered, nil
}
// VerifyOrderPublic returns the currently persisted public order state without
// triggering any upstream reconciliation. Signed resume-token recovery is the
// only public recovery path allowed to query upstream state.
@@ -20,10 +20,13 @@ import (
)
type paymentOrderLifecycleQueryProvider struct {
lastQueryTradeNo string
queryCalls int
responses []*payment.QueryOrderResponse
resp *payment.QueryOrderResponse
key string
lastQueryTradeNo string
lastCancelTradeNo string
queryCalls int
cancelCalls int
responses []*payment.QueryOrderResponse
resp *payment.QueryOrderResponse
}
type paymentOrderLifecycleRedeemRepo struct {
@@ -38,10 +41,15 @@ func (p *paymentOrderLifecycleQueryProvider) Name() string {
return "payment-order-lifecycle-query-provider"
}
func (p *paymentOrderLifecycleQueryProvider) ProviderKey() string { return payment.TypeAlipay }
func (p *paymentOrderLifecycleQueryProvider) ProviderKey() string {
if p.key != "" {
return p.key
}
return payment.TypeAlipay
}
func (p *paymentOrderLifecycleQueryProvider) SupportedTypes() []payment.PaymentType {
return []payment.PaymentType{payment.TypeAlipay}
return []payment.PaymentType{p.ProviderKey()}
}
func (p *paymentOrderLifecycleQueryProvider) CreatePayment(context.Context, payment.CreatePaymentRequest) (*payment.CreatePaymentResponse, error) {
@@ -69,6 +77,12 @@ func (p *paymentOrderLifecycleQueryProvider) Refund(context.Context, payment.Ref
panic("unexpected call")
}
func (p *paymentOrderLifecycleQueryProvider) CancelPayment(_ context.Context, tradeNo string) error {
p.lastCancelTradeNo = tradeNo
p.cancelCalls++
return nil
}
func (r *paymentOrderLifecycleRedeemRepo) Create(context.Context, *RedeemCode) error {
panic("unexpected call")
}
@@ -435,6 +449,222 @@ func TestVerifyOrderByOutTradeNoRejectsPaidQueryWithZeroAmount(t *testing.T) {
require.Empty(t, redeemRepo.useCalls)
}
func TestVerifyOrderByOutTradeNoDoesNotCancelUnpaidUpstreamOrder(t *testing.T) {
ctx := context.Background()
client := newPaymentOrderLifecycleTestClient(t)
user, err := client.User.Create().
SetEmail("checkpaid-pending@example.com").
SetPasswordHash("hash").
SetUsername("checkpaid-pending-user").
Save(ctx)
require.NoError(t, err)
order, err := client.PaymentOrder.Create().
SetUserID(user.ID).
SetUserEmail(user.Email).
SetUserName(user.Username).
SetAmount(88).
SetPayAmount(88).
SetFeeRate(0).
SetRechargeCode("CHECKPAID-PENDING").
SetOutTradeNo("sub2_checkpaid_pending").
SetPaymentType(payment.TypeAlipay).
SetPaymentTradeNo("").
SetOrderType(payment.OrderTypeBalance).
SetStatus(OrderStatusPending).
SetExpiresAt(time.Now().Add(time.Hour)).
SetClientIP("127.0.0.1").
SetSrcHost("api.example.com").
Save(ctx)
require.NoError(t, err)
registry := payment.NewRegistry()
provider := &paymentOrderLifecycleQueryProvider{
resp: &payment.QueryOrderResponse{
TradeNo: order.OutTradeNo,
Status: payment.ProviderStatusPending,
Amount: 0,
},
}
registry.Register(provider)
svc := &PaymentService{
entClient: client,
registry: registry,
providersLoaded: true,
}
got, err := svc.VerifyOrderByOutTradeNo(ctx, order.OutTradeNo, user.ID)
require.NoError(t, err)
require.Equal(t, OrderStatusPending, got.Status)
require.Equal(t, order.OutTradeNo, provider.lastQueryTradeNo)
require.Zero(t, provider.cancelCalls)
reloaded, err := client.PaymentOrder.Get(ctx, order.ID)
require.NoError(t, err)
require.Equal(t, OrderStatusPending, reloaded.Status)
}
func TestCancelOrderStillClosesUnpaidUpstreamOrder(t *testing.T) {
ctx := context.Background()
client := newPaymentOrderLifecycleTestClient(t)
user, err := client.User.Create().
SetEmail("cancel-pending@example.com").
SetPasswordHash("hash").
SetUsername("cancel-pending-user").
Save(ctx)
require.NoError(t, err)
order, err := client.PaymentOrder.Create().
SetUserID(user.ID).
SetUserEmail(user.Email).
SetUserName(user.Username).
SetAmount(88).
SetPayAmount(88).
SetFeeRate(0).
SetRechargeCode("CANCEL-PENDING").
SetOutTradeNo("sub2_cancel_pending").
SetPaymentType(payment.TypeAlipay).
SetPaymentTradeNo("").
SetOrderType(payment.OrderTypeBalance).
SetStatus(OrderStatusPending).
SetExpiresAt(time.Now().Add(time.Hour)).
SetClientIP("127.0.0.1").
SetSrcHost("api.example.com").
Save(ctx)
require.NoError(t, err)
registry := payment.NewRegistry()
provider := &paymentOrderLifecycleQueryProvider{
resp: &payment.QueryOrderResponse{
TradeNo: order.OutTradeNo,
Status: payment.ProviderStatusPending,
Amount: 0,
},
}
registry.Register(provider)
svc := &PaymentService{
entClient: client,
registry: registry,
providersLoaded: true,
}
outcome, err := svc.CancelOrder(ctx, order.ID, user.ID)
require.NoError(t, err)
require.Equal(t, checkPaidResultCancelled, outcome)
require.Equal(t, order.OutTradeNo, provider.lastCancelTradeNo)
require.Equal(t, 1, provider.cancelCalls)
reloaded, err := client.PaymentOrder.Get(ctx, order.ID)
require.NoError(t, err)
require.Equal(t, OrderStatusCancelled, reloaded.Status)
}
func TestReconcilePendingWxpayOrdersBackfillsPaidOrder(t *testing.T) {
ctx := context.Background()
client := newPaymentOrderLifecycleTestClient(t)
user, err := client.User.Create().
SetEmail("wxpay-reconcile@example.com").
SetPasswordHash("hash").
SetUsername("wxpay-reconcile-user").
Save(ctx)
require.NoError(t, err)
order, err := client.PaymentOrder.Create().
SetUserID(user.ID).
SetUserEmail(user.Email).
SetUserName(user.Username).
SetAmount(50).
SetPayAmount(50).
SetFeeRate(0).
SetRechargeCode("WXPAY-RECONCILE").
SetOutTradeNo("sub2_wxpay_reconcile").
SetPaymentType(payment.TypeWxpay).
SetPaymentTradeNo("").
SetOrderType(payment.OrderTypeBalance).
SetStatus(OrderStatusPending).
SetExpiresAt(time.Now().Add(time.Hour)).
SetClientIP("127.0.0.1").
SetSrcHost("api.example.com").
Save(ctx)
require.NoError(t, err)
userRepo := &mockUserRepo{
getByIDUser: &User{
ID: user.ID,
Email: user.Email,
Username: user.Username,
Balance: 0,
},
}
userRepo.updateBalanceFn = func(ctx context.Context, id int64, amount float64) error {
require.Equal(t, user.ID, id)
if userRepo.getByIDUser != nil {
userRepo.getByIDUser.Balance += amount
}
return nil
}
redeemRepo := &paymentOrderLifecycleRedeemRepo{
codesByCode: map[string]*RedeemCode{
order.RechargeCode: {
ID: 1,
Code: order.RechargeCode,
Type: RedeemTypeBalance,
Value: order.Amount,
Status: StatusUnused,
},
},
}
redeemService := NewRedeemService(
redeemRepo,
userRepo,
nil,
nil,
nil,
client,
nil,
nil,
)
registry := payment.NewRegistry()
provider := &paymentOrderLifecycleQueryProvider{
key: payment.TypeWxpay,
resp: &payment.QueryOrderResponse{
TradeNo: "wxpay-upstream-trade-123",
Status: payment.ProviderStatusPaid,
Amount: 50,
Metadata: map[string]string{
"trade_state": "SUCCESS",
},
},
}
registry.Register(provider)
svc := &PaymentService{
entClient: client,
registry: registry,
redeemService: redeemService,
userRepo: userRepo,
providersLoaded: true,
}
recovered, err := svc.ReconcilePendingWxpayOrders(ctx)
require.NoError(t, err)
require.Equal(t, 1, recovered)
require.Equal(t, order.OutTradeNo, provider.lastQueryTradeNo)
require.Zero(t, provider.cancelCalls)
reloaded, err := client.PaymentOrder.Get(ctx, order.ID)
require.NoError(t, err)
require.Equal(t, OrderStatusCompleted, reloaded.Status)
require.Equal(t, "wxpay-upstream-trade-123", reloaded.PaymentTradeNo)
require.Equal(t, 50.0, userRepo.getByIDUser.Balance)
require.Len(t, redeemRepo.useCalls, 1)
}
func TestVerifyOrderByOutTradeNoUsesOutTradeNoWhenPaymentTradeNoAlreadyExistsForAlipay(t *testing.T) {
ctx := context.Background()
client := newPaymentOrderLifecycleTestClient(t)
@@ -46,7 +46,7 @@ func (s *PaymentService) GetPublicOrderByResumeToken(ctx context.Context, token
return nil, invalidResumeTokenMatchError()
}
if order.Status == OrderStatusPending || order.Status == OrderStatusExpired {
result := s.checkPaid(ctx, order)
result := s.reconcilePaid(ctx, order)
if result == checkPaidResultAlreadyPaid {
order, err = s.entClient.PaymentOrder.Get(ctx, order.ID)
if err != nil {