Znc_V2/app/main/api/internal/logic/admin_agent/adminbatchunfreezeagentcommissionlogic.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/Masterminds/squirrel"
	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
)

type AdminBatchUnfreezeAgentCommissionLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewAdminBatchUnfreezeAgentCommissionLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminBatchUnfreezeAgentCommissionLogic {
	return &AdminBatchUnfreezeAgentCommissionLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *AdminBatchUnfreezeAgentCommissionLogic) AdminBatchUnfreezeAgentCommission(req *types.AdminBatchUnfreezeAgentCommissionReq) (resp *types.AdminBatchUnfreezeAgentCommissionResp, err error) {
	// 构建查询条件：状态为1（冻结中）
	builder := l.svcCtx.AgentCommissionModel.SelectBuilder().Where(squirrel.Eq{"status": 1})

	// 如果指定了代理商ID，则只查询该代理商的冻结佣金
	if req.AgentId != nil && *req.AgentId > 0 {
		builder = builder.Where(squirrel.Eq{"agent_id": *req.AgentId})
	}

	// 查询所有冻结中的佣金记录
	commissions, err := l.svcCtx.AgentCommissionModel.FindAll(l.ctx, builder, "")
	if err != nil {
		return nil, err
	}

	// 如果没有冻结的佣金，直接返回
	if len(commissions) == 0 {
		resp = &types.AdminBatchUnfreezeAgentCommissionResp{
			Success: true,
			Count:   0,
			Amount:  0,
		}
		return
	}

	// 计算总金额
	var totalAmount float64
	for _, commission := range commissions {
		totalAmount += commission.Amount
	}

	// 开始事务
	err = l.svcCtx.AgentCommissionModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error {
		// 按代理商分组更新钱包余额
		agentWalletMap := make(map[int64]*model.AgentWallet)

		// 遍历所有冻结的佣金，更新状态
		for _, commission := range commissions {
			// 更新佣金状态为已结算
			commission.Status = 0
			err := l.svcCtx.AgentCommissionModel.UpdateWithVersion(ctx, session, commission)
			if err != nil {
				// 如果是版本冲突错误，重新查询最新的数据后重试
				if errors.Is(err, model.ErrNoRowsUpdate) {
					latestCommission, findErr := l.svcCtx.AgentCommissionModel.FindOne(ctx, commission.Id)
					if findErr != nil {
						return findErr
					}
					// 检查状态是否已被其他操作修改
					if latestCommission.Status != 1 {
						return xerr.NewErrCodeMsg(xerr.SERVER_COMMON_ERROR, fmt.Sprintf("佣金 %d 的状态已被其他操作修改，当前状态: %d", commission.Id, latestCommission.Status))
					}
					// 重新更新状态
					latestCommission.Status = 0
					updateErr := l.svcCtx.AgentCommissionModel.UpdateWithVersion(ctx, session, latestCommission)
					if updateErr != nil {
						return updateErr
					}
					// 更新引用，使用最新的数据
					commission.Version = latestCommission.Version
				} else {
					return err
				}
			}

			// 累加到对应代理商的钱包数据
			if wallet, exists := agentWalletMap[commission.AgentId]; exists {
				wallet.Balance += commission.Amount
				wallet.FrozenBalance -= commission.Amount
			} else {
				// 查询该代理商的钱包
				wallet, err := l.svcCtx.AgentWalletModel.FindOneByAgentId(ctx, commission.AgentId)
				if err != nil {
					return err
				}
				wallet.Balance += commission.Amount
				wallet.FrozenBalance -= commission.Amount
				agentWalletMap[commission.AgentId] = wallet
			}
		}

		// 更新所有受影响代理商的钱包
		for _, wallet := range agentWalletMap {
			err := l.svcCtx.AgentWalletModel.UpdateWithVersion(ctx, session, wallet)
			if err != nil {
				// 如果是版本冲突错误，重新查询最新的数据后重试
				if errors.Is(err, model.ErrNoRowsUpdate) {
					latestWallet, findErr := l.svcCtx.AgentWalletModel.FindOneByAgentId(ctx, wallet.AgentId)
					if findErr != nil {
						return findErr
					}
					// 重新累加金额
					latestWallet.Balance = wallet.Balance
					latestWallet.FrozenBalance = wallet.FrozenBalance
					updateErr := l.svcCtx.AgentWalletModel.UpdateWithVersion(ctx, session, latestWallet)
					if updateErr != nil {
						return updateErr
					}
				} else {
					return err
				}
			}
		}

		return nil
	})

	if err != nil {
		return nil, xerr.NewErrMsg("批量解冻失败: " + err.Error())
	}

	resp = &types.AdminBatchUnfreezeAgentCommissionResp{
		Success: true,
		Count:   int64(len(commissions)),
		Amount:  totalAmount,
	}
	return
}