🐛 代理信息修改

internal/controller/agent/agent_ctrlV1.go

@@ -23,7 +23,7 @@ func (a AgentCtrlV1) List(ctx context.Context, req *v1.AgentListReq) (res *utils
 // Add 添加下级代理
 func (a AgentCtrlV1) Add(ctx context.Context, req *v1.AgentAddReq) (res *utils.Response[any], err error) {
 	defer utils.Pass(&err)
-	return service.Global.Agent.Add(ctx, req), nil
+	panic("implement me")
 }
 
 // Ban 禁用代理

internal/ent/schema/sys_user.go

@@ -46,14 +46,14 @@ func (SysUser) Edges() []ent.Edge {
 		edge.To("children", SysUser.Type).
 			From("parent").
 			Unique(),
-		edge.To("roles", SysRole.Type),                                               // 用户拥有角色
-		edge.To("agent", SysAgent.Type).Unique(),                                     // 用户拥有代理
-		edge.To("images", CpImage.Type).Annotations(entsql.OnDelete(entsql.Cascade)), // 用户拥有的图片
-		edge.To("cpuser", CpUser.Type).Annotations(entsql.OnDelete(entsql.Cascade)),  // 用户拥有的cp用户
-		edge.To("cp_task", CpTask.Type),
-		edge.To("cp_task_logs", CpTaskLog.Type), // 用户拥有的cp任务日志
-		edge.To("created_cards", SysCard.Type),  // 用户创建的卡
-		edge.To("used_cards", SysCard.Type),     // 用户使用的卡
+		edge.To("roles", SysRole.Type).Annotations(entsql.OnDelete(entsql.Cascade)),           // 用户拥有角色
+		edge.To("agent", SysAgent.Type).Unique().Annotations(entsql.OnDelete(entsql.Cascade)), // 用户拥有代理
+		edge.To("images", CpImage.Type).Annotations(entsql.OnDelete(entsql.Cascade)),          // 用户拥有的图片
+		edge.To("cpuser", CpUser.Type).Annotations(entsql.OnDelete(entsql.Cascade)),           // 用户拥有的cp用户
+		edge.To("cp_task", CpTask.Type).Annotations(entsql.OnDelete(entsql.Cascade)),
+		edge.To("cp_task_logs", CpTaskLog.Type).Annotations(entsql.OnDelete(entsql.Cascade)), // 用户拥有的cp任务日志
+		edge.To("created_cards", SysCard.Type),                                               // 用户创建的卡
+		edge.To("used_cards", SysCard.Type),                                                  // 用户使用的卡
 		edge.To("cashback_cards", SysCard.Type),
 	}
 }

internal/service/agent/agent_service.go

@@ -60,57 +60,6 @@ func (s *AgentServiceV1) List(ctx context.Context, req *v1.AgentListReq) *utils.
 	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) {
@@ -127,19 +76,22 @@ func (s *AgentServiceV1) Ban(ctx context.Context, req *v1.AgentBanReq) *utils.Re
 }
 
 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)
+		currentUser := utils.GetUserInfoFromCtx(ctx)
+		// 参数校验
+		if user.Edges.Agent == nil {
+			return utils.FailParam("用户不是代理")
+		}
 		// 判断是否是自己的下级代理
 		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())
