101 lines
2.8 KiB
Go
101 lines
2.8 KiB
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 == "pending" && order.PaymentScene == "wxmini" &&
|
||
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
|
||
}
|
||
sessionKey, err := l.svcCtx.XpayService.GetSessionKey(l.ctx, userID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
status, err := l.svcCtx.XpayService.QueryOrder(l.ctx, openid, order.OrderNo, sessionKey)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
if service.IsXpayPaidStatus(status.Status) {
|
||
credited, fulfillErr := fulfillQueryOrderPaid(l.ctx, l.svcCtx, order, "", status.PaidFee)
|
||
if fulfillErr != nil {
|
||
return fulfillErr
|
||
}
|
||
if credited {
|
||
wxOrderID := ""
|
||
_ = l.svcCtx.XpayService.NotifyProvideGoods(l.ctx, openid, order.OrderNo, wxOrderID, sessionKey)
|
||
_ = l.svcCtx.XpayService.MarkNotified(l.ctx, order.OrderNo)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
if service.IsXpayClosedStatus(status.Status) {
|
||
order.Status = "closed"
|
||
return l.svcCtx.OrderModel.UpdateWithVersion(l.ctx, nil, order)
|
||
}
|
||
|
||
return nil
|
||
}
|