fix add 修改钱包余额

app/main/api/internal/logic/admin_agent/adminupdateagentwalletbalancelogic.go

@@ -0,0 +1,83 @@
+package admin_agent
+
+import (
+	"context"
+	"errors"
+	"fmt"
+
+	"tydata-server/app/main/api/internal/svc"
+	"tydata-server/app/main/api/internal/types"
+	"tydata-server/app/main/model"
+	"tydata-server/common/xerr"
+
+	"github.com/zeromicro/go-zero/core/logx"
+	"github.com/zeromicro/go-zero/core/stores/sqlx"
+)
+
+type AdminUpdateAgentWalletBalanceLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewAdminUpdateAgentWalletBalanceLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUpdateAgentWalletBalanceLogic {
+	return &AdminUpdateAgentWalletBalanceLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *AdminUpdateAgentWalletBalanceLogic) AdminUpdateAgentWalletBalance(req *types.AdminUpdateAgentWalletBalanceReq) (resp *types.AdminUpdateAgentWalletBalanceResp, err error) {
+	// 参数校验
+	if req.AgentId <= 0 {
+		return nil, xerr.NewErrMsg("代理ID无效")
+	}
+	if req.Amount == 0 {
+		return nil, xerr.NewErrMsg("修改金额不能为0")
+	}
+
+	// 查询代理钱包信息
+	wallet, err := l.svcCtx.AgentWalletModel.FindOneByAgentId(l.ctx, req.AgentId)
+	if err != nil {
+		if errors.Is(err, model.ErrNotFound) {
+			return nil, xerr.NewErrMsg("代理钱包不存在")
+		}
+		return nil, err
+	}
+
+	// 计算新余额
+	newBalance := wallet.Balance + req.Amount
+
+	// 校验余额不能为负数
+	if newBalance < 0 {
+		return nil, xerr.NewErrMsg(fmt.Sprintf("操作后余额不能为负数，当前余额: %.2f，操作金额: %.2f", wallet.Balance, req.Amount))
+	}
+
+	// 更新余额
+	updateErr := l.svcCtx.AgentWalletModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error {
+		// 使用版本号更新
+		wallet.Balance = newBalance
+		err := l.svcCtx.AgentWalletModel.UpdateWithVersion(transCtx, session, wallet)
+		if err != nil {
+			return err
+		}
+
+		// TODO: 这里可以添加记录余额变更日志的逻辑
+		l.Logger.Infof("代理钱包余额变更 - AgentId: %d, 原余额: %.2f, 变更金额: %.2f, 新余额: %.2f",
+			req.AgentId, wallet.Balance-req.Amount, req.Amount, newBalance)
+
+		return nil
+	})
+
+	if updateErr != nil {
+		l.Logger.Errorf("更新代理钱包余额失败: %+v", updateErr)
+		return nil, xerr.NewErrMsg("更新余额失败")
+	}
+
+	resp = &types.AdminUpdateAgentWalletBalanceResp{
+		Success: true,
+		Balance: newBalance,
+	}
+	return
+}