136 lines
4.1 KiB
Go
136 lines
4.1 KiB
Go
package agent
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"ycc-server/app/main/model"
|
|
"ycc-server/common/ctxdata"
|
|
"ycc-server/common/xerr"
|
|
"ycc-server/pkg/lzkit/lzUtils"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
|
|
"ycc-server/app/main/api/internal/svc"
|
|
"ycc-server/app/main/api/internal/types"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
type ApplyUpgradeLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewApplyUpgradeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ApplyUpgradeLogic {
|
|
return &ApplyUpgradeLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *ApplyUpgradeLogic) ApplyUpgrade(req *types.ApplyUpgradeReq) (resp *types.ApplyUpgradeResp, err error) {
|
|
userID, err := ctxdata.GetUidFromCtx(l.ctx)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err)
|
|
}
|
|
|
|
// 1. 获取代理信息
|
|
agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID)
|
|
if err != nil {
|
|
if errors.Is(err, model.ErrNotFound) {
|
|
return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "")
|
|
}
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err)
|
|
}
|
|
|
|
fromLevel := agent.Level
|
|
toLevel := req.ToLevel
|
|
|
|
// 2. 验证升级条件
|
|
if !l.canUpgrade(agent.Level, toLevel, 1) {
|
|
return nil, errors.Wrapf(xerr.NewErrMsg("升级条件不满足"), "")
|
|
}
|
|
|
|
// 3. 计算升级费用和返佣
|
|
upgradeFee := l.svcCtx.AgentService.GetUpgradeFee(fromLevel, toLevel)
|
|
rebateAmount := l.svcCtx.AgentService.GetUpgradeRebate(fromLevel, toLevel)
|
|
|
|
// 4. 查找原直接上级（用于返佣）
|
|
var rebateAgentId int64
|
|
parent, err := l.svcCtx.AgentService.FindDirectParent(l.ctx, agent.Id)
|
|
if err != nil && !errors.Is(err, model.ErrNotFound) {
|
|
return nil, errors.Wrapf(err, "查找直接上级失败")
|
|
}
|
|
if parent != nil {
|
|
rebateAgentId = parent.Id
|
|
}
|
|
|
|
// 5. 使用事务处理升级
|
|
var upgradeId int64
|
|
err = l.svcCtx.AgentWalletModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error {
|
|
// 5.1 创建升级记录
|
|
upgradeRecord := &model.AgentUpgrade{
|
|
AgentId: agent.Id,
|
|
FromLevel: fromLevel,
|
|
ToLevel: toLevel,
|
|
UpgradeType: 1, // 自主付费
|
|
UpgradeFee: upgradeFee,
|
|
RebateAmount: rebateAmount,
|
|
Status: 1, // 待处理
|
|
}
|
|
if rebateAgentId > 0 {
|
|
upgradeRecord.RebateAgentId = sql.NullInt64{Int64: rebateAgentId, Valid: true}
|
|
}
|
|
|
|
upgradeResult, err := l.svcCtx.AgentUpgradeModel.Insert(transCtx, session, upgradeRecord)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "创建升级记录失败")
|
|
}
|
|
upgradeId, _ = upgradeResult.LastInsertId()
|
|
|
|
// 5.2 处理支付（这里假设支付已在外层处理，只记录订单号）
|
|
// 实际支付应该在创建升级记录之前完成
|
|
// 注意：支付订单号需要从支付回调中获取，这里暂时留空
|
|
|
|
// 5.3 执行升级操作
|
|
if err := l.svcCtx.AgentService.ProcessUpgrade(transCtx, agent.Id, toLevel, 1, upgradeFee, rebateAmount, "", 0); err != nil {
|
|
return errors.Wrapf(err, "执行升级操作失败")
|
|
}
|
|
|
|
// 5.4 更新升级记录状态
|
|
upgradeRecord.Id = upgradeId
|
|
upgradeRecord.Status = 2 // 已完成
|
|
upgradeRecord.Remark = lzUtils.StringToNullString("升级成功")
|
|
if err := l.svcCtx.AgentUpgradeModel.UpdateWithVersion(transCtx, session, upgradeRecord); err != nil {
|
|
return errors.Wrapf(err, "更新升级记录失败")
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 返回响应（订单号需要从支付回调中获取，这里暂时返回空）
|
|
return &types.ApplyUpgradeResp{
|
|
UpgradeId: upgradeId,
|
|
OrderNo: "", // 需要从支付回调中获取
|
|
}, nil
|
|
}
|
|
|
|
// canUpgrade 检查是否可以升级
|
|
func (l *ApplyUpgradeLogic) canUpgrade(fromLevel, toLevel int64, upgradeType int64) bool {
|
|
if upgradeType == 1 { // 自主付费
|
|
if fromLevel == 1 { // 普通
|
|
return toLevel == 2 || toLevel == 3 // 可以升级为黄金或钻石
|
|
} else if fromLevel == 2 { // 黄金
|
|
return toLevel == 3 // 可以升级为钻石
|
|
}
|
|
}
|
|
return false
|
|
}
|