恢复余额功能 & 管理员修改余额功能

web/services/user.go

@@ -28,12 +28,28 @@ func (s *userService) Get(q *q.Query, uid int32) (*m.User, error) {
 	return user, nil
 }
 
-func (s *userService) UpdateBalance(q *q.Query, user *m.User, amount decimal.Decimal) error {
+func (s *userService) UpdateBalanceByAdmin(user *m.User, newBalance decimal.Decimal, adminId *int32) error {
+	if user == nil {
+		return core.NewServErr("用户不存在")
+	}
+
+	amount := newBalance.Sub(user.Balance)
+	if amount.IsZero() {
+		return nil
+	}
+
+	return q.Q.Transaction(func(q *q.Query) error {
+		return s.UpdateBalance(q, user, amount, "管理员修改余额", adminId)
+	})
+}
+
+func (s *userService) UpdateBalance(q *q.Query, user *m.User, amount decimal.Decimal, remark string, adminId *int32) error {
 	balance := user.Balance.Add(amount)
 	if balance.IsNegative() {
 		return core.NewServErr("用户余额不足")
 	}
 
+	// 更新余额
 	_, err := q.User.
 		Where(
 			q.User.ID.Eq(user.ID),
@@ -46,6 +62,19 @@ func (s *userService) UpdateBalance(q *q.Query, user *m.User, amount decimal.Dec
 		return core.NewServErr("更新用户余额失败", err)
 	}
 
+	// 新增动账记录
+	err = q.BalanceActivity.Create(&m.BalanceActivity{
+		UserID:      user.ID,
+		AdminID:     adminId,
+		Amount:      amount.StringFixed(2),
+		BalancePrev: user.Balance.StringFixed(2),
+		BalanceCurr: balance.StringFixed(2),
+		Remark:      &remark,
+	})
+	if err != nil {
+		return core.NewServErr("新增动账记录失败", err)
+	}
+
 	return nil
 }