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

实现用户咨询数据收集接口

Browse Source
This commit is contained in:
luorijun
2025-12-18 14:22:56 +08:00
parent 0207720943
commit 8f2e71849f
14 changed files with 606 additions and 121 deletions
Download Patch File Download Diff File Expand all files Collapse all files

120
web/services/resource.go
View File

@@ -6,7 +6,6 @@ import (
"fmt"
"platform/pkg/u"
"platform/web/core"
g "platform/web/globals"
m "platform/web/models"
q "platform/web/queries"
"time"
@@ -19,64 +18,67 @@ var Resource = &resourceService{}
type resourceService struct{}
func (s *resourceService) CreateResourceByBalance(uid int32, now time.Time, data *CreateResourceData) error {
return g.Redsync.WithLock(userBalanceKey(uid), func() error {
return q.Q.Transaction(func(q *q.Query) error {
// 找到用户
user, err := q.User.
Where(q.User.ID.Eq(uid)).
Take()
if err != nil {
return err
}
// 检查余额
amount, err := data.GetAmount()
if err != nil {
return err
}
balance := user.Balance.Sub(amount)
if balance.IsNegative() {
return ErrBalanceNotEnough
}
// 找到用户
user, err := q.User.
Where(q.User.ID.Eq(uid)).
Take()
if err != nil {
return err
}
// 更新用户余额
_, err = q.User.
Where(q.User.ID.Eq(uid), q.User.Balance.Eq(user.Balance)).
UpdateSimple(q.User.Balance.Value(balance))
if err != nil {
return core.NewServErr("更新用户余额失败", err)
}
// 检查余额
amount, err := data.GetAmount()
if err != nil {
return err
}
newBalance := user.Balance.Sub(amount)
if newBalance.IsNegative() {
return ErrBalanceNotEnough
}
// 保存套餐
resource, err := createResource(q, uid, now, data)
if err != nil {
return core.NewServErr("创建套餐失败", err)
}
return q.Q.Transaction(func(q *q.Query) error {
// 生成账单
subject, err := data.GetSubject()
if err != nil {
return err
}
err = q.Bill.Create(newForConsume(uid, Bill.GenNo(), subject, amount, resource))
if err != nil {
return core.NewServErr("生成账单失败", err)
}
// 更新用户余额
_, err = q.User.
Where(
q.User.ID.Eq(uid),
q.User.Balance.Eq(user.Balance),
).
UpdateSimple(q.User.Balance.Value(newBalance))
if err != nil {
return core.NewServErr("更新用户余额失败", err)
}
return nil
})
// 保存套餐
resource, err := createResource(q, uid, now, data)
if err != nil {
return core.NewServErr("创建套餐失败", err)
}
// 生成账单
subject, err := data.GetSubject()
if err != nil {
return err
}
err = q.Bill.Create(newForConsume(uid, Bill.GenNo(), subject, amount, resource))
if err != nil {
return core.NewServErr("生成账单失败", err)
}
return nil
})
}
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 *CreateResourceData, trade *m.Trade) error { // 检查交易
if trade == nil {
return core.NewBizErr("交易数据不能为空")
}
if trade.Status != m.TradeStatusSuccess {
return core.NewBizErr("交易状态不正确")
}
return q.Q.Transaction(func(q *q.Query) error {
// 检查交易
if trade == nil {
return core.NewBizErr("交易数据不能为空")
}
if trade.Status != m.TradeStatusSuccess {
return core.NewBizErr("交易状态不正确")
}
// 保存套餐
resource, err := createResource(q, uid, now, data)
@@ -171,7 +173,7 @@ type CreateResourceData struct {
type CreateShortResourceData struct {
Live int32 `json:"live" validate:"required,min=180"`
Mode m.ResourceMode `json:"mode" validate:"required"`
Quota int32 `json:"quota"`
Quota int32 `json:"quota" validate:"required"`
Expire *int32 `json:"expire"`
name string
@@ -182,7 +184,7 @@ type CreateLongResourceData struct {
Live int32 `json:"live" validate:"required"`
Mode m.ResourceMode `json:"mode" validate:"required"`
Quota int32 `json:"quota" validate:"required"`
Expire *int32 `json:"expire" validate:"required"`
Expire *int32 `json:"expire"`
name string
price *decimal.Decimal
@@ -193,25 +195,25 @@ func (c *CreateResourceData) GetType() m.TradeType {
}
func (c *CreateResourceData) GetSubject() (string, error) {
switch c.Type {
switch {
default:
return "", errors.New("无效的套餐类型")
case m.ResourceTypeShort:
case c.Type == m.ResourceTypeShort && c.Short != nil:
return c.Short.GetSubject()
case m.ResourceTypeLong:
case c.Type == m.ResourceTypeLong && c.Long != nil:
return c.Long.GetSubject()
}
}
func (c *CreateResourceData) GetAmount() (decimal.Decimal, error) {
switch c.Type {
switch {
default:
return decimal.Zero, errors.New("无效的套餐类型")
case m.ResourceTypeShort:
case c.Type == m.ResourceTypeShort && c.Short != nil:
return c.Short.GetAmount()
case m.ResourceTypeLong:
case c.Type == m.ResourceTypeLong && c.Long != nil:
return c.Long.GetAmount()
}
}
@@ -339,6 +341,7 @@ func (data *CreateLongResourceData) GetAmount() (decimal.Decimal, error) {
return *data.price, nil
}
// 交易后创建套餐
type ResourceOnTradeComplete struct{}
func (r ResourceOnTradeComplete) Check(t m.TradeType) (ProductInfo, bool) {
@@ -352,6 +355,7 @@ func (r ResourceOnTradeComplete) OnTradeComplete(info ProductInfo, trade *m.Trad
return Resource.CreateResourceByTrade(trade.UserID, time.Time(*trade.CompletedAt), info.(*CreateResourceData), trade)
}
// 服务错误
type ResourceServiceErr string
func (e ResourceServiceErr) Error() string {