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

支付功能动态化扩展

Browse Source
This commit is contained in:
luorijun
2026-03-18 13:07:06 +08:00
parent 99853b8514
commit 9d996acf5f
16 changed files with 1402 additions and 195 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
web/services/product.go Normal file
View File

@@ -0,0 +1,12 @@
package services
import q "platform/web/queries"
var Product = &productService{}
type productService struct{}
// 获取产品价格
func (s *productService) GetPrice(code string) {
q.ProductSku.Where(q.ProductSku.Code.Eq(code)).Find()
}

262
web/services/resource.go
View File

@@ -11,6 +11,7 @@ import (
"time"
"github.com/shopspring/decimal"
"gorm.io/gorm"
)
var Resource = &resourceService{}
@@ -27,8 +28,14 @@ func (s *resourceService) CreateResourceByBalance(uid int32, now time.Time, data
return err
}
// 获取 sku
sku, err := s.GetSku(data)
if err != nil {
return err
}
// 检查余额
amount, err := data.GetAmount()
_, amount, err := s.GetPrice(sku, data.Count(), &uid)
if err != nil {
return err
}
@@ -57,11 +64,7 @@ func (s *resourceService) CreateResourceByBalance(uid int32, now time.Time, data
}
// 生成账单
subject, err := data.GetSubject()
if err != nil {
return err
}
err = q.Bill.Create(newForConsume(uid, Bill.GenNo(), subject, amount, resource))
err = q.Bill.Create(newForConsume(uid, Bill.GenNo(), sku.Name, amount, resource))
if err != nil {
return core.NewServErr("生成账单失败", err)
}
@@ -70,7 +73,7 @@ func (s *resourceService) CreateResourceByBalance(uid int32, now time.Time, data
})
}
func (s *resourceService) CreateResourceByTrade(uid int32, now time.Time, data *CreateResourceData, trade *m.Trade) error { // 检查交易
func (s *resourceService) CreateResourceByTrade(uid int32, now time.Time, data *CreateResourceByTradeData, trade *m.Trade) error { // 检查交易
if trade == nil {
return core.NewBizErr("交易数据不能为空")
}
@@ -81,21 +84,13 @@ func (s *resourceService) CreateResourceByTrade(uid int32, now time.Time, data *
return q.Q.Transaction(func(q *q.Query) error {
// 保存套餐
resource, err := createResource(q, uid, now, data)
resource, err := createResource(q, uid, now, data.Req)
if err != nil {
return core.NewServErr("创建套餐失败", err)
}
// 生成账单
subject, err := data.GetSubject()
if err != nil {
return err
}
amount, err := data.GetAmount()
if err != nil {
return err
}
err = q.Bill.Create(newForConsume(uid, Bill.GenNo(), subject, amount, resource, trade))
err = q.Bill.Create(newForConsume(uid, Bill.GenNo(), data.GetSubject(), data.GetAmount(), resource, trade))
if err != nil {
return core.NewServErr("生成账单失败", err)
}
@@ -164,6 +159,46 @@ func createResource(q *q.Query, uid int32, now time.Time, data *CreateResourceDa
return &resource, nil
}
func (s *resourceService) GetSku(data *CreateResourceData) (*m.ProductSku, error) {
sku, err := q.ProductSku.Where(q.ProductSku.Code.Eq(data.Code())).Take()
if err != nil {
return nil, core.NewServErr("产品不可用", err)
}
return sku, nil
}
func (s *resourceService) GetPrice(sku *m.ProductSku, count int32, uid *int32) (decimal.Decimal, decimal.Decimal, error) {
// 根据用户 id 查询特殊优惠
var uSku *m.ProductSkuUser
if uid != nil {
var err error
uSku, err = q.ProductSkuUser.Where(
q.ProductSkuUser.UserID.Eq(*uid),
q.ProductSkuUser.ProductSkuID.Eq(sku.ID),
).Take()
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
return decimal.Zero, decimal.Zero, core.NewServErr("客户特殊价查询失败", err)
}
}
// 返回计算价格
price := sku.Price
if uSku != nil && uSku.Price != nil {
price = *uSku.Price
}
discount := sku.Discount
if uSku != nil && uSku.Discount != nil {
discount = *uSku.Discount
}
before := price.Mul(decimal.NewFromInt32(count))
after := before.Mul(decimal.NewFromFloat32(discount))
return before, after, nil
}
type CreateResourceData struct {
Type m.ResourceType `json:"type" validate:"required"`
Short *CreateShortResourceData `json:"short,omitempty"`
@@ -171,7 +206,7 @@ type CreateResourceData struct {
}
type CreateShortResourceData struct {
Live int32 `json:"live" validate:"required,min=180"`
Live int32 `json:"live" validate:"required"`
Mode m.ResourceMode `json:"mode" validate:"required"`
Quota int32 `json:"quota" validate:"required"`
Expire *int32 `json:"expire"`
@@ -190,31 +225,33 @@ type CreateLongResourceData struct {
price *decimal.Decimal
}
func (c *CreateResourceData) GetType() m.TradeType {
return m.TradeTypePurchase
}
func (c *CreateResourceData) GetSubject() (string, error) {
func (c *CreateResourceData) Count() int32 {
switch {
default:
return "", errors.New("无效的套餐类型")
return 0
case c.Type == m.ResourceTypeShort && c.Short != nil:
return c.Short.GetSubject()
return c.Short.Quota
case c.Type == m.ResourceTypeLong && c.Long != nil:
return c.Long.GetSubject()
return c.Long.Quota
}
}
func (c *CreateResourceData) GetAmount() (decimal.Decimal, error) {
func (c *CreateResourceData) Code() string {
switch {
default:
return decimal.Zero, errors.New("无效的套餐类型")
return ""
case c.Type == m.ResourceTypeShort && c.Short != nil:
return c.Short.GetAmount()
return fmt.Sprintf(
"mode=%s,live=%d,expire=%d",
c.Short.Mode.Code(), c.Short.Live, u.Else(c.Short.Expire, 0),
)
case c.Type == m.ResourceTypeLong && c.Long != nil:
return c.Long.GetAmount()
return fmt.Sprintf(
"mode=%s,live=%d,expire=%d",
c.Long.Mode.Code(), c.Long.Live, u.Else(c.Long.Expire, 0),
)
}
}
@@ -227,132 +264,63 @@ func (c *CreateResourceData) Deserialize(str string) error {
return json.Unmarshal([]byte(str), c)
}
func (data *CreateShortResourceData) GetSubject() (string, error) {
if data.name == "" {
var mode string
switch data.Mode {
default:
return "", errors.New("无效的套餐模式")
case m.ResourceModeTime:
mode = "包时"
case m.ResourceModeQuota:
mode = "包量"
}
data.name = fmt.Sprintf("短效动态%s %v 分钟", mode, data.Live/60)
}
return data.name, nil
}
func (data *CreateShortResourceData) GetAmount() (decimal.Decimal, error) {
if data.price == nil {
var factor int32
switch data.Mode {
default:
return decimal.Zero, errors.New("无效的套餐模式")
case m.ResourceModeTime:
if data.Expire == nil {
return decimal.Zero, errors.New("包时套餐过期时间不能为空")
}
factor = data.Quota * *data.Expire
case m.ResourceModeQuota:
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))
if dec.IsZero() {
return decimal.Zero, errors.New("计算金额错误")
}
data.price = &dec
}
return *data.price, nil
}
func (data *CreateLongResourceData) GetSubject() (string, error) {
if data.name == "" {
var mode string
switch data.Mode {
default:
return "", errors.New("无效的套餐模式")
case m.ResourceModeTime:
mode = "包时"
case m.ResourceModeQuota:
mode = "包量"
}
data.name = fmt.Sprintf("长效动态%s %d 小时", mode, data.Live)
}
return data.name, nil
}
func (data *CreateLongResourceData) GetAmount() (decimal.Decimal, error) {
if data.price == nil {
var factor int32 = 0
switch data.Mode {
default:
return decimal.Zero, errors.New("无效的套餐模式")
case m.ResourceModeTime:
if data.Expire == nil {
return decimal.Zero, errors.New("包时套餐过期时间不能为空")
}
factor = *data.Expire * data.Quota
case m.ResourceModeQuota:
factor = data.Quota
}
var base int32
switch data.Live {
default:
return decimal.Zero, errors.New("无效的套餐时长")
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))
if dec.IsZero() {
return decimal.Zero, errors.New("计算金额错误")
}
data.price = &dec
}
return *data.price, nil
}
// 交易后创建套餐
type ResourceOnTradeComplete struct{}
func (r ResourceOnTradeComplete) Check(t m.TradeType) (ProductInfo, bool) {
if t == m.TradeTypePurchase {
return &CreateResourceData{}, true
return &CreateResourceByTradeData{}, true
}
return nil, false
}
func (r ResourceOnTradeComplete) OnTradeComplete(info ProductInfo, trade *m.Trade) error {
return Resource.CreateResourceByTrade(trade.UserID, time.Time(*trade.CompletedAt), info.(*CreateResourceData), trade)
return Resource.CreateResourceByTrade(trade.UserID, time.Time(*trade.CompletedAt), info.(*CreateResourceByTradeData), trade)
}
type CreateResourceByTradeData struct {
Subject string `json:"subject"`
Amount decimal.Decimal `json:"amount"`
Req *CreateResourceData `json:"data"`
}
func (e CreateResourceByTradeData) GetType() m.TradeType {
return m.TradeTypePurchase
}
func (e CreateResourceByTradeData) GetSubject() string {
return e.Subject
}
func (e CreateResourceByTradeData) GetAmount() decimal.Decimal {
return e.Amount
}
func (e CreateResourceByTradeData) Serialize() (string, error) {
bytes, err := json.Marshal(e)
return string(bytes), err
}
func (e *CreateResourceByTradeData) Deserialize(str string) error {
return json.Unmarshal([]byte(str), e)
}
func NewCreateResourceByTradeData(req *CreateResourceData) (*CreateResourceByTradeData, error) {
sku, err := Resource.GetSku(req)
if err != nil {
return nil, err
}
_, amount, err := Resource.GetPrice(sku, req.Count(), nil)
if err != nil {
return nil, err
}
return &CreateResourceByTradeData{
Subject: sku.Name,
Amount: amount,
Req: req,
}, nil
}
// 服务错误

29
web/services/trade.go
View File

@@ -32,19 +32,13 @@ type tradeService struct {
}
// 创建交易
func (s *tradeService) CreateTrade(uid int32, now time.Time, data *CreateTradeData) (*CreateTradeResult, error) {
platform := data.Platform
method := data.Method
tType := data.Product.GetType()
func (s *tradeService) CreateTrade(uid int32, now time.Time, payment *CreateTradeData, product ProductInfo) (*CreateTradeResult, error) {
platform := payment.Platform
method := payment.Method
tType := product.GetType()
expire := time.Now().Add(30 * time.Minute)
subject, err := data.Product.GetSubject()
if err != nil {
return nil, err
}
amount, err := data.Product.GetAmount()
if err != nil {
return nil, err
}
subject := product.GetSubject()
amount := product.GetAmount()
// 实际支付金额，只在创建真实订单时使用
amountReal := amount
@@ -53,10 +47,10 @@ func (s *tradeService) CreateTrade(uid int32, now time.Time, data *CreateTradeDa
}
// 附加优惠券
if data.CouponCode != nil {
if payment.CouponCode != nil {
coupon, err := q.Coupon.
Where(
q.Coupon.Code.Eq(*data.CouponCode),
q.Coupon.Code.Eq(*payment.CouponCode),
q.Coupon.Status.Eq(int(m.CouponStatusUnused)),
).
Take()
@@ -248,7 +242,7 @@ func (s *tradeService) CreateTrade(uid int32, now time.Time, data *CreateTradeDa
}
// 缓存产品数据
serialized, err := data.Product.Serialize()
serialized, err := product.Serialize()
if err != nil {
return nil, core.NewServErr("序列化产品信息失败", err)
}
@@ -684,7 +678,6 @@ type CreateTradeData struct {
Platform m.TradePlatform `json:"platform" validate:"required"`
Method m.TradeMethod `json:"method" validate:"required"`
CouponCode *string `json:"coupon_code"`
Product ProductInfo
}
type CreateTradeResult struct {
@@ -717,8 +710,8 @@ type OnTradeCompletedData struct {
type ProductInfo interface {
GetType() m.TradeType
GetSubject() (string, error)
GetAmount() (decimal.Decimal, error)
GetSubject() string
GetAmount() decimal.Decimal
Serialize() (string, error)
Deserialize(str string) error
}

24
web/services/user.go
View File

@@ -25,14 +25,8 @@ func (s *userService) UpdateBalanceByTrade(uid int32, info *RechargeProductInfo,
}
// 生成账单
subject, err := info.GetSubject()
if err != nil {
return err
}
amount, err := info.GetAmount()
if err != nil {
return err
}
subject := info.GetSubject()
amount := info.GetAmount()
err = q.Bill.Create(newForRecharge(uid, Bill.GenNo(), subject, amount, trade))
if err != nil {
return core.NewServErr("生成账单失败", err)
@@ -54,10 +48,7 @@ func updateBalance(q *q.Query, uid int32, info *RechargeProductInfo) error {
return core.NewServErr("查询用户失败", err)
}
amount, err := info.GetAmount()
if err != nil {
return err
}
amount := info.GetAmount()
balance := user.Balance.Add(amount)
if balance.IsNegative() {
return core.NewServErr("用户余额不足")
@@ -85,13 +76,12 @@ func (r *RechargeProductInfo) GetType() m.TradeType {
return m.TradeTypeRecharge
}
func (r *RechargeProductInfo) GetSubject() (string, error) {
amount, _ := r.GetAmount()
return fmt.Sprintf("账户充值 - %s元", amount.StringFixed(2)), nil
func (r *RechargeProductInfo) GetSubject() string {
return fmt.Sprintf("账户充值 - %s元", r.GetAmount().StringFixed(2))
}
func (r *RechargeProductInfo) GetAmount() (decimal.Decimal, error) {
return decimal.NewFromInt(int64(r.Amount)).Div(decimal.NewFromInt(100)), nil
func (r *RechargeProductInfo) GetAmount() decimal.Decimal {
return decimal.NewFromInt(int64(r.Amount)).Div(decimal.NewFromInt(100))
}
func (r *RechargeProductInfo) Serialize() (string, error) {