mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-03 15:00:27 +08:00
254 lines
6.6 KiB
Go
254 lines
6.6 KiB
Go
package accounts
|
|
|
|
import (
|
|
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/errors"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/goman"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/remotelogs"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/userconfigs"
|
|
_ "github.com/go-sql-driver/mysql"
|
|
"github.com/iwind/TeaGo/Tea"
|
|
"github.com/iwind/TeaGo/dbs"
|
|
"github.com/iwind/TeaGo/lists"
|
|
"github.com/iwind/TeaGo/maps"
|
|
"github.com/iwind/TeaGo/types"
|
|
"time"
|
|
)
|
|
|
|
func init() {
|
|
dbs.OnReadyDone(func() {
|
|
goman.New(func() {
|
|
// 自动支付账单任务
|
|
var ticker = time.NewTicker(12 * time.Hour)
|
|
for range ticker.C {
|
|
if SharedUserAccountDAO.Instance != nil {
|
|
err := SharedUserAccountDAO.Instance.RunTx(func(tx *dbs.Tx) error {
|
|
return SharedUserAccountDAO.PayBills(tx)
|
|
})
|
|
if err != nil {
|
|
remotelogs.Error("USER_ACCOUNT_DAO", "pay bills task failed: "+err.Error())
|
|
}
|
|
}
|
|
}
|
|
})
|
|
})
|
|
}
|
|
|
|
type UserAccountDAO dbs.DAO
|
|
|
|
func NewUserAccountDAO() *UserAccountDAO {
|
|
return dbs.NewDAO(&UserAccountDAO{
|
|
DAOObject: dbs.DAOObject{
|
|
DB: Tea.Env,
|
|
Table: "edgeUserAccounts",
|
|
Model: new(UserAccount),
|
|
PkName: "id",
|
|
},
|
|
}).(*UserAccountDAO)
|
|
}
|
|
|
|
var SharedUserAccountDAO *UserAccountDAO
|
|
|
|
func init() {
|
|
dbs.OnReady(func() {
|
|
SharedUserAccountDAO = NewUserAccountDAO()
|
|
})
|
|
}
|
|
|
|
// FindUserAccountWithUserId 根据用户ID查找用户账户
|
|
func (this *UserAccountDAO) FindUserAccountWithUserId(tx *dbs.Tx, userId int64) (*UserAccount, error) {
|
|
if userId <= 0 {
|
|
return nil, errors.New("invalid userId '" + types.String(userId) + "'")
|
|
}
|
|
|
|
// 用户是否存在
|
|
user, err := models.SharedUserDAO.FindEnabledUser(tx, userId, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if user == nil {
|
|
return nil, errors.New("invalid userId '" + types.String(userId) + "'")
|
|
}
|
|
|
|
account, err := this.Query(tx).
|
|
Attr("userId", userId).
|
|
Find()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if account != nil {
|
|
return account.(*UserAccount), nil
|
|
}
|
|
|
|
var op = NewUserAccountOperator()
|
|
op.UserId = userId
|
|
_, err = this.SaveInt64(tx, op)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return this.FindUserAccountWithUserId(tx, userId)
|
|
}
|
|
|
|
// FindUserAccountWithAccountId 根据ID查找用户账户
|
|
func (this *UserAccountDAO) FindUserAccountWithAccountId(tx *dbs.Tx, accountId int64) (*UserAccount, error) {
|
|
one, err := this.Query(tx).
|
|
Pk(accountId).
|
|
Find()
|
|
if one != nil {
|
|
return one.(*UserAccount), nil
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
// UpdateUserAccount 操作用户账户
|
|
func (this *UserAccountDAO) UpdateUserAccount(tx *dbs.Tx, accountId int64, delta float32, eventType userconfigs.AccountEventType, description string, params maps.Map) error {
|
|
account, err := this.FindUserAccountWithAccountId(tx, accountId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if account == nil {
|
|
return errors.New("invalid account id '" + types.String(accountId) + "'")
|
|
}
|
|
var userId = int64(account.UserId)
|
|
var deltaFloat64 = float64(delta)
|
|
if deltaFloat64 < 0 && account.Total < -deltaFloat64 {
|
|
return errors.New("not enough account quota to decrease")
|
|
}
|
|
|
|
// 操作账户
|
|
err = this.Query(tx).
|
|
Pk(account.Id).
|
|
Set("total", dbs.SQL("total+:delta")).
|
|
Param("delta", delta).
|
|
UpdateQuickly()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 生成日志
|
|
err = SharedUserAccountLogDAO.CreateAccountLog(tx, userId, accountId, delta, 0, eventType, description, params)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// UpdateUserAccountFrozen 操作用户账户冻结余额
|
|
func (this *UserAccountDAO) UpdateUserAccountFrozen(tx *dbs.Tx, userId int64, delta float32, eventType userconfigs.AccountEventType, description string, params maps.Map) error {
|
|
account, err := this.FindUserAccountWithUserId(tx, userId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
var deltaFloat64 = float64(delta)
|
|
if deltaFloat64 < 0 && account.TotalFrozen < -deltaFloat64 {
|
|
return errors.New("not enough account frozen quota to decrease")
|
|
}
|
|
|
|
// 操作账户
|
|
err = this.Query(tx).
|
|
Pk(account.Id).
|
|
Set("totalFrozen", dbs.SQL("total+:delta")).
|
|
Param("delta", delta).
|
|
UpdateQuickly()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 生成日志
|
|
err = SharedUserAccountLogDAO.CreateAccountLog(tx, userId, int64(account.Id), 0, delta, eventType, description, params)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// CountAllAccounts 计算所有账户数量
|
|
func (this *UserAccountDAO) CountAllAccounts(tx *dbs.Tx, keyword string) (int64, error) {
|
|
var query = this.Query(tx)
|
|
if len(keyword) > 0 {
|
|
query.Where("userId IN (SELECT id FROM " + models.SharedUserDAO.Table + " WHERE state=1 AND (username LIKE :keyword OR fullname LIKE :keyword))")
|
|
query.Param("keyword", keyword)
|
|
} else {
|
|
query.Where("userId IN (SELECT id FROM " + models.SharedUserDAO.Table + " WHERE state=1)")
|
|
}
|
|
return query.Count()
|
|
}
|
|
|
|
// ListAccounts 列出单页账户
|
|
func (this *UserAccountDAO) ListAccounts(tx *dbs.Tx, keyword string, offset int64, size int64) (result []*UserAccount, err error) {
|
|
var query = this.Query(tx)
|
|
if len(keyword) > 0 {
|
|
query.Where("userId IN (SELECT id FROM " + models.SharedUserDAO.Table + " WHERE state=1 AND (username LIKE :keyword OR fullname LIKE :keyword))")
|
|
query.Param("keyword", keyword)
|
|
} else {
|
|
query.Where("userId IN (SELECT id FROM " + models.SharedUserDAO.Table + " WHERE state=1)")
|
|
}
|
|
_, err = query.
|
|
DescPk().
|
|
Offset(offset).
|
|
Limit(size).
|
|
Slice(&result).
|
|
FindAll()
|
|
return
|
|
}
|
|
|
|
// PayBills 尝试自动支付账单
|
|
func (this *UserAccountDAO) PayBills(tx *dbs.Tx) error {
|
|
bills, err := models.SharedUserBillDAO.FindUnpaidBills(tx, 10000)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 先支付久远的
|
|
lists.Reverse(bills)
|
|
|
|
for _, bill := range bills {
|
|
if bill.Amount <= 0 {
|
|
err = models.SharedUserBillDAO.UpdateUserBillIsPaid(tx, int64(bill.Id), true)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
continue
|
|
}
|
|
|
|
account, err := SharedUserAccountDAO.FindUserAccountWithUserId(tx, int64(bill.UserId))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if account == nil || account.Total < bill.Amount {
|
|
continue
|
|
}
|
|
|
|
// 扣款
|
|
err = SharedUserAccountDAO.UpdateUserAccount(tx, int64(account.Id), -float32(bill.Amount), userconfigs.AccountEventTypePayBill, "支付账单"+bill.Code, maps.Map{"billId": bill.Id})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 改为已支付
|
|
err = models.SharedUserBillDAO.UpdateUserBillIsPaid(tx, int64(bill.Id), true)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// CheckUserAccount 检查用户账户
|
|
func (this *UserAccountDAO) CheckUserAccount(tx *dbs.Tx, userId int64, accountId int64) error {
|
|
exists, err := this.Query(tx).
|
|
Pk(accountId).
|
|
Attr("userId", userId).
|
|
Exist()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !exists {
|
|
return models.ErrNotFound
|
|
}
|
|
return nil
|
|
}
|