1.4.5.2
This commit is contained in:
unknown
320
EdgeAPI/internal/rpc/services/service_user_bill_plus.go
Normal file
320
EdgeAPI/internal/rpc/services/service_user_bill_plus.go
Normal file
@@ -0,0 +1,320 @@
|
||||
//go:build plus
|
||||
|
||||
package services
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
|
||||
"github.com/TeaOSLab/EdgeAPI/internal/db/models/accounts"
|
||||
"github.com/TeaOSLab/EdgeAPI/internal/errors"
|
||||
"github.com/TeaOSLab/EdgeAPI/internal/utils/regexputils"
|
||||
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
||||
"github.com/TeaOSLab/EdgeCommon/pkg/userconfigs"
|
||||
"github.com/iwind/TeaGo/dbs"
|
||||
"github.com/iwind/TeaGo/maps"
|
||||
timeutil "github.com/iwind/TeaGo/utils/time"
|
||||
"time"
|
||||
)
|
||||
|
||||
// UserBillService 账单相关服务
|
||||
type UserBillService struct {
|
||||
BaseService
|
||||
}
|
||||
|
||||
// GenerateAllUserBills 手工生成订单
|
||||
func (this *UserBillService) GenerateAllUserBills(ctx context.Context, req *pb.GenerateAllUserBillsRequest) (*pb.RPCSuccess, error) {
|
||||
_, err := this.ValidateAdmin(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 校验Month
|
||||
if !regexputils.YYYYMM.MatchString(req.Month) {
|
||||
return nil, errors.New("invalid month '" + req.Month + "'")
|
||||
}
|
||||
if req.Month >= timeutil.Format("Ym") {
|
||||
return nil, errors.New("invalid month '" + req.Month + "'")
|
||||
}
|
||||
|
||||
var tx = this.NullTx()
|
||||
priceConfig, err := models.SharedSysSettingDAO.ReadUserPriceConfig(tx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if priceConfig == nil {
|
||||
return this.Success()
|
||||
}
|
||||
|
||||
// 套餐
|
||||
if priceConfig.EnablePlans {
|
||||
err = models.SharedUserBillDAO.GenerateBillsWithPlans(tx, req.Month, priceConfig)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return this.Success()
|
||||
}
|
||||
|
||||
// 流量|带宽
|
||||
err = models.SharedUserBillDAO.GenerateTrafficAndBandwidthBills(tx, req.Month+"01", req.Month+"31", req.Month, req.Day, priceConfig)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return this.Success()
|
||||
}
|
||||
|
||||
// CountAllUserBills 计算所有账单数量
|
||||
func (this *UserBillService) CountAllUserBills(ctx context.Context, req *pb.CountAllUserBillsRequest) (*pb.RPCCountResponse, error) {
|
||||
_, userId, err := this.ValidateAdminAndUser(ctx, true)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var tx = this.NullTx()
|
||||
if userId > 0 {
|
||||
req.UserId = userId
|
||||
}
|
||||
|
||||
count, err := models.SharedUserBillDAO.CountAllUserBills(tx, req.PaidFlag, req.UserId, req.Month, req.TrafficRelated, req.MinDailyBillDays, req.MinMonthlyBillDays)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return this.SuccessCount(count)
|
||||
}
|
||||
|
||||
// ListUserBills 列出单页账单
|
||||
func (this *UserBillService) ListUserBills(ctx context.Context, req *pb.ListUserBillsRequest) (*pb.ListUserBillsResponse, error) {
|
||||
_, userId, err := this.ValidateAdminAndUser(ctx, true)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var tx = this.NullTx()
|
||||
if userId > 0 {
|
||||
req.UserId = userId
|
||||
}
|
||||
|
||||
// 计费参数设置
|
||||
var overdueDailyDay = ""
|
||||
var overdueMonthlyDay = ""
|
||||
priceConfig, err := models.SharedSysSettingDAO.ReadUserPriceConfig(tx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if priceConfig.IsOn && priceConfig.UnpaidBillPolicy.IsOn {
|
||||
if priceConfig.UnpaidBillPolicy.MinDailyBillDays > 0 {
|
||||
overdueDailyDay = timeutil.Format("Ymd", time.Now().AddDate(0, 0, -int(priceConfig.UnpaidBillPolicy.MinDailyBillDays)))
|
||||
}
|
||||
if priceConfig.UnpaidBillPolicy.MinMonthlyBillDays > 0 {
|
||||
overdueMonthlyDay = timeutil.Format("Ymd", time.Now().AddDate(0, 0, -int(priceConfig.UnpaidBillPolicy.MinMonthlyBillDays)))
|
||||
}
|
||||
}
|
||||
|
||||
bills, err := models.SharedUserBillDAO.ListUserBills(tx, req.PaidFlag, req.UserId, req.Month, req.Offset, req.Size)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var pbUserBills = []*pb.UserBill{}
|
||||
for _, bill := range bills {
|
||||
user, err := models.SharedUserDAO.FindBasicUserWithoutState(tx, int64(bill.UserId))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if user == nil {
|
||||
user = &models.User{Id: bill.UserId}
|
||||
}
|
||||
|
||||
// overdue
|
||||
var isOverdue = false
|
||||
if bill.CanPay && !bill.IsPaid && bill.Amount > 0 {
|
||||
switch bill.PricePeriod {
|
||||
case userconfigs.PricePeriodDaily:
|
||||
if len(overdueDailyDay) > 0 && len(bill.CreatedDay) > 0 && bill.CreatedDay < overdueDailyDay {
|
||||
isOverdue = true
|
||||
}
|
||||
case userconfigs.PricePeriodMonthly:
|
||||
if len(overdueMonthlyDay) > 0 && len(bill.CreatedDay) > 0 && bill.CreatedDay < overdueMonthlyDay {
|
||||
isOverdue = true
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pbUserBills = append(pbUserBills, &pb.UserBill{
|
||||
Id: int64(bill.Id),
|
||||
User: &pb.User{
|
||||
Id: int64(bill.UserId),
|
||||
Fullname: user.Fullname,
|
||||
Username: user.Username,
|
||||
IsDeleted: user.State == models.UserStateDisabled,
|
||||
},
|
||||
Type: bill.Type,
|
||||
TypeName: models.SharedUserBillDAO.BillTypeName(bill.Type),
|
||||
Description: bill.Description,
|
||||
Amount: bill.Amount,
|
||||
Month: bill.Month,
|
||||
DayFrom: bill.DayFrom,
|
||||
DayTo: bill.DayTo,
|
||||
CanPay: bill.CanPay,
|
||||
IsPaid: bill.IsPaid,
|
||||
PaidAt: int64(bill.PaidAt),
|
||||
Code: bill.Code,
|
||||
PricePeriod: bill.PricePeriod,
|
||||
IsOverdue: isOverdue,
|
||||
})
|
||||
}
|
||||
return &pb.ListUserBillsResponse{UserBills: pbUserBills}, nil
|
||||
}
|
||||
|
||||
// FindUserBill 查找账单信息
|
||||
func (this *UserBillService) FindUserBill(ctx context.Context, req *pb.FindUserBillRequest) (*pb.FindUserBillResponse, error) {
|
||||
_, userId, err := this.ValidateAdminAndUser(ctx, true)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var tx = this.NullTx()
|
||||
|
||||
var bill *models.UserBill
|
||||
if req.UserBillId > 0 {
|
||||
bill, err = models.SharedUserBillDAO.FindUserBill(tx, req.UserBillId)
|
||||
} else if len(req.Code) > 0 {
|
||||
bill, err = models.SharedUserBillDAO.FindUserBillWithCode(tx, req.Code)
|
||||
} else {
|
||||
return nil, errors.New("'userBillId' or 'code' required")
|
||||
}
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if bill == nil {
|
||||
return &pb.FindUserBillResponse{UserBill: nil}, nil
|
||||
}
|
||||
|
||||
// 检查用户权限
|
||||
if userId > 0 {
|
||||
err = models.SharedUserBillDAO.CheckUserBill(tx, userId, int64(bill.Id))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
// 用户
|
||||
var pbUser = &pb.User{Id: int64(bill.UserId)}
|
||||
user, err := models.SharedUserDAO.FindEnabledUser(tx, int64(bill.UserId), nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if user != nil {
|
||||
pbUser = &pb.User{
|
||||
Id: int64(user.Id),
|
||||
Username: user.Username,
|
||||
Fullname: user.Fullname,
|
||||
}
|
||||
}
|
||||
|
||||
return &pb.FindUserBillResponse{
|
||||
UserBill: &pb.UserBill{
|
||||
Id: int64(bill.Id),
|
||||
User: pbUser,
|
||||
Type: bill.Type,
|
||||
TypeName: models.SharedUserBillDAO.BillTypeName(bill.Type),
|
||||
Description: bill.Description,
|
||||
Amount: bill.Amount,
|
||||
Month: bill.Month,
|
||||
DayFrom: bill.DayFrom,
|
||||
DayTo: bill.DayTo,
|
||||
CanPay: bill.CanPay,
|
||||
IsPaid: bill.IsPaid,
|
||||
PaidAt: int64(bill.PaidAt),
|
||||
Code: bill.Code,
|
||||
PricePeriod: bill.PricePeriod,
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
// PayUserBill 支付账单
|
||||
func (this *UserBillService) PayUserBill(ctx context.Context, req *pb.PayUserBillRequest) (*pb.RPCSuccess, error) {
|
||||
_, userId, err := this.ValidateAdminAndUser(ctx, true)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = this.RunTx(func(tx *dbs.Tx) error {
|
||||
// 检查用户
|
||||
if userId > 0 {
|
||||
err = models.SharedUserBillDAO.CheckUserBill(tx, userId, req.UserBillId)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
// 是否存在
|
||||
bill, err := models.SharedUserBillDAO.FindUserBill(tx, req.UserBillId)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if bill == nil {
|
||||
return nil
|
||||
}
|
||||
userId = int64(bill.UserId)
|
||||
|
||||
// 是否已支付
|
||||
if bill.IsPaid {
|
||||
return nil
|
||||
}
|
||||
|
||||
if bill.Amount <= 0 {
|
||||
// 直接修改为已支付
|
||||
return models.SharedUserBillDAO.UpdateUserBillIsPaid(tx, req.UserBillId, true)
|
||||
}
|
||||
|
||||
if !bill.CanPay {
|
||||
return errors.New("can not pay now")
|
||||
}
|
||||
|
||||
// 余额是否足够
|
||||
account, err := accounts.SharedUserAccountDAO.FindUserAccountWithUserId(tx, userId)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if account == nil {
|
||||
return errors.New("can not find user account")
|
||||
}
|
||||
|
||||
if account.Total < bill.Amount {
|
||||
return errors.New("not enough balance to pay")
|
||||
}
|
||||
|
||||
err = accounts.SharedUserAccountDAO.UpdateUserAccount(tx, int64(account.Id), -bill.Amount, userconfigs.AccountEventTypePayBill, "支付账单"+bill.Code, maps.Map{"billId": bill.Id})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 修改为已支付
|
||||
return models.SharedUserBillDAO.UpdateUserBillIsPaid(tx, req.UserBillId, true)
|
||||
})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return this.Success()
|
||||
}
|
||||
|
||||
// SumUserUnpaidBills 计算用户所有未支付账单总额
|
||||
func (this *UserBillService) SumUserUnpaidBills(ctx context.Context, req *pb.SumUserUnpaidBillsRequest) (*pb.SumUserUnpaidBillsResponse, error) {
|
||||
_, userId, err := this.ValidateAdminAndUser(ctx, true)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var tx = this.NullTx()
|
||||
if userId > 0 {
|
||||
req.UserId = userId
|
||||
}
|
||||
|
||||
sum, err := models.SharedUserBillDAO.SumUnpaidUserBill(tx, userId)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &pb.SumUserUnpaidBillsResponse{Amount: sum}, nil
|
||||
}
|
||||
Reference in New Issue
Block a user