chore: 并入 iotlcm-worker-svr 历史到 worker-svr/（原提交 96f0f4e）

worker-svr/internal/service/balance/alipay.go

@@ -0,0 +1,137 @@
+package balance
+
+import (
+	"context"
+	"errors"
+	"net/http"
+	"strings"
+	"sync"
+	"time"
+
+	"github.com/gogf/gf/v2/errors/gerror"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/smartwalle/alipay/v3"
+)
+
+type alipayConfig struct {
+	AppID           string `yaml:"appId"`
+	PrivateKey      string `yaml:"privateKey"`
+	AlipayPublicKey string `yaml:"alipayPublicKey"`
+	IsProduction    bool   `yaml:"isProduction"`
+}
+
+type alipayTradeQueryer struct {
+	once      sync.Once
+	client    *alipay.Client
+	clientErr error
+}
+
+func newAlipayTradeQueryer() *alipayTradeQueryer {
+	return &alipayTradeQueryer{}
+}
+
+func (q *alipayTradeQueryer) Query(ctx context.Context, outTradeNo string) (*TradeQueryResult, error) {
+	client, err := q.getClient(ctx)
+	if err != nil {
+		return nil, err
+	}
+	rsp, err := q.tradeQueryWithRetry(ctx, client, alipay.TradeQuery{OutTradeNo: outTradeNo})
+	if err != nil {
+		if isAlipayTradeNotFound(err.Error()) {
+			return &TradeQueryResult{OutTradeNo: outTradeNo, TradeStatus: string(alipay.TradeStatusWaitBuyerPay)}, nil
+		}
+		return nil, gerror.Wrap(err, "query Alipay trade failed")
+	}
+	if rsp == nil {
+		return nil, gerror.New("Alipay trade query returned no result")
+	}
+	if rsp.IsFailure() {
+		if isAlipayTradeNotFound(string(rsp.Code), rsp.SubCode, rsp.SubMsg) {
+			return &TradeQueryResult{OutTradeNo: outTradeNo, TradeStatus: string(alipay.TradeStatusWaitBuyerPay)}, nil
+		}
+		return nil, gerror.Newf("Alipay trade query failed: code=%s, subCode=%s, subMsg=%s", rsp.Code, rsp.SubCode, rsp.SubMsg)
+	}
+	return &TradeQueryResult{
+		OutTradeNo:   rsp.OutTradeNo,
+		TradeNo:      rsp.TradeNo,
+		TradeStatus:  string(rsp.TradeStatus),
+		TotalAmount:  rsp.TotalAmount,
+		BuyerLogonID: rsp.BuyerLogonId,
+	}, nil
+}
+
+func (q *alipayTradeQueryer) getClient(ctx context.Context) (*alipay.Client, error) {
+	q.once.Do(func() {
+		cfg := alipayConfig{}
+		if err := g.Cfg().MustGet(ctx, "alipay").Scan(&cfg); err != nil {
+			q.clientErr = gerror.Wrap(err, "read Alipay config failed")
+			return
+		}
+		if strings.TrimSpace(cfg.AppID) == "" || strings.TrimSpace(cfg.PrivateKey) == "" || strings.TrimSpace(cfg.AlipayPublicKey) == "" {
+			q.clientErr = gerror.New("Alipay credentials are not configured")
+			return
+		}
+		q.client, q.clientErr = alipay.New(cfg.AppID, cfg.PrivateKey, cfg.IsProduction, alipay.WithHTTPClient(&http.Client{Timeout: 10 * time.Second}))
+		if q.clientErr != nil {
+			q.clientErr = gerror.Wrap(q.clientErr, "initialize Alipay client failed")
+			return
+		}
+		if err := q.client.LoadAliPayPublicKey(cfg.AlipayPublicKey); err != nil {
+			q.clientErr = gerror.Wrap(err, "load Alipay public key failed")
+		}
+	})
+	if q.clientErr != nil {
+		return nil, q.clientErr
+	}
+	return q.client, nil
+}
+
+func (q *alipayTradeQueryer) tradeQueryWithRetry(ctx context.Context, client *alipay.Client, query alipay.TradeQuery) (*alipay.TradeQueryRsp, error) {
+	var err error
+	for attempt := 1; attempt <= 3; attempt++ {
+		rsp, queryErr := client.TradeQuery(ctx, query)
+		if queryErr == nil {
+			return rsp, nil
+		}
+		err = queryErr
+		if !isRetryableAlipayError(err) || attempt == 3 {
+			return nil, err
+		}
+		if err = sleepWithContext(ctx, time.Duration(attempt)*300*time.Millisecond); err != nil {
+			return nil, err
+		}
+	}
+	return nil, err
+}
+
+func isRetryableAlipayError(err error) bool {
+	if err == nil {
+		return false
+	}
+	if errors.Is(err, context.DeadlineExceeded) {
+		return true
+	}
+	msg := strings.ToLower(err.Error())
+	for _, marker := range []string{"invalid character '<'", "unexpected end of json", "bad response", "eof", "timeout", "connection reset", "connection refused", "tls handshake", "temporary", "too many requests", "429", "500", "502", "503", "504"} {
+		if strings.Contains(msg, marker) {
+			return true
+		}
+	}
+	return false
+}
+
+func isAlipayTradeNotFound(parts ...string) bool {
+	msg := strings.ToLower(strings.Join(parts, " "))
+	return strings.Contains(msg, "acq.trade_not_exist") || strings.Contains(msg, "trade_not_exist") || strings.Contains(msg, "transaction not exist") || strings.Contains(msg, "交易不存在") || strings.Contains(msg, "订单不存在")
+}
+
+func sleepWithContext(ctx context.Context, delay time.Duration) error {
+	timer := time.NewTimer(delay)
+	defer timer.Stop()
+	select {
+	case <-ctx.Done():
+		return ctx.Err()
+	case <-timer.C:
+		return nil
+	}
+}