You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
paopao-ce/internal/dao/jinzhu/wallet.go

159 lines
3.8 KiB

// Copyright 2022 ROC. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.
package jinzhu
2 years ago
import (
"github.com/rocboss/paopao-ce/internal/conf"
"github.com/rocboss/paopao-ce/internal/core"
"github.com/rocboss/paopao-ce/internal/dao/jinzhu/dbr"
2 years ago
"gorm.io/gorm"
)
var (
_ core.WalletService = (*walletServant)(nil)
)
type walletServant struct {
db *gorm.DB
}
func newWalletService(db *gorm.DB) core.WalletService {
return &walletServant{
db: db,
}
}
func (d *walletServant) GetRechargeByID(id int64) (*core.WalletRecharge, error) {
recharge := &dbr.WalletRecharge{
Model: &dbr.Model{
2 years ago
ID: id,
},
}
return recharge.Get(d.db)
2 years ago
}
func (d *walletServant) CreateRecharge(userId, amount int64) (*core.WalletRecharge, error) {
recharge := &dbr.WalletRecharge{
2 years ago
UserID: userId,
Amount: amount,
}
return recharge.Create(d.db)
}
func (d *walletServant) GetUserWalletBills(userID int64, offset, limit int) ([]*core.WalletStatement, error) {
statement := &dbr.WalletStatement{
UserID: userID,
}
return statement.List(d.db, &dbr.ConditionsT{
"ORDER": "id DESC",
}, offset, limit)
}
func (d *walletServant) GetUserWalletBillCount(userID int64) (int64, error) {
statement := &dbr.WalletStatement{
UserID: userID,
}
return statement.Count(d.db, &dbr.ConditionsT{})
2 years ago
}
func (d *walletServant) HandleRechargeSuccess(recharge *core.WalletRecharge, tradeNo string) error {
user, _ := (&dbr.User{
Model: &dbr.Model{
2 years ago
ID: recharge.UserID,
},
}).Get(d.db)
2 years ago
return d.db.Transaction(func(tx *gorm.DB) error {
2 years ago
// 扣除金额
if err := tx.Model(user).Update("balance", gorm.Expr("balance + ?", recharge.Amount)).Error; err != nil {
// 返回任何错误都会回滚事务
return err
}
// 新增账单
if err := tx.Create(&dbr.WalletStatement{
2 years ago
UserID: user.ID,
ChangeAmount: recharge.Amount,
BalanceSnapshot: user.Balance + recharge.Amount,
Reason: "用户充值",
}).Error; err != nil {
return err
}
// 标记为已付款
if err := tx.Model(recharge).Updates(map[string]any{
2 years ago
"trade_no": tradeNo,
"trade_status": "TRADE_SUCCESS",
}).Error; err != nil {
return err
}
// 返回 nil 提交事务
return nil
})
}
func (d *walletServant) HandlePostAttachmentBought(post *core.Post, user *core.User) error {
return d.db.Transaction(func(tx *gorm.DB) error {
2 years ago
// 扣除金额
if err := tx.Model(user).Update("balance", gorm.Expr("balance - ?", post.AttachmentPrice)).Error; err != nil {
// 返回任何错误都会回滚事务
return err
}
// 新增账单
if err := tx.Create(&dbr.WalletStatement{
2 years ago
PostID: post.ID,
UserID: user.ID,
ChangeAmount: -post.AttachmentPrice,
BalanceSnapshot: user.Balance - post.AttachmentPrice,
Reason: "购买附件支出",
}).Error; err != nil {
return err
}
// 新增附件购买记录
if err := tx.Create(&dbr.PostAttachmentBill{
2 years ago
PostID: post.ID,
UserID: user.ID,
PaidAmount: post.AttachmentPrice,
}).Error; err != nil {
return err
}
// 对附件主新增账单
income := int64(float64(post.AttachmentPrice) * conf.AppSetting.AttachmentIncomeRate)
2 years ago
if income > 0 {
master := &dbr.User{
Model: &dbr.Model{
2 years ago
ID: post.UserID,
},
}
master, _ = master.Get(d.db)
2 years ago
if err := tx.Model(master).Update("balance", gorm.Expr("balance + ?", income)).Error; err != nil {
// 返回任何错误都会回滚事务
return err
}
// 新增账单
if err := tx.Create(&dbr.WalletStatement{
2 years ago
PostID: post.ID,
UserID: master.ID,
ChangeAmount: income,
BalanceSnapshot: master.Balance + income,
Reason: "出售附件收入",
}).Error; err != nil {
return err
}
}
// 返回 nil 提交事务
return nil
})
}