qnc-server-v3/app/main/api/internal/logic/pay/paymentchecklogic.go

package pay

import (
	"context"
	"strings"

	"qnc-server/app/main/api/internal/service"
	"qnc-server/app/main/api/internal/svc"
	"qnc-server/app/main/api/internal/types"
	"qnc-server/app/main/model"
	"qnc-server/common/ctxdata"
	"qnc-server/common/xerr"
	"github.com/pkg/errors"
	"github.com/zeromicro/go-zero/core/logx"
)

type PaymentCheckLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewPaymentCheckLogic(ctx context.Context, svcCtx *svc.ServiceContext) *PaymentCheckLogic {
	return &PaymentCheckLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *PaymentCheckLogic) PaymentCheck(req *types.PaymentCheckReq) (resp *types.PaymentCheckResp, err error) {
	if strings.HasPrefix(req.OrderNo, "U_") {
		order, findErr := l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, req.OrderNo)
		if findErr != nil {
			return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询升级订单失败: %v", findErr)
		}
		return &types.PaymentCheckResp{Type: "agent_upgrade", Status: order.Status}, nil
	}

	order, findErr := l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, req.OrderNo)
	if findErr != nil {
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询订单失败: %v", findErr)
	}

	// xpay 轮询：pending 时主动查微信单并到账（不通知微信发货）
	if order.Status == model.OrderStatusPending && model.IsXpayOrder(order) &&
		l.svcCtx.XpayService != nil && l.svcCtx.XpayService.Enabled() {
		if syncErr := l.syncXpayOrderStatus(order); syncErr != nil {
			l.Errorf("[xpay] 轮询查单失败 order_no=%s err=%v", req.OrderNo, syncErr)
		} else {
			order, _ = l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, req.OrderNo)
		}
	}

	return &types.PaymentCheckResp{Type: "query", Status: order.Status}, nil
}

func (l *PaymentCheckLogic) syncXpayOrderStatus(order *model.Order) error {
	userID, err := ctxdata.GetUidFromCtx(l.ctx)
	if err != nil {
		return err
	}
	if order.UserId != userID {
		return errors.New("无权查询此订单")
	}

	openid, err := l.svcCtx.XpayService.GetWxMiniOpenID(l.ctx, l.svcCtx.UserAuthModel, userID)
	if err != nil {
		return err
	}
	status, err := l.svcCtx.XpayService.QueryOrder(l.ctx, openid, order.OrderNo, "")
	if err != nil {
		return err
	}

	if service.IsXpayPaidStatus(status.Status) {
		_, fulfillErr := fulfillQueryOrderPaid(l.ctx, l.svcCtx, order, status.WxOrderID, status.PaidFee)
		return fulfillErr
	}

	if service.IsXpayClosedStatus(status.Status) {
		order.Status = "closed"
		return l.svcCtx.OrderModel.UpdateWithVersion(l.ctx, nil, order)
	}

	return nil
}