-		}
+		needBalance := utils.NeedBalance(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("操作成功")
@@ -185,11 +137,13 @@ func (s *AgentServiceV1) AgentPutInfoReq(ctx context.Context, req *v1.AgentPutIn
 		oldBalance = utils.RevertBalance(user.Balance, user.Edges.Agent.Rate)
 		tx.SysAgent.UpdateOne(user.Edges.Agent).SetLevel(req.Level).SetRate(req.Rate).SaveX(ctx)
 	}
+	err = tx.Commit()
 	//按照新费率计算余额
 	newBalance := utils.TransformBalance(oldBalance, req.Rate)
-	// 6. 更新用户余额
-	balance.BalanceService.ChangeUserBalance(ctx, user.ID, newBalance.Sub(oldBalance))
-	err = tx.Commit()
+	// 6. 更新用户余额(如果更新了费率或者第一次改为代理)
+	if user.Edges.Agent == nil || !user.Edges.Agent.Rate.Equal(req.Rate) {
+		balance.BalanceService.ChangeUserBalance(ctx, user.ID, newBalance.Sub(user.Balance))
+	}
 	utils.OkOrPanic(err, func(err error) error {
 		return oops.
 			WithContext(ctx).

internal/service/balance/balance_service.go

@@ -4,33 +4,40 @@ import (
 	"context"
 	"likedream-goframe/internal/common/errors"
 	"likedream-goframe/internal/db"
+	"likedream-goframe/internal/ent"
 	"likedream-goframe/internal/ent/sysuser"
 	"likedream-goframe/internal/utils"
-	"sync"
 
-	"github.com/samber/oops"
 	"github.com/shopspring/decimal"
 )
 
-var BalanceService = &BalanceServiceV1{
-	lock: sync.Mutex{},
-}
+var BalanceService = &BalanceServiceV1{}
 
 type BalanceServiceV1 struct {
-	lock sync.Mutex
 }
 
-func (v *BalanceServiceV1) ChangeUserBalance(ctx context.Context, userId int64, balance decimal.Decimal) {
-	v.lock.Lock()
-	defer v.lock.Unlock()
-	tx, err := db.Global.Tx(ctx)
-	utils.OkOrPanic(err, oops.WithContext(ctx).Wrap)
-	defer tx.Rollback()
-	user := tx.SysUser.Query().Where(sysuser.ID(userId)).WithRoles().OnlyX(ctx)
-	newBalance := user.Balance.Add(balance)
-	if newBalance.LessThan(decimal.Zero) {
-		utils.OkOrPanic(errors.NewBusinessError(400, "余额不足"))
-	}
-	tx.SysUser.Update().Where(sysuser.ID(userId)).SetBalance(newBalance).SaveX(ctx)
-	tx.Commit()
+func (v *BalanceServiceV1) ChangeUserBalance(ctx context.Context, userId int64, balance decimal.Decimal) error {
+	return db.WithTx(ctx, func(c context.Context, tx *ent.Tx) {
+		// 使用 SELECT ... FOR UPDATE 锁定用户记录
+		user, err := tx.SysUser.Query().
+			Where(sysuser.ID(userId)).
+			ForUpdate().
+			Only(ctx)
+		if err != nil {
+			utils.OkOrPanic(err)
+		}
+
+		newBalance := user.Balance.Add(balance)
+		if newBalance.LessThan(decimal.Zero) {
+			utils.OkOrPanic(errors.NewBusinessError(400, "余额不足"))
+		}
+
+		// 更新余额
+		tx.SysUser.Update().
+			Where(sysuser.ID(userId)).
+			SetBalance(newBalance).
+			ExecX(ctx)
+		tx.Commit()
+	}, nil)
+
 }

internal/service/user_login/user_login_service.go

@@ -81,19 +81,18 @@ func (s *UserLoginServiceV1) Register(ctx context.Context, req *v1.UserRegisterR
 			With("提交事务失败").
 			Wrap(err)
 	})
-
 	// 注册成功后发送欢迎邮件（异步发送，不影响注册结果）
 	go func() {
+		ctx2 := context.Background()
 		token := make([]byte, 32)
 		_, _ = rand.Read(token)
 		tokenStr := base64.URLEncoding.EncodeToString(token)
-		set := redis.GetClient().Set(ctx, cst.RedisVerifyCode+tokenStr, req.Email, time.Duration(cst.RedisVerifyCodeExpire)*time.Second)
-		err2 := set.Err()
-		utils.OkOrPanic(err2)
+		err2 := redis.GetClient().SetEx(ctx2, cst.RedisVerifyCode+tokenStr, req.Email, time.Duration(cst.RedisVerifyCodeExpire)*time.Second).Err()
+		utils.OkOrPanic(err2, oops.WithContext(ctx2).Wrap)
 		// 创建新的context避免原context被取消
-		emailCtx := context.Background()
 		emailService := email.NewV1()
-		emailService.SendSignupEmail(emailCtx, req.Email, req.Username, tokenStr)
+		emailService.SendSignupEmail(ctx2, req.Email, req.Username, tokenStr)
+
 	}()
 
 	return utils.SuccessWithMsg("注册成功")

internal/utils/user.go

@@ -28,6 +28,14 @@ func CalculateBalance(balance decimal.Decimal, rate decimal.Decimal, amount deci
 	return balance.Sub(amount.Div(amountRate).Mul(rate)).Truncate(2)
 }
 
+func NeedBalance(rate decimal.Decimal, amount decimal.Decimal, amountRate decimal.Decimal) decimal.Decimal {
+	// 如果当前级别费率为0(超级管理员)
+	if rate.Equal(decimal.NewFromInt(0)) {
+		return decimal.Zero
+	}
+	return amount.Mul(rate).Div(amountRate)
+}
+
 // ConsumeBalance 消费余额
 func ConsumeBalance(balance decimal.Decimal, rate decimal.Decimal) decimal.Decimal {
 	return balance.Mul(rate)