重构交易相关逻辑,优化异步任务处理,更新支付状态查询逻辑
This commit is contained in:
45
README.md
45
README.md
@@ -1,5 +1,13 @@
|
|||||||
## TODO
|
## TODO
|
||||||
|
|
||||||
|
创建交易订单后添加一个关闭订单的异步任务
|
||||||
|
|
||||||
|
支付回调需要判断可能重复调用的情况
|
||||||
|
|
||||||
|
实现订单状态查询的 SSE 接口
|
||||||
|
|
||||||
|
考虑将重复量比较大的异步任务修改成定时调度任务
|
||||||
|
|
||||||
### 长期
|
### 长期
|
||||||
|
|
||||||
模型字段修改,特定枚举字段使用自定义类型代替通用 int32
|
模型字段修改,特定枚举字段使用自定义类型代替通用 int32
|
||||||
@@ -8,28 +16,16 @@
|
|||||||
|
|
||||||
## 业务逻辑
|
## 业务逻辑
|
||||||
|
|
||||||
### 代理服务与节点的增量更新
|
### 支付处理流程
|
||||||
|
|
||||||
代理服务定时提交增量更新数据包,格式为:
|
1. 创建订单,推送异步检查任务
|
||||||
|
2. sse 接口推送订单状态
|
||||||
|
|
||||||
| version(1) | proxy_id(4) | count(2) | edge_id(4) | mask(1) | info(...) |
|
- 支付回调更新支付状态
|
||||||
|------------|-------------|----------|------------|---------|-----------|
|
- 异步任务更新支付状态
|
||||||
| 数据包版本 | 代理服务 id | 更新的节点数量 | 节点 id | 节点更新项 | 节点更新内容 |
|
- 主动查询更新支付状态
|
||||||
|
|
||||||
其中 mask 部分,每个位代表一个节点更新项,具体如下:
|
更新支付状态后,缓存结果以便查询
|
||||||
|
|
||||||
| 位数(从前到后) | 节点更新项 |
|
|
||||||
|----------|----------|
|
|
||||||
| 0 | 保留位,预留扩展 |
|
|
||||||
| 1 | 空,固定为 0 |
|
|
||||||
| 2 | 空,固定为 0 |
|
|
||||||
| 3 | 端口 |
|
|
||||||
| 4 | 省份 |
|
|
||||||
| 5 | 城市 |
|
|
||||||
| 6 | 运营商 |
|
|
||||||
| 7 | 状态 |
|
|
||||||
|
|
||||||
节点更新的具体内容大小视数据结构而定
|
|
||||||
|
|
||||||
### 产品字典表
|
### 产品字典表
|
||||||
|
|
||||||
@@ -37,14 +33,3 @@
|
|||||||
|-------|------|
|
|-------|------|
|
||||||
| short | 短效代理 |
|
| short | 短效代理 |
|
||||||
| long | 长效代理 |
|
| long | 长效代理 |
|
||||||
|
|
||||||
## 业务流程
|
|
||||||
|
|
||||||
### 短效动态
|
|
||||||
|
|
||||||
1. 获取远端连接配置
|
|
||||||
2. 查询并计算实际连接状况
|
|
||||||
3. 每个网关:
|
|
||||||
1. 根据负载分配算法,计算是否需要为此网关分配新的连接
|
|
||||||
2. 如果需要新的连接且预计连接数大于已配置的连接数,则更新远端连接配置
|
|
||||||
3. 为新增连接分配可用端口,并更新网关端口配置
|
|
||||||
@@ -3,6 +3,7 @@ package handlers
|
|||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/shopspring/decimal"
|
"github.com/shopspring/decimal"
|
||||||
|
"github.com/valyala/fasthttp/fasthttpadaptor"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
"net/http"
|
"net/http"
|
||||||
trade2 "platform/web/domains/trade"
|
trade2 "platform/web/domains/trade"
|
||||||
@@ -13,13 +14,25 @@ import (
|
|||||||
|
|
||||||
"github.com/gofiber/fiber/v2"
|
"github.com/gofiber/fiber/v2"
|
||||||
"github.com/smartwalle/alipay/v3"
|
"github.com/smartwalle/alipay/v3"
|
||||||
"github.com/valyala/fasthttp/fasthttpadaptor"
|
|
||||||
"github.com/wechatpay-apiv3/wechatpay-go/services/payments"
|
"github.com/wechatpay-apiv3/wechatpay-go/services/payments"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// region TradeCheckSSE
|
||||||
|
|
||||||
|
func TradeCheckSSE(c *fiber.Ctx) error {
|
||||||
|
// 设置响应头
|
||||||
|
c.Set("Content-Type", "text/event-stream")
|
||||||
|
c.Set("Cache-Control", "no-cache")
|
||||||
|
c.Set("Connection", "keep-alive")
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// endregion
|
||||||
|
|
||||||
// region AlipayCallback
|
// region AlipayCallback
|
||||||
|
|
||||||
func AlipayCallback(c *fiber.Ctx) error {
|
func AlipayCallback(c *fiber.Ctx) (err error) {
|
||||||
|
|
||||||
// 解析请求
|
// 解析请求
|
||||||
httpRequest := new(http.Request)
|
httpRequest := new(http.Request)
|
||||||
@@ -36,21 +49,34 @@ func AlipayCallback(c *fiber.Ctx) error {
|
|||||||
|
|
||||||
slog.Debug("支付宝支付回调", "notification", fmt.Sprintf("%+v", notification))
|
slog.Debug("支付宝支付回调", "notification", fmt.Sprintf("%+v", notification))
|
||||||
|
|
||||||
// todo 退款通知
|
|
||||||
if isRefund(notification) {
|
|
||||||
return act(c)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 查询交易信息
|
// 查询交易信息
|
||||||
trade, err := q.Q.Trade.Where(q.Trade.InnerNo.Eq(notification.OutTradeNo)).Take()
|
trade, err := q.Q.Trade.Where(q.Trade.InnerNo.Eq(notification.OutTradeNo)).Take()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
// 跳过测试通知
|
return c.SendString("success")
|
||||||
return act(c)
|
|
||||||
}
|
}
|
||||||
switch notification.NotifyType {
|
switch alipay.TradeStatus(notification.NotifyType) {
|
||||||
|
|
||||||
|
// 等待支付
|
||||||
|
case alipay.TradeStatusWaitBuyerPay:
|
||||||
|
// 不需要处理
|
||||||
|
|
||||||
|
// 支付关闭
|
||||||
|
case alipay.TradeStatusClosed:
|
||||||
|
switch trade2.Type(trade.Type) {
|
||||||
|
|
||||||
|
// 购买产品
|
||||||
|
case trade2.TypePurchase:
|
||||||
|
err = s.Resource.CancelResource(notification.OutTradeNo, time.Now(), true)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 余额充值
|
||||||
|
case trade2.TypeRecharge:
|
||||||
|
}
|
||||||
|
|
||||||
// 支付成功
|
// 支付成功
|
||||||
case string(alipay.TradeStatusSuccess):
|
case alipay.TradeStatusSuccess:
|
||||||
|
|
||||||
// 收集交易状态
|
// 收集交易状态
|
||||||
payment, err := decimal.NewFromString(notification.TotalAmount)
|
payment, err := decimal.NewFromString(notification.TotalAmount)
|
||||||
@@ -61,7 +87,7 @@ func AlipayCallback(c *fiber.Ctx) error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
verified := &s.TransactionVerifyResult{
|
verified := &s.TradeSuccessResult{
|
||||||
TransId: notification.TradeNo,
|
TransId: notification.TradeNo,
|
||||||
Payment: payment,
|
Payment: payment,
|
||||||
Time: paidAt,
|
Time: paidAt,
|
||||||
@@ -83,48 +109,18 @@ func AlipayCallback(c *fiber.Ctx) error {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// 支付关闭
|
// 交易结束
|
||||||
case string(alipay.TradeStatusClosed):
|
case alipay.TradeStatusFinished:
|
||||||
switch trade2.Type(trade.Type) {
|
// 结束交易状态
|
||||||
|
|
||||||
// 购买产品
|
|
||||||
case trade2.TypePurchase:
|
|
||||||
err = s.Resource.CancelResource(notification.OutTradeNo, time.Now(), true)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return act(c)
|
return c.SendString("success")
|
||||||
}
|
|
||||||
|
|
||||||
type AdapterWriter struct {
|
|
||||||
c *fiber.Ctx
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a AdapterWriter) Header() http.Header {
|
|
||||||
panic("implement me")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a AdapterWriter) Write(bytes []byte) (int, error) {
|
|
||||||
return a.c.Write(bytes)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a AdapterWriter) WriteHeader(statusCode int) {
|
|
||||||
a.c.Status(statusCode)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func isRefund(notification *alipay.Notification) bool {
|
func isRefund(notification *alipay.Notification) bool {
|
||||||
return notification.OutBizNo != "" || notification.RefundFee != "" || notification.GmtRefund != ""
|
return notification.OutBizNo != "" || notification.RefundFee != "" || notification.GmtRefund != ""
|
||||||
}
|
}
|
||||||
|
|
||||||
func act(c *fiber.Ctx) error {
|
|
||||||
g.Alipay.ACKNotification(AdapterWriter{c: c})
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// endregion
|
// endregion
|
||||||
|
|
||||||
// region WechatPayCallback
|
// region WechatPayCallback
|
||||||
@@ -162,7 +158,7 @@ func WechatPayCallback(c *fiber.Ctx) error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
verified := &s.TransactionVerifyResult{
|
verified := &s.TradeSuccessResult{
|
||||||
TransId: *content.TransactionId,
|
TransId: *content.TransactionId,
|
||||||
Payment: payment,
|
Payment: payment,
|
||||||
Time: paidAt,
|
Time: paidAt,
|
||||||
|
|||||||
@@ -179,9 +179,9 @@ func RechargePrepareAlipay(c *fiber.Ctx) error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
var result *s.TransactionPrepareResult
|
var result *s.TradeCreateResult
|
||||||
err = q.Q.Transaction(func(tx *q.Query) error {
|
err = q.Q.Transaction(func(tx *q.Query) error {
|
||||||
result, err = s.Transaction.PrepareTransaction(tx, authContext.Payload.Id, now, &s.TransactionPrepareData{
|
result, err = s.Trade.SendCreateTradeByQrcode(tx, authContext.Payload.Id, now, &s.TradeCreateData{
|
||||||
Subject: "账户充值 - " + amount.StringFixed(2) + "元",
|
Subject: "账户充值 - " + amount.StringFixed(2) + "元",
|
||||||
Amount: amount,
|
Amount: amount,
|
||||||
ExpireAt: time.Now().Add(30 * time.Minute),
|
ExpireAt: time.Now().Add(30 * time.Minute),
|
||||||
@@ -215,7 +215,7 @@ func RechargeConfirmAlipay(c *fiber.Ctx) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 验证支付结果
|
// 验证支付结果
|
||||||
result, err := s.Transaction.VerifyTransaction(&s.TransactionVerifyData{
|
result, err := s.Trade.VerifyTrade(&s.TradeVerifyData{
|
||||||
TradeNo: req.TradeNo,
|
TradeNo: req.TradeNo,
|
||||||
Method: trade2.MethodAlipay,
|
Method: trade2.MethodAlipay,
|
||||||
})
|
})
|
||||||
@@ -251,9 +251,9 @@ func RechargePrepareWechat(c *fiber.Ctx) error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
var result *s.TransactionPrepareResult
|
var result *s.TradeCreateResult
|
||||||
err = q.Q.Transaction(func(tx *q.Query) error {
|
err = q.Q.Transaction(func(tx *q.Query) error {
|
||||||
result, err = s.Transaction.PrepareTransaction(tx, authContext.Payload.Id, now, &s.TransactionPrepareData{
|
result, err = s.Trade.SendCreateTradeByQrcode(tx, authContext.Payload.Id, now, &s.TradeCreateData{
|
||||||
Subject: "账户充值 - " + amount.StringFixed(2) + "元",
|
Subject: "账户充值 - " + amount.StringFixed(2) + "元",
|
||||||
Amount: amount,
|
Amount: amount,
|
||||||
ExpireAt: now.Add(30 * time.Minute),
|
ExpireAt: now.Add(30 * time.Minute),
|
||||||
@@ -289,7 +289,7 @@ func RechargeConfirmWechat(c *fiber.Ctx) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 验证支付结果
|
// 验证支付结果
|
||||||
result, err := s.Transaction.VerifyTransaction(&s.TransactionVerifyData{
|
result, err := s.Trade.VerifyTrade(&s.TradeVerifyData{
|
||||||
TradeNo: req.TradeNo,
|
TradeNo: req.TradeNo,
|
||||||
Method: trade2.MethodWeChat,
|
Method: trade2.MethodWeChat,
|
||||||
})
|
})
|
||||||
|
|||||||
@@ -17,7 +17,7 @@ type VerifierReq struct {
|
|||||||
|
|
||||||
func SmsCode(c *fiber.Ctx) error {
|
func SmsCode(c *fiber.Ctx) error {
|
||||||
|
|
||||||
_, err := auth.Protect(c, []auth.PayloadType{auth.PayloadSecuredServer}, []string{})
|
_, err := auth.Protect(c, []auth.PayloadType{auth.PayloadInternalServer}, []string{})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -56,6 +56,8 @@ func ApplyRouters(app *fiber.App) {
|
|||||||
// 交易
|
// 交易
|
||||||
trade := api.Group("/trade")
|
trade := api.Group("/trade")
|
||||||
trade.Post("/callback/alipay", handlers.AlipayCallback)
|
trade.Post("/callback/alipay", handlers.AlipayCallback)
|
||||||
|
trade.Post("/callback/wechat", handlers.WechatPayCallback)
|
||||||
|
trade.Post("/check", handlers.TradeCheckSSE)
|
||||||
|
|
||||||
// 账单
|
// 账单
|
||||||
bill := api.Group("/bill")
|
bill := api.Group("/bill")
|
||||||
|
|||||||
@@ -87,7 +87,7 @@ func (s *resourceService) CreateResource(uid int32, now time.Time, ser *CreateRe
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *resourceService) PrepareResource(uid int32, now time.Time, method trade2.Method, ser *CreateResourceSerializer) (*TransactionPrepareResult, error) {
|
func (s *resourceService) PrepareResource(uid int32, now time.Time, method trade2.Method, ser *CreateResourceSerializer) (*TradeCreateResult, error) {
|
||||||
|
|
||||||
data, err := ser.ToData()
|
data, err := ser.ToData()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -98,12 +98,12 @@ func (s *resourceService) PrepareResource(uid int32, now time.Time, method trade
|
|||||||
amount := data.GetPrice()
|
amount := data.GetPrice()
|
||||||
|
|
||||||
// 保存到数据库
|
// 保存到数据库
|
||||||
var result *TransactionPrepareResult
|
var result *TradeCreateResult
|
||||||
err = q.Q.Transaction(func(q *q.Query) error {
|
err = q.Q.Transaction(func(q *q.Query) error {
|
||||||
var err error
|
var err error
|
||||||
|
|
||||||
// 生成交易订单
|
// 生成交易订单
|
||||||
result, err = Transaction.PrepareTransaction(q, uid, now, &TransactionPrepareData{
|
result, err = Trade.SendCreateTradeByQrcode(q, uid, now, &TradeCreateData{
|
||||||
Subject: "购买套餐 - " + name,
|
Subject: "购买套餐 - " + name,
|
||||||
Amount: amount,
|
Amount: amount,
|
||||||
ExpireAt: time.Now().Add(30 * time.Minute),
|
ExpireAt: time.Now().Add(30 * time.Minute),
|
||||||
@@ -140,7 +140,7 @@ func (s *resourceService) PrepareResource(uid int32, now time.Time, method trade
|
|||||||
return result, nil
|
return result, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *resourceService) CompleteResource(tradeNo string, now time.Time, opResult ...*TransactionVerifyResult) error {
|
func (s *resourceService) CompleteResource(tradeNo string, now time.Time, opResult ...*TradeSuccessResult) error {
|
||||||
|
|
||||||
// 获取请求缓存
|
// 获取请求缓存
|
||||||
reqStr, err := g.Redis.Get(context.Background(), tradeNo).Result()
|
reqStr, err := g.Redis.Get(context.Background(), tradeNo).Result()
|
||||||
@@ -153,12 +153,12 @@ func (s *resourceService) CompleteResource(tradeNo string, now time.Time, opResu
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 检查交易结果
|
// 检查交易结果
|
||||||
var rs *TransactionVerifyResult
|
var rs *TradeSuccessResult
|
||||||
if len(opResult) > 0 && opResult[0] != nil {
|
if len(opResult) > 0 && opResult[0] != nil {
|
||||||
rs = opResult[0]
|
rs = opResult[0]
|
||||||
} else {
|
} else {
|
||||||
var err error
|
var err error
|
||||||
rs, err = Transaction.VerifyTransaction(&TransactionVerifyData{
|
rs, err = Trade.VerifyTrade(&TradeVerifyData{
|
||||||
TradeNo: tradeNo,
|
TradeNo: tradeNo,
|
||||||
Method: cache.Method,
|
Method: cache.Method,
|
||||||
})
|
})
|
||||||
@@ -175,10 +175,19 @@ func (s *resourceService) CompleteResource(tradeNo string, now time.Time, opResu
|
|||||||
// 保存交易信息
|
// 保存交易信息
|
||||||
err = q.Q.Transaction(func(q *q.Query) error {
|
err = q.Q.Transaction(func(q *q.Query) error {
|
||||||
|
|
||||||
|
// 完成交易
|
||||||
|
_, err = Trade.OnTradeCreated(q, &OnTradeCreateData{
|
||||||
|
TradeNo: tradeNo,
|
||||||
|
TradeSuccessResult: *rs,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("完成交易失败: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
// 保存套餐
|
// 保存套餐
|
||||||
resource, err := createResource(q, cache.Uid, now, data)
|
resource, err := createResource(q, cache.Uid, now, data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("创建套餐失败: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// 更新账单
|
// 更新账单
|
||||||
@@ -189,22 +198,13 @@ func (s *resourceService) CompleteResource(tradeNo string, now time.Time, opResu
|
|||||||
ResourceID: &resource.ID,
|
ResourceID: &resource.ID,
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("更新账单失败: %w", err)
|
||||||
}
|
|
||||||
|
|
||||||
// 完成交易
|
|
||||||
_, err = Transaction.CompleteTransaction(q, &TransactionCompleteData{
|
|
||||||
TradeNo: tradeNo,
|
|
||||||
TransactionVerifyResult: *rs,
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// 删除缓存
|
// 删除缓存
|
||||||
err = g.Redis.Del(context.Background(), tradeNo).Err()
|
err = g.Redis.Del(context.Background(), tradeNo).Err()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return fmt.Errorf("删除缓存失败: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
@@ -229,14 +229,14 @@ func (s *resourceService) CancelResource(tradeNo string, now time.Time, opRevoke
|
|||||||
|
|
||||||
// 取消交易
|
// 取消交易
|
||||||
if len(opRevoked) <= 0 {
|
if len(opRevoked) <= 0 {
|
||||||
err = Transaction.RevokeTransaction(tradeNo, cache.Method)
|
err = Trade.SendCancelTrade(tradeNo, cache.Method)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// 更新订单状态
|
// 更新订单状态
|
||||||
err = Transaction.FinishTransaction(q.Q, tradeNo, now)
|
err = Trade.OnTradeCanceled(q.Q, tradeNo, now)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -16,6 +16,7 @@ import (
|
|||||||
"platform/web/globals/orm"
|
"platform/web/globals/orm"
|
||||||
m "platform/web/models"
|
m "platform/web/models"
|
||||||
q "platform/web/queries"
|
q "platform/web/queries"
|
||||||
|
"platform/web/tasks"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/smartwalle/alipay/v3"
|
"github.com/smartwalle/alipay/v3"
|
||||||
@@ -23,12 +24,12 @@ import (
|
|||||||
"gorm.io/gorm"
|
"gorm.io/gorm"
|
||||||
)
|
)
|
||||||
|
|
||||||
var Transaction = &transactionService{}
|
var Trade = &tradeService{}
|
||||||
|
|
||||||
type transactionService struct {
|
type tradeService struct {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *transactionService) PrepareTransaction(q *q.Query, uid int32, now time.Time, data *TransactionPrepareData) (*TransactionPrepareResult, error) {
|
func (s *tradeService) SendCreateTradeByQrcode(q *q.Query, uid int32, now time.Time, data *TradeCreateData) (*TradeCreateResult, error) {
|
||||||
var subject = data.Subject
|
var subject = data.Subject
|
||||||
var expire = data.ExpireAt
|
var expire = data.ExpireAt
|
||||||
var tType = data.Type
|
var tType = data.Type
|
||||||
@@ -184,15 +185,118 @@ func (s *transactionService) PrepareTransaction(q *q.Query, uid int32, now time.
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return &TransactionPrepareResult{
|
// 提交异步任务更新订单状态
|
||||||
|
_, err = g.Asynq.Enqueue(tasks.NewUpdateTrade(tradeNo, method))
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return &TradeCreateResult{
|
||||||
TradeNo: tradeNo,
|
TradeNo: tradeNo,
|
||||||
PayURL: payUrl,
|
PayURL: payUrl,
|
||||||
Bill: &bill,
|
Bill: &bill,
|
||||||
Trade: &trade,
|
Trade: &trade,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
func (s *tradeService) SendCreateTradeByRedirect() {
|
||||||
|
panic("todo")
|
||||||
|
}
|
||||||
|
|
||||||
func (s *transactionService) VerifyTransaction(data *TransactionVerifyData) (*TransactionVerifyResult, error) {
|
func (s *tradeService) OnTradeCreated(q *q.Query, data *OnTradeCreateData) (*m.Trade, error) {
|
||||||
|
var transId = data.TransId
|
||||||
|
var tradeNo = data.TradeNo
|
||||||
|
var payment = data.Payment
|
||||||
|
var paidAt = data.Time
|
||||||
|
|
||||||
|
// 获取交易信息
|
||||||
|
trade, err := q.Trade.
|
||||||
|
Where(q.Trade.InnerNo.Eq(tradeNo)).
|
||||||
|
First()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查交易状态
|
||||||
|
switch trade2.Status(trade.Status) {
|
||||||
|
|
||||||
|
// 如果已退款或取消,则返回错误
|
||||||
|
case trade2.StatusCanceled, trade2.StatusRefunded:
|
||||||
|
return nil, errors.New("交易已取消或已退款")
|
||||||
|
|
||||||
|
// 如果是未支付,则更新支付状态
|
||||||
|
case trade2.StatusPending:
|
||||||
|
trade.Status = int32(trade2.StatusSuccess)
|
||||||
|
trade.OuterNo = &transId
|
||||||
|
trade.Payment = payment
|
||||||
|
trade.PaidAt = u.P(orm.LocalDateTime(paidAt))
|
||||||
|
trade.PayURL = u.P("")
|
||||||
|
_, err = q.Trade.Updates(trade)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
case trade2.StatusSuccess:
|
||||||
|
}
|
||||||
|
|
||||||
|
return trade, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *tradeService) SendCancelTrade(tradeNo string, method trade2.Method) error {
|
||||||
|
|
||||||
|
switch method {
|
||||||
|
|
||||||
|
case trade2.MethodAlipay:
|
||||||
|
resp, err := g.Alipay.TradeCancel(context.Background(), alipay.TradeCancel{
|
||||||
|
OutTradeNo: tradeNo,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if resp.Code != alipay.CodeSuccess {
|
||||||
|
slog.Warn("支付宝交易取消失败", "code", resp.Code, "sub_code", resp.SubCode, "msg", resp.Msg)
|
||||||
|
return errors.New("交易取消失败")
|
||||||
|
}
|
||||||
|
|
||||||
|
case trade2.MethodWeChat:
|
||||||
|
resp, err := g.WechatPay.Native.CloseOrder(context.Background(), native.CloseOrderRequest{
|
||||||
|
Mchid: &env.WechatPayMchId,
|
||||||
|
OutTradeNo: &tradeNo,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if resp.Response.StatusCode != http.StatusNoContent {
|
||||||
|
body, _ := io.ReadAll(resp.Response.Body)
|
||||||
|
slog.Warn("微信交易取消失败", "code", resp.Response.StatusCode, "body", string(body))
|
||||||
|
return errors.New("交易取消失败")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
func (s *tradeService) OnTradeCanceled(q *q.Query, tradeNo string, now time.Time) error {
|
||||||
|
_, err := q.Trade.
|
||||||
|
Where(q.Trade.InnerNo.Eq(tradeNo)).
|
||||||
|
Select(q.Trade.Status, q.Trade.CancelAt, q.Trade.PayURL).
|
||||||
|
Updates(m.Trade{
|
||||||
|
Status: int32(trade2.StatusCanceled),
|
||||||
|
CancelAt: u.P(orm.LocalDateTime(now)),
|
||||||
|
PayURL: u.P(""),
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *tradeService) SendRefundTrade(tradeNo string, method trade2.Method) error {
|
||||||
|
panic("todo")
|
||||||
|
}
|
||||||
|
func (s *tradeService) OnTradeRefunded(q *q.Query, tradeNo string, now time.Time) error {
|
||||||
|
panic("todo")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *tradeService) VerifyTrade(data *TradeVerifyData) (*TradeSuccessResult, error) {
|
||||||
var tradeNo = data.TradeNo
|
var tradeNo = data.TradeNo
|
||||||
var method = data.Method
|
var method = data.Method
|
||||||
|
|
||||||
@@ -253,104 +357,14 @@ func (s *transactionService) VerifyTransaction(data *TransactionVerifyData) (*Tr
|
|||||||
return nil, ErrTransactionNotSupported
|
return nil, ErrTransactionNotSupported
|
||||||
}
|
}
|
||||||
|
|
||||||
return &TransactionVerifyResult{
|
return &TradeSuccessResult{
|
||||||
TransId: transId,
|
TransId: transId,
|
||||||
Payment: payment,
|
Payment: payment,
|
||||||
Time: paidAt,
|
Time: paidAt,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *transactionService) CompleteTransaction(q *q.Query, data *TransactionCompleteData) (*TransactionCompleteResult, error) {
|
type TradeCreateData struct {
|
||||||
var transId = data.TransId
|
|
||||||
var tradeNo = data.TradeNo
|
|
||||||
var payment = data.Payment
|
|
||||||
var paidAt = data.Time
|
|
||||||
|
|
||||||
// 获取交易信息
|
|
||||||
trade, err := q.Trade.
|
|
||||||
Where(q.Trade.InnerNo.Eq(tradeNo)).
|
|
||||||
First()
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// 检查交易状态
|
|
||||||
switch trade2.Status(trade.Status) {
|
|
||||||
|
|
||||||
// 如果已退款或取消,则返回错误
|
|
||||||
case trade2.StatusCanceled, trade2.StatusRefunded:
|
|
||||||
return nil, errors.New("交易已取消或已退款")
|
|
||||||
|
|
||||||
// 如果是未支付,则更新支付状态
|
|
||||||
case trade2.StatusPending:
|
|
||||||
trade.Status = int32(trade2.StatusSuccess)
|
|
||||||
trade.OuterNo = &transId
|
|
||||||
trade.Payment = payment
|
|
||||||
trade.PaidAt = u.P(orm.LocalDateTime(paidAt))
|
|
||||||
trade.PayURL = u.P("")
|
|
||||||
_, err = q.Trade.Updates(trade)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
case trade2.StatusSuccess:
|
|
||||||
}
|
|
||||||
|
|
||||||
return &TransactionCompleteResult{
|
|
||||||
Trade: trade,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *transactionService) RevokeTransaction(tradeNo string, method trade2.Method) error {
|
|
||||||
|
|
||||||
switch method {
|
|
||||||
|
|
||||||
case trade2.MethodAlipay:
|
|
||||||
resp, err := g.Alipay.TradeCancel(context.Background(), alipay.TradeCancel{
|
|
||||||
OutTradeNo: tradeNo,
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if resp.Code != alipay.CodeSuccess {
|
|
||||||
slog.Warn("支付宝交易取消失败", "code", resp.Code, "sub_code", resp.SubCode, "msg", resp.Msg)
|
|
||||||
return errors.New("交易取消失败")
|
|
||||||
}
|
|
||||||
|
|
||||||
case trade2.MethodWeChat:
|
|
||||||
resp, err := g.WechatPay.Native.CloseOrder(context.Background(), native.CloseOrderRequest{
|
|
||||||
Mchid: &env.WechatPayMchId,
|
|
||||||
OutTradeNo: &tradeNo,
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if resp.Response.StatusCode != http.StatusNoContent {
|
|
||||||
body, _ := io.ReadAll(resp.Response.Body)
|
|
||||||
slog.Warn("微信交易取消失败", "code", resp.Response.StatusCode, "body", string(body))
|
|
||||||
return errors.New("交易取消失败")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *transactionService) FinishTransaction(q *q.Query, tradeNo string, now time.Time) error {
|
|
||||||
_, err := q.Trade.
|
|
||||||
Where(q.Trade.InnerNo.Eq(tradeNo)).
|
|
||||||
Select(q.Trade.Status, q.Trade.CancelAt, q.Trade.PayURL).
|
|
||||||
Updates(m.Trade{
|
|
||||||
Status: int32(trade2.StatusCanceled),
|
|
||||||
CancelAt: u.P(orm.LocalDateTime(now)),
|
|
||||||
PayURL: u.P(""),
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type TransactionPrepareData struct {
|
|
||||||
Subject string
|
Subject string
|
||||||
Amount decimal.Decimal
|
Amount decimal.Decimal
|
||||||
ExpireAt time.Time
|
ExpireAt time.Time
|
||||||
@@ -359,40 +373,44 @@ type TransactionPrepareData struct {
|
|||||||
CouponCode string
|
CouponCode string
|
||||||
}
|
}
|
||||||
|
|
||||||
type TransactionPrepareResult struct {
|
type TradeCreateResult struct {
|
||||||
TradeNo string
|
TradeNo string
|
||||||
PayURL string
|
PayURL string
|
||||||
Bill *m.Bill
|
Bill *m.Bill
|
||||||
Trade *m.Trade
|
Trade *m.Trade
|
||||||
}
|
}
|
||||||
|
|
||||||
type TransactionVerifyData struct {
|
type TradeVerifyData struct {
|
||||||
TradeNo string
|
TradeNo string
|
||||||
Method trade2.Method
|
Method trade2.Method
|
||||||
}
|
}
|
||||||
|
|
||||||
type TransactionVerifyResult struct {
|
type TradeSuccessResult struct {
|
||||||
TransId string
|
TransId string
|
||||||
Payment decimal.Decimal
|
Payment decimal.Decimal
|
||||||
Time time.Time
|
Time time.Time
|
||||||
}
|
}
|
||||||
|
|
||||||
type TransactionCompleteData struct {
|
type OnTradeCreateData struct {
|
||||||
TradeNo string
|
TradeNo string
|
||||||
TransactionVerifyResult
|
TradeSuccessResult
|
||||||
}
|
}
|
||||||
|
|
||||||
type TransactionCompleteResult struct {
|
type TradeResult int
|
||||||
Trade *m.Trade
|
|
||||||
}
|
|
||||||
|
|
||||||
type TransactionErr string
|
const (
|
||||||
|
TradeSuccess TradeResult = iota + 1
|
||||||
|
TradeCanceled
|
||||||
|
TradeClosed
|
||||||
|
)
|
||||||
|
|
||||||
func (e TransactionErr) Error() string {
|
type TradeErr string
|
||||||
|
|
||||||
|
func (e TradeErr) Error() string {
|
||||||
return string(e)
|
return string(e)
|
||||||
}
|
}
|
||||||
|
|
||||||
var (
|
var (
|
||||||
ErrTransactionNotPaid = TransactionErr("交易未支付")
|
ErrTransactionNotPaid = TradeErr("交易未支付")
|
||||||
ErrTransactionNotSupported = TransactionErr("不支持的支付方式")
|
ErrTransactionNotSupported = TradeErr("不支持的支付方式")
|
||||||
)
|
)
|
||||||
@@ -8,14 +8,14 @@ var User = &userService{}
|
|||||||
|
|
||||||
type userService struct{}
|
type userService struct{}
|
||||||
|
|
||||||
func (s *userService) RechargeConfirm(tradeNo string, verified *TransactionVerifyResult) error {
|
func (s *userService) RechargeConfirm(tradeNo string, verified *TradeSuccessResult) error {
|
||||||
|
|
||||||
err := q.Q.Transaction(func(tx *q.Query) error {
|
err := q.Q.Transaction(func(tx *q.Query) error {
|
||||||
|
|
||||||
// 更新交易状态
|
// 更新交易状态
|
||||||
result, err := Transaction.CompleteTransaction(tx, &TransactionCompleteData{
|
trade, err := Trade.OnTradeCreated(tx, &OnTradeCreateData{
|
||||||
TradeNo: tradeNo,
|
TradeNo: tradeNo,
|
||||||
TransactionVerifyResult: *verified,
|
TradeSuccessResult: *verified,
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@@ -23,14 +23,14 @@ func (s *userService) RechargeConfirm(tradeNo string, verified *TransactionVerif
|
|||||||
|
|
||||||
// 更新用户余额
|
// 更新用户余额
|
||||||
user, err := tx.User.
|
user, err := tx.User.
|
||||||
Where(tx.User.ID.Eq(result.Trade.UserID)).Take()
|
Where(tx.User.ID.Eq(trade.UserID)).Take()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = tx.User.
|
_, err = tx.User.
|
||||||
Where(tx.User.ID.Eq(user.ID)).
|
Where(tx.User.ID.Eq(user.ID)).
|
||||||
UpdateSimple(tx.User.Balance.Value(user.Balance.Add(result.Trade.Amount)))
|
UpdateSimple(tx.User.Balance.Value(user.Balance.Add(trade.Amount)))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|||||||
19
web/tasks/trade.go
Normal file
19
web/tasks/trade.go
Normal file
@@ -0,0 +1,19 @@
|
|||||||
|
package tasks
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"github.com/hibiken/asynq"
|
||||||
|
"log/slog"
|
||||||
|
trade2 "platform/web/domains/trade"
|
||||||
|
)
|
||||||
|
|
||||||
|
const UpdateTrade = "trade:update"
|
||||||
|
|
||||||
|
func NewUpdateTrade(tradeNo string, method trade2.Method) *asynq.Task {
|
||||||
|
bytes, err := json.Marshal(tradeNo)
|
||||||
|
if err != nil {
|
||||||
|
slog.Error("序列化更新交易任务失败", "error", err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return asynq.NewTask(UpdateTrade, bytes)
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user