84 lines
2.3 KiB
Go
84 lines
2.3 KiB
Go
|
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
|
||
|
|
}
|