diff --git a/internal/db/models/user_bill_dao.go b/internal/db/models/user_bill_dao.go index 0395eb85..31fc6201 100644 --- a/internal/db/models/user_bill_dao.go +++ b/internal/db/models/user_bill_dao.go @@ -68,6 +68,17 @@ func init() { }) } +// FindUserBill 查找单个账单 +func (this *UserBillDAO) FindUserBill(tx *dbs.Tx, billId int64) (*UserBill, error) { + one, err := this.Query(tx). + Pk(billId). + Find() + if err != nil || one == nil { + return nil, err + } + return one.(*UserBill), nil +} + // CountAllUserBills 计算账单数量 func (this *UserBillDAO) CountAllUserBills(tx *dbs.Tx, isPaid int32, userId int64, month string) (int64, error) { query := this.Query(tx) @@ -330,3 +341,33 @@ func (this *UserBillDAO) GenerateBillCode(tx *dbs.Tx) (string, error) { } return this.GenerateBillCode(tx) } + +// CheckUserBill 检查用户账单 +func (this *UserBillDAO) CheckUserBill(tx *dbs.Tx, userId int64, billId int64) error { + if userId <= 0 || billId <= 0 { + return ErrNotFound + } + exists, err := this.Query(tx). + Pk(billId). + Attr("userId", userId). + Exist() + if err != nil { + return err + } + if !exists { + return ErrNotFound + } + return nil +} + +// SumUnpaidUserBill 计算某个用户未支付总额 +func (this *UserBillDAO) SumUnpaidUserBill(tx *dbs.Tx, userId int64) (float32, error) { + sum, err := this.Query(tx). + Attr("userId", userId). + Attr("isPaid", 0). + Sum("amount", 0) + if err != nil { + return 0, err + } + return float32(sum), nil +} diff --git a/internal/rpc/services/service_user_bill.go b/internal/rpc/services/service_user_bill.go index 63e6b307..55c3a1db 100644 --- a/internal/rpc/services/service_user_bill.go +++ b/internal/rpc/services/service_user_bill.go @@ -3,8 +3,12 @@ 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/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" "regexp" ) @@ -97,3 +101,142 @@ func (this *UserBillService) ListUserBills(ctx context.Context, req *pb.ListUser } return &pb.ListUserBillsResponse{UserBills: result}, nil } + +// FindUserBill 查找账单信息 +func (this *UserBillService) FindUserBill(ctx context.Context, req *pb.FindUserBillRequest) (*pb.FindUserBillResponse, error) { + _, userId, err := this.ValidateAdminAndUser(ctx, 0, 0) + if err != nil { + return nil, err + } + + var tx = this.NullTx() + + // 检查用户 + if userId > 0 { + err = models.SharedUserBillDAO.CheckUserBill(tx, userId, req.UserBillId) + if err != nil { + return nil, err + } + } + + bill, err := models.SharedUserBillDAO.FindUserBill(tx, req.UserBillId) + if err != nil { + return nil, err + } + + if bill == nil { + return &pb.FindUserBillResponse{UserBill: nil}, nil + } + + // 用户 + 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: float32(bill.Amount), + Month: bill.Month, + IsPaid: bill.IsPaid == 1, + PaidAt: int64(bill.PaidAt), + Code: bill.Code, + }, + }, nil +} + +// PayUserBill 支付账单 +func (this *UserBillService) PayUserBill(ctx context.Context, req *pb.PayUserBillRequest) (*pb.RPCSuccess, error) { + _, userId, err := this.ValidateAdminAndUser(ctx, 0, 0) + 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 == 1 { + return nil + } + + if bill.Amount <= 0 { + // 直接修改为已支付 + return models.SharedUserBillDAO.UpdateUserBillIsPaid(tx, req.UserBillId, true) + } + + // 余额是否足够 + 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), -float32(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, 0, 0) + 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 +}