mirror of
https://github.com/bufanyun/hotgo.git
synced 2025-01-24 03:29:05 +08:00
120 lines
3.0 KiB
Go
120 lines
3.0 KiB
Go
package payment
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/os/gtime"
|
||
"github.com/gogf/gf/v2/util/grand"
|
||
"hotgo/internal/consts"
|
||
"hotgo/internal/library/payment/alipay"
|
||
"hotgo/internal/library/payment/qqpay"
|
||
"hotgo/internal/library/payment/wxpay"
|
||
"hotgo/internal/model/input/payin"
|
||
"hotgo/utility/validate"
|
||
)
|
||
|
||
// PayClient 支付客户端
|
||
type PayClient interface {
|
||
// CreateOrder 创建订单
|
||
CreateOrder(ctx context.Context, in payin.CreateOrderInp) (res *payin.CreateOrderModel, err error)
|
||
// Notify 异步通知
|
||
Notify(ctx context.Context, in payin.NotifyInp) (res *payin.NotifyModel, err error)
|
||
// Refund 订单退款
|
||
Refund(ctx context.Context, in payin.RefundInp) (res *payin.RefundModel, err error)
|
||
}
|
||
|
||
func New(name ...string) PayClient {
|
||
var (
|
||
payType = consts.PayTypeWxPay
|
||
client PayClient
|
||
)
|
||
|
||
if len(name) > 0 && name[0] != "" {
|
||
payType = name[0]
|
||
}
|
||
|
||
switch payType {
|
||
case consts.PayTypeAliPay:
|
||
client = alipay.New(config)
|
||
case consts.PayTypeWxPay:
|
||
client = wxpay.New(config)
|
||
case consts.PayTypeQQPay:
|
||
client = qqpay.New(config)
|
||
default:
|
||
panic(fmt.Sprintf("暂不支持的支付方式:%v", payType))
|
||
}
|
||
return client
|
||
}
|
||
|
||
// GenOrderSn 生成业务订单号
|
||
func GenOrderSn() string {
|
||
orderSn := fmt.Sprintf("HG@%v%v", gtime.Now().Format("YmdHis"), grand.S(4))
|
||
count, err := g.Model("pay_log").Where("order_sn", orderSn).Count()
|
||
if err != nil {
|
||
panic(fmt.Sprintf("payment.GenOrderSn err:%+v", err))
|
||
}
|
||
if count > 0 {
|
||
return GenOrderSn()
|
||
}
|
||
return orderSn
|
||
}
|
||
|
||
// GenOutTradeNo 生成商户订单号
|
||
func GenOutTradeNo() string {
|
||
outTradeNo := fmt.Sprintf("%v%v", gtime.Now().Format("YmdHis"), grand.N(10000000, 99999999))
|
||
count, err := g.Model("pay_log").Where("out_trade_no", outTradeNo).Count()
|
||
if err != nil {
|
||
panic(fmt.Sprintf("payment.GenOutTradeNo err:%+v", err))
|
||
}
|
||
if count > 0 {
|
||
return GenOutTradeNo()
|
||
}
|
||
return outTradeNo
|
||
}
|
||
|
||
// GenRefundSn 生成退款订单号
|
||
func GenRefundSn() string {
|
||
outTradeNo := fmt.Sprintf("%v%v", gtime.Now().Format("YmdHis"), grand.N(10000, 99999))
|
||
count, err := g.Model("pay_refund").Where("refund_trade_no", outTradeNo).Count()
|
||
if err != nil {
|
||
panic(fmt.Sprintf("payment.GenRefundSn err:%+v", err))
|
||
}
|
||
if count > 0 {
|
||
return GenRefundSn()
|
||
}
|
||
return outTradeNo
|
||
}
|
||
|
||
// AutoTradeType 根据userAgent自动识别交易方式，在实际支付场景中你可以手动调整识别规则
|
||
func AutoTradeType(payType, userAgent string) (tradeType string) {
|
||
isMobile := validate.IsMobileVisit(userAgent)
|
||
switch payType {
|
||
case consts.PayTypeAliPay:
|
||
if isMobile {
|
||
return consts.TradeTypeAliWap
|
||
}
|
||
return consts.TradeTypeAliWeb
|
||
case consts.PayTypeWxPay:
|
||
if isMobile {
|
||
if validate.IsWxBrowserVisit(userAgent) {
|
||
return consts.TradeTypeWxMP
|
||
}
|
||
|
||
if validate.IsWxMiniProgramVisit(userAgent) {
|
||
return consts.TradeTypeWxMini
|
||
}
|
||
|
||
return consts.TradeTypeWxH5
|
||
}
|
||
return consts.TradeTypeWxScan
|
||
case consts.PayTypeQQPay:
|
||
if isMobile {
|
||
return consts.TradeTypeQQWap
|
||
}
|
||
return consts.TradeTypeQQWeb
|
||
default:
|
||
}
|
||
return
|
||
}
|