sugarBecomer
201 lines
7.0 KiB
Go
201 lines
7.0 KiB
Go
package agent
|
|
|
|
import (
|
|
"context"
|
|
v1 "likedream-goframe/internal/api/v1"
|
|
"likedream-goframe/internal/cst"
|
|
"likedream-goframe/internal/db"
|
|
"likedream-goframe/internal/ent"
|
|
"likedream-goframe/internal/ent/sysrole"
|
|
"likedream-goframe/internal/ent/sysuser"
|
|
"likedream-goframe/internal/service/balance"
|
|
"likedream-goframe/internal/utils"
|
|
|
|
"github.com/samber/oops"
|
|
"github.com/shopspring/decimal"
|
|
)
|
|
|
|
type AgentServiceV1 struct{}
|
|
|
|
func NewV1() *AgentServiceV1 {
|
|
return &AgentServiceV1{}
|
|
}
|
|
|
|
func (s *AgentServiceV1) List(ctx context.Context, req *v1.AgentListReq) *utils.Response[[]v1.AgentListRes] {
|
|
query := db.Global.SysUser.Query()
|
|
if !utils.IsAdminFromCtx(ctx) {
|
|
query.Where(sysuser.HasParentWith(sysuser.IDEQ(utils.GetUserInfoFromCtx(ctx).ID)))
|
|
}
|
|
if req.Email != "" {
|
|
query.Where(sysuser.EmailContains(req.Email))
|
|
}
|
|
if req.Username != "" {
|
|
query.Where(sysuser.UsernameContains(req.Username))
|
|
}
|
|
query.Order(ent.Desc(sysuser.FieldCreateTime)).WithAgent()
|
|
//查询总数
|
|
total := query.CountX(ctx)
|
|
query.Limit(req.Size).Offset((req.Page - 1) * req.Size)
|
|
list := query.AllX(ctx)
|
|
res := make([]v1.AgentListRes, 0)
|
|
for _, item := range list {
|
|
one := v1.AgentListRes{
|
|
ID: item.ID,
|
|
Username: item.Username,
|
|
Email: item.Email,
|
|
InviteCode: item.InviteCode,
|
|
Level: 0,
|
|
Rate: decimal.NewFromInt(0),
|
|
CashBack: decimal.NewFromInt(0),
|
|
Balance: item.Balance,
|
|
Block: item.Block,
|
|
}
|
|
|
|
if item.Edges.Agent != nil {
|
|
one.Level = item.Edges.Agent.Level
|
|
one.Rate = item.Edges.Agent.Rate
|
|
}
|
|
res = append(res, one)
|
|
}
|
|
return utils.SuccessWithTotal[[]v1.AgentListRes](res, total)
|
|
}
|
|
|
|
func (s *AgentServiceV1) Add(ctx context.Context, req *v1.AgentAddReq) *utils.Response[any] {
|
|
// 参数校验
|
|
currentUser := utils.GetUserInfoFromCtx(ctx)
|
|
agent := currentUser.QueryAgent().OnlyX(ctx)
|
|
if agent.Level >= req.Level {
|
|
return utils.FailParam("下级代理级别不能高于当前代理级别")
|
|
}
|
|
if agent.Rate.Compare(req.Rate) <= 0 {
|
|
return utils.FailParam("下级代理费率比例不能低于当前代理比例")
|
|
}
|
|
tx, err := db.Global.Tx(ctx)
|
|
utils.OkOrPanic(err, func(err error) error {
|
|
return oops.
|
|
WithContext(ctx).
|
|
With("创建事务失败").
|
|
Wrap(err)
|
|
})
|
|
defer tx.Rollback()
|
|
user := tx.SysUser.Query().Where(sysuser.IDEQ(currentUser.ID)).OnlyX(ctx)
|
|
user.Balance = utils.CalculateBalance(user.Balance, req.Rate, req.Balance, req.Rate)
|
|
if user.Balance.LessThan(decimal.NewFromInt(0)) {
|
|
return utils.FailParam("余额不足")
|
|
}
|
|
// 扣除余额
|
|
_, err = tx.SysUser.UpdateOne(user).SetBalance(user.Balance).Save(ctx)
|
|
utils.OkOrPanic(err, func(err error) error {
|
|
return oops.
|
|
WithContext(ctx).
|
|
With("更新用户余额失败").
|
|
Wrap(err)
|
|
})
|
|
// 创建用户
|
|
tmp := tx.SysUser.Create().
|
|
SetUsername(req.Username).
|
|
SetPassword(utils.PasswordBcryptHash(req.Password)).
|
|
SetEmail(req.Email).
|
|
SetParent(user)
|
|
save := tmp.SaveX(ctx)
|
|
// 关联角色
|
|
role := tx.SysRole.Query().Where(sysrole.NameEQ(cst.KeyRoleAgent)).AllX(ctx)
|
|
tx.SysUser.UpdateOne(save).AddRoles(role...).ExecX(ctx)
|
|
// 创建代理
|
|
tx.SysAgent.Create().
|
|
SetLevel(req.Level).
|
|
SetRate(req.Rate).
|
|
SaveX(ctx)
|
|
|
|
err = tx.Commit()
|
|
return utils.SuccessWithMsg("")
|
|
}
|
|
|
|
func (s *AgentServiceV1) Ban(ctx context.Context, req *v1.AgentBanReq) *utils.Response[any] {
|
|
user := db.Global.SysUser.Query().Where(sysuser.IDEQ(req.ID)).OnlyX(ctx)
|
|
if !utils.IsAdminFromCtx(ctx) {
|
|
// 参数校验
|
|
currentUser := utils.GetUserInfoFromCtx(ctx)
|
|
// 判断是否是自己的下级代理
|
|
parent := user.QueryParent().OnlyX(ctx)
|
|
if parent.ID != currentUser.ID {
|
|
return utils.FailParam("非法的下级代理参数")
|
|
}
|
|
}
|
|
db.Global.SysUser.UpdateOne(user).SetBlock(req.Block).ExecX(ctx)
|
|
return utils.SuccessWithMsg("操作成功")
|
|
}
|
|
|
|
func (s *AgentServiceV1) Recharge(ctx context.Context, req *v1.AgentRechargeReq) *utils.Response[any] {
|
|
// 参数校验
|
|
user := db.Global.SysUser.Query().Where(sysuser.IDEQ(req.ID)).WithAgent().OnlyX(ctx)
|
|
if !utils.IsAdminFromCtx(ctx) {
|
|
currentUser := db.Global.SysUser.Query().Where(sysuser.ID(utils.GetUserInfoFromCtx(ctx).ID)).WithAgent().WithRoles().OnlyX(ctx)
|
|
// 判断是否是自己的下级代理
|
|
parent := user.QueryParent().WithAgent().FirstX(ctx)
|
|
if parent == nil || parent.ID != currentUser.ID {
|
|
return utils.FailParam("非法的下级代理参数")
|
|
}
|
|
if !utils.HasRole(currentUser.Edges.Roles, cst.KeyRoleAdmin) {
|
|
needBalance := utils.CalculateBalance(parent.Balance, parent.Edges.Agent.Rate, req.Amount, user.Edges.Agent.Rate)
|
|
balance.BalanceService.ChangeUserBalance(ctx, parent.ID, needBalance.Neg())
|
|
}
|
|
}
|
|
balance.BalanceService.ChangeUserBalance(ctx, user.ID, req.Amount)
|
|
return utils.SuccessWithMsg("操作成功")
|
|
}
|
|
|
|
func (s *AgentServiceV1) AgentPutInfoReq(ctx context.Context, req *v1.AgentPutInfoReq) *utils.Response[any] {
|
|
// 1. 检测用户是否是自己下级
|
|
tx, err := db.Global.Tx(ctx)
|
|
utils.OkOrPanic(err, func(err error) error {
|
|
return oops.
|
|
WithContext(ctx).
|
|
With("创建事务失败").
|
|
Wrap(err)
|
|
})
|
|
defer tx.Rollback()
|
|
user := tx.SysUser.Query().Where(sysuser.ID(req.ID)).WithParent().WithAgent().FirstX(ctx)
|
|
if !utils.IsAdminFromCtx(ctx) {
|
|
currentUser := tx.SysUser.Query().Where(sysuser.ID(utils.GetUserInfoFromCtx(ctx).ID)).WithAgent().WithRoles().OnlyX(ctx)
|
|
if user.Edges.Parent == nil || user.Edges.Parent.ID != currentUser.ID {
|
|
return utils.FailParam("非法的下级代理参数")
|
|
}
|
|
// 2. 如果没有代理权限
|
|
if !utils.HasRole(currentUser.Edges.Roles, cst.KeyRoleAdmin) && !utils.HasRole(currentUser.Edges.Roles, cst.KeyRoleAgent) {
|
|
return utils.FailParam("代理权限不足")
|
|
}
|
|
// 3. 检测代理费率或level是否低于自己
|
|
if currentUser.Edges.Agent != nil && (req.Level > currentUser.Edges.Agent.Level || req.Rate.LessThan(currentUser.Edges.Agent.Rate)) {
|
|
return utils.FailParam("代理等级或费率低于自己")
|
|
}
|
|
}
|
|
// 5. 检测是否存在代理信息，
|
|
// 将用户余额先计算为比例为1的用户余额
|
|
oldBalance := user.Balance
|
|
if user.Edges.Agent == nil {
|
|
// 创建代理信息
|
|
tx.SysAgent.Create().SetLevel(req.Level).SetRate(req.Rate).SetUsers(user).SaveX(ctx)
|
|
// 设置用户角色
|
|
userRole := tx.SysRole.Query().Where(sysrole.LabelEQ(cst.KeyRoleUser)).OnlyX(ctx)
|
|
agentRole := tx.SysRole.Query().Where(sysrole.LabelEQ(cst.KeyRoleAgent)).OnlyX(ctx)
|
|
tx.SysUser.UpdateOne(user).RemoveRoles(userRole).AddRoles(agentRole).ExecX(ctx)
|
|
} else {
|
|
// 按照费率折算用户余额
|
|
oldBalance = utils.RevertBalance(user.Balance, user.Edges.Agent.Rate)
|
|
tx.SysAgent.UpdateOne(user.Edges.Agent).SetLevel(req.Level).SetRate(req.Rate).SaveX(ctx)
|
|
}
|
|
//按照新费率计算余额
|
|
newBalance := utils.TransformBalance(oldBalance, req.Rate)
|
|
// 6. 更新用户余额
|
|
balance.BalanceService.ChangeUserBalance(ctx, user.ID, newBalance.Sub(oldBalance))
|
|
err = tx.Commit()
|
|
utils.OkOrPanic(err, func(err error) error {
|
|
return oops.
|
|
WithContext(ctx).
|
|
With("更新代理信息失败").
|
|
Wrap(err)
|
|
})
|
|
return utils.SuccessWithMsg("操作成功")
|
|
}
|