lanhu/platform
2
0
Fork 0
Code Issues Actions Packages Releases 13 Activity

重构支付接口，以动态支持多种产品类型，整合长短效套餐的购买支付逻辑；引入 decimal 类型的金额计算；

Browse Source
This commit is contained in:
luorijun
2025-05-20 17:14:07 +08:00
parent 689964e3ef
commit 6f1bc72912
45 changed files with 891 additions and 754 deletions
Download Patch File Download Diff File Expand all files Collapse all files

455
web/services/resource.go
View File

@@ -5,6 +5,8 @@ import (
"database/sql"
"encoding/json"
"fmt"
"github.com/gofiber/fiber/v2"
"github.com/shopspring/decimal"
bill2 "platform/web/domains/bill"
resource2 "platform/web/domains/resource"
trade2 "platform/web/domains/trade"
@@ -12,27 +14,97 @@ import (
"platform/web/globals/orm"
m "platform/web/models"
q "platform/web/queries"
"strings"
"time"
"github.com/gofiber/fiber/v2"
)
var Resource = &resourceService{}
type resourceService struct{}
func (s *resourceService) PrepareResource(ctx context.Context, data *CreateResourceData, uid int32, method trade2.Method) (*TransactionPrepareResult, error) {
func (s *resourceService) CreateResource(uid int32, now time.Time, ser *CreateResourceSerializer) error {
data, err := ser.ToData()
if err != nil {
return err
}
var name = data.GetName()
var amount = data.GetPrice()
// 保存交易信息
err = q.Q.Transaction(func(q *q.Query) error {
// 检查用户
user, err := q.User.
Where(q.User.ID.Eq(uid)).
Take()
if err != nil {
return err
}
// 检查余额
if user.Balance.Cmp(amount) < 0 {
return fiber.NewError(fiber.StatusBadRequest, "余额不足")
}
// 保存套餐
resource, err := createResource(q, uid, now, data)
if err != nil {
return err
}
// 生成账单
bill := m.Bill{
UserID: uid,
ResourceID: resource.ID,
BillNo: ID.GenReadable("bil"),
Info: "购买套餐 - " + name,
Type: int32(bill2.TypeConsume),
Amount: amount,
}
err = q.Bill.
Omit(q.Bill.TradeID, q.Bill.RefundID).
Create(&bill)
if err != nil {
return err
}
// 更新用户余额
_, err = q.User.UpdateSimple(q.User.Balance.Value(user.Balance.Sub(amount)))
if err != nil {
return err
}
if err != nil {
return err
}
return nil
}, &sql.TxOptions{Isolation: sql.LevelRepeatableRead})
if err != nil {
return err
}
return nil
}
func (s *resourceService) PrepareResource(uid int32, now time.Time, method trade2.Method, ser *CreateResourceSerializer) (*TransactionPrepareResult, error) {
data, err := ser.ToData()
if err != nil {
return nil, err
}
name := data.GetName()
amount := data.GetPrice()
// 保存到数据库
var result *TransactionPrepareResult
err := q.Q.Transaction(func(q *q.Query) error {
err = q.Q.Transaction(func(q *q.Query) error {
var err error
// 生成交易订单
result, err = Transaction.PrepareTransaction(ctx, q, uid, &TransactionPrepareData{
Subject: "购买套餐 - " + data.GetName(),
result, err = Transaction.PrepareTransaction(q, uid, now, &TransactionPrepareData{
Subject: "购买套餐 - " + name,
Amount: amount,
ExpireAt: time.Now().Add(30 * time.Minute),
Type: trade2.TypeRecharge,
@@ -43,16 +115,18 @@ func (s *resourceService) PrepareResource(ctx context.Context, data *CreateResou
}
// 保存请求缓存
reqStr, err := json.Marshal(&CreateResourceCache{
CreateResourceData: *data,
Uid: uid,
TradeId: result.Trade.ID,
BillId: result.Bill.ID,
})
if err != nil {
resourceSerializer := CreateResourceSerializer{}
if err := resourceSerializer.ByData(data); err != nil {
return err
}
err = g.Redis.Set(ctx, result.TradeNo, reqStr, 30*time.Minute).Err()
err = g.Redis.Set(context.Background(), result.TradeNo, &CreateResourceCache{
Uid: uid,
TradeId: result.Trade.ID,
BillId: result.Bill.ID,
Method: method,
CreateResourceSerializer: resourceSerializer,
}, 30*time.Minute).Err()
if err != nil {
return err
}
@@ -66,10 +140,10 @@ func (s *resourceService) PrepareResource(ctx context.Context, data *CreateResou
return result, nil
}
func (s *resourceService) CompleteResource(ctx context.Context, tradeNo string, rs *TransactionVerifyResult) error {
func (s *resourceService) CompleteResource(tradeNo string, now time.Time, opResult ...*TransactionVerifyResult) error {
// 获取请求缓存
reqStr, err := g.Redis.Get(ctx, tradeNo).Result()
reqStr, err := g.Redis.Get(context.Background(), tradeNo).Result()
if err != nil {
return err
}
@@ -78,11 +152,31 @@ func (s *resourceService) CompleteResource(ctx context.Context, tradeNo string,
return err
}
// 检查交易结果
var rs *TransactionVerifyResult
if len(opResult) > 0 {
rs = opResult[0]
} else {
var err error
rs, err = Transaction.VerifyTransaction(&TransactionVerifyData{
TradeNo: tradeNo,
Method: cache.Method,
})
if err != nil {
return err
}
}
data, err := cache.ToData()
if err != nil {
return err
}
// 保存交易信息
err = q.Q.Transaction(func(q *q.Query) error {
// 保存套餐
resource, err := createResource(q, &cache.CreateResourceData, cache.Uid)
resource, err := createResource(q, cache.Uid, now, data)
if err != nil {
return err
}
@@ -99,7 +193,7 @@ func (s *resourceService) CompleteResource(ctx context.Context, tradeNo string,
}
// 完成交易
_, err = Transaction.CompleteTransaction(ctx, q, &TransactionCompleteData{
_, err = Transaction.CompleteTransaction(q, &TransactionCompleteData{
TradeNo: tradeNo,
TransactionVerifyResult: *rs,
})
@@ -108,7 +202,7 @@ func (s *resourceService) CompleteResource(ctx context.Context, tradeNo string,
}
// 删除缓存
err = g.Redis.Del(ctx, tradeNo).Err()
err = g.Redis.Del(context.Background(), tradeNo).Err()
if err != nil {
return err
}
@@ -122,57 +216,27 @@ func (s *resourceService) CompleteResource(ctx context.Context, tradeNo string,
return nil
}
func (s *resourceService) CreateResource(data *CreateResourceData, uid int32) error {
func (s *resourceService) CancelResource(tradeNo string, now time.Time, opRevoked ...bool) error {
// 删除请求缓存
cacheStr, err := g.Redis.GetDel(context.Background(), tradeNo).Result()
if err != nil {
return err
}
cache := new(CreateResourceCache)
if err := json.Unmarshal([]byte(cacheStr), cache); err != nil {
return err
}
// 保存交易信息
err := q.Q.Transaction(func(q *q.Query) error {
amount := data.GetPrice()
// 检查用户
user, err := q.User.
Where(q.User.ID.Eq(uid)).
Take()
// 取消交易
if len(opRevoked) <= 0 {
err = Transaction.RevokeTransaction(tradeNo, cache.Method)
if err != nil {
return err
}
}
// 检查余额
if user.Balance < amount {
return fiber.NewError(fiber.StatusBadRequest, "余额不足")
}
// 保存套餐
resource, err := createResource(q, data, uid)
if err != nil {
return err
}
// 生成账单
bill := m.Bill{
UserID: uid,
ResourceID: resource.ID,
BillNo: ID.GenReadable("bil"),
Info: "购买套餐 - " + data.GetName(),
Type: int32(bill2.TypeConsume),
Amount: data.GetPrice(),
}
err = q.Bill.
Omit(q.Bill.TradeID, q.Bill.RefundID).
Create(&bill)
if err != nil {
return err
}
// 更新用户余额
_, err = q.User.
Where(q.User.ID.Eq(uid)).
UpdateSimple(q.User.Balance.Sub(amount))
if err != nil {
return err
}
return nil
}, &sql.TxOptions{Isolation: sql.LevelRepeatableRead})
// 更新订单状态
err = Transaction.FinishTransaction(q.Q, tradeNo, now)
if err != nil {
return err
}
@@ -180,81 +244,44 @@ func (s *resourceService) CreateResource(data *CreateResourceData, uid int32) er
return nil
}
type CreateResourceData struct {
Type int32 `json:"type" validate:"required"`
Live int32 `json:"live" validate:"required"`
Expire int32 `json:"expire" validate:"required"`
Quota int32 `json:"quota" validate:"required"`
DailyLimit int32 `json:"daily_limit" validate:"required"`
func createResource(q *q.Query, uid int32, now time.Time, data CreateResourceData) (*m.Resource, error) {
name string
price float64
}
func (data *CreateResourceData) GetName() string {
if data.name == "" {
sb := strings.Builder{}
sb.WriteString("短效动态")
switch data.Type {
case 1:
sb.WriteString("包时 ")
case 2:
sb.WriteString("包量 ")
}
sb.WriteString(fmt.Sprintf("%d 分钟", data.Live/60))
data.name = sb.String()
}
return data.name
}
func (data *CreateResourceData) GetPrice() float64 {
if data.price == 0 {
var count int
switch data.Type {
case 1:
count = int(data.DailyLimit)
case 2:
count = int(data.Quota)
}
seconds := int(data.Live)
if seconds == 180 {
seconds = 150
}
times := int(data.Expire)
if data.Type == 2 {
times = 1
}
data.price = float64(count*seconds*times) / 30000
}
return data.price
}
type CreateResourceCache struct {
CreateResourceData `json:"data"`
Uid int32 `json:"uid"`
TradeId int32 `json:"trade_id"`
BillId int32 `json:"bill_id"`
}
func createResource(q *q.Query, data *CreateResourceData, uid int32) (*m.Resource, error) {
// 创建套餐
resource := m.Resource{
// 套餐基本信息
var resource = m.Resource{
UserID: uid,
ResourceNo: ID.GenReadable("res"),
Active: true,
Type: int32(resource2.TypeShort),
Short: &m.ResourceShort{
Type: data.Type,
}
switch data := data.(type) {
// 短效套餐
case *CreateShortResourceData:
var duration = time.Duration(data.Expire) * 24 * time.Hour
resource.Type = int32(resource2.TypeShort)
resource.Short = &m.ResourceShort{
Type: data.Mode,
Live: data.Live,
Quota: data.Quota,
Expire: orm.LocalDateTime(time.Now().Add(time.Duration(data.Expire) * 24 * time.Hour)),
Expire: orm.LocalDateTime(now.Add(duration)),
DailyLimit: data.DailyLimit,
},
}
// 长效套餐
case *CreateLongResourceData:
var duration = time.Duration(data.Expire) * 24 * time.Hour
resource.Type = int32(resource2.TypeLong)
resource.Long = &m.ResourceLong{
Type: data.Mode,
Live: data.Live,
Quota: data.Quota,
Expire: orm.LocalDateTime(now.Add(duration)),
DailyLimit: data.DailyLimit,
}
default:
return nil, fmt.Errorf("不支持的套餐类型")
}
err := q.Resource.Create(&resource)
if err != nil {
return nil, err
@@ -263,24 +290,154 @@ func createResource(q *q.Query, data *CreateResourceData, uid int32) (*m.Resourc
return &resource, nil
}
func (s *resourceService) CancelResource(ctx context.Context, tradeNo string, at time.Time, method trade2.Method) error {
// 删除请求缓存
_, err := g.Redis.Del(ctx, tradeNo).Result()
if err != nil {
return err
}
type CreateResourceData interface {
GetName() string
GetPrice() decimal.Decimal
}
// 取消交易
err = Transaction.RevokeTransaction(ctx, tradeNo, method)
if err != nil {
return err
}
type CreateShortResourceData struct {
Live int32 `json:"live" validate:"required min=180"`
Mode int32 `json:"mode" validate:"required"`
Expire int32 `json:"expire"`
DailyLimit int32 `json:"daily_limit" validate:"min=2000"`
Quota int32 `json:"quota" validate:"min=10000"`
// 更新订单状态
err = Transaction.FinishTransaction(ctx, q.Q, tradeNo, at)
if err != nil {
return err
}
name string
price *decimal.Decimal
}
func (data *CreateShortResourceData) GetName() string {
if data.name == "" {
var mode string
switch data.Mode {
case 1:
mode = "包时"
case 2:
mode = "包量"
}
data.name = fmt.Sprintf("短效动态%s %d 天", mode, data.Live)
}
return data.name
}
func (data *CreateShortResourceData) GetPrice() decimal.Decimal {
if data.price == nil {
var factor int32
switch data.Mode {
case 1:
factor = data.DailyLimit * data.Expire
case 2:
factor = data.Quota
}
var base = data.Live
if base == 180 {
base = 150
}
var dec = decimal.Decimal{}.
Add(decimal.NewFromInt32(base * factor)).
Div(decimal.NewFromInt(30000))
data.price = &dec
}
return *data.price
}
type CreateLongResourceData struct {
Live int32 `json:"live" validate:"required oneof=1,4,8,12,24"`
Mode int32 `json:"mode" validate:"required oneof=1,2"`
Expire int32 `json:"expire"`
DailyLimit int32 `json:"daily_limit" validate:"min=100"`
Quota int32 `json:"quota" validate:"min=500"`
name string
price *decimal.Decimal
}
func (data *CreateLongResourceData) GetName() string {
if data.name == "" {
var mode string
switch data.Mode {
case 1:
mode = "包时"
case 2:
mode = "包量"
}
data.name = fmt.Sprintf("长效动态%s %d 天", mode, data.Live)
}
return data.name
}
func (data *CreateLongResourceData) GetPrice() decimal.Decimal {
if data.price == nil {
var factor int32 = 0
switch resource2.Mode(data.Mode) {
case resource2.ModeTime:
factor = data.Expire * data.DailyLimit
case resource2.ModeCount:
factor = data.Quota
}
var base int32
switch data.Live {
case 1:
base = 30
case 4:
base = 80
case 8:
base = 120
case 12:
base = 180
case 24:
base = 350
}
// data.price = big.NewRat(int64(base*factor), 100)
var dec = decimal.Decimal{}.
Add(decimal.NewFromInt32(base * factor)).
Div(decimal.NewFromInt(100))
data.price = &dec
}
return *data.price
}
type CreateResourceCache struct {
Uid int32 `json:"uid"`
TradeId int32 `json:"trade_id"`
BillId int32 `json:"bill_id"`
Method trade2.Method `json:"method"`
CreateResourceSerializer
}
type CreateResourceSerializer struct {
Type resource2.Type `json:"type" validate:"required"`
Short *CreateShortResourceData `json:"short,omitempty"`
Long *CreateLongResourceData `json:"long,omitempty"`
}
func (s *CreateResourceSerializer) ToData() (CreateResourceData, error) {
switch s.Type {
case resource2.TypeShort:
return s.Short, nil
case resource2.TypeLong:
return s.Long, nil
}
return nil, fmt.Errorf("不支持的套餐类型")
}
func (s *CreateResourceSerializer) ByData(data CreateResourceData) error {
switch data := data.(type) {
case *CreateShortResourceData:
s.Type = resource2.TypeShort
s.Short = data
case *CreateLongResourceData:
s.Type = resource2.TypeLong
s.Long = data
default:
return fmt.Errorf("不支持的套餐类型")
}
return nil
}