163 lines
5.9 KiB
Go
163 lines
5.9 KiB
Go
package agent
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"fmt"
|
||
"qnc-server/common/ctxdata"
|
||
"qnc-server/common/xerr"
|
||
"time"
|
||
|
||
"github.com/pkg/errors"
|
||
|
||
"qnc-server/app/user/cmd/api/internal/svc"
|
||
"qnc-server/app/user/cmd/api/internal/types"
|
||
"qnc-server/app/user/model"
|
||
|
||
"github.com/zeromicro/go-zero/core/logx"
|
||
)
|
||
|
||
type ActivateAgentMembershipLogic struct {
|
||
logx.Logger
|
||
ctx context.Context
|
||
svcCtx *svc.ServiceContext
|
||
}
|
||
|
||
func NewActivateAgentMembershipLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ActivateAgentMembershipLogic {
|
||
return &ActivateAgentMembershipLogic{
|
||
Logger: logx.WithContext(ctx),
|
||
ctx: ctx,
|
||
svcCtx: svcCtx,
|
||
}
|
||
}
|
||
func (l *ActivateAgentMembershipLogic) ActivateAgentMembership(req *types.AgentActivateMembershipReq) (resp *types.AgentActivateMembershipResp, err error) {
|
||
userID, err := ctxdata.GetUidFromCtx(l.ctx)
|
||
if err != nil {
|
||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户ID失败: %v", err)
|
||
}
|
||
// 查询用户代理信息
|
||
agentModel, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID)
|
||
if err != nil {
|
||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败: %v", err)
|
||
}
|
||
// 定义等级顺序映射
|
||
levelOrder := map[string]int{
|
||
"": 1,
|
||
model.AgentLeveNameNormal: 1,
|
||
model.AgentLeveNameVIP: 2,
|
||
model.AgentLeveNameSVIP: 3,
|
||
}
|
||
|
||
// 验证请求等级合法性
|
||
if _, valid := levelOrder[req.Type]; !valid {
|
||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "无效的代理等级: %s", req.Type)
|
||
}
|
||
|
||
// 如果存在代理记录，进行等级验证
|
||
if agentModel != nil {
|
||
currentLevel, exists := levelOrder[agentModel.LevelName]
|
||
if !exists {
|
||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR),
|
||
"非法的当前代理等级: %s", agentModel.LevelName)
|
||
}
|
||
|
||
requestedLevel := levelOrder[req.Type]
|
||
if requestedLevel < currentLevel {
|
||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR),
|
||
"禁止降级操作（当前等级：%s，请求等级：%s）", agentModel.LevelName, req.Type)
|
||
}
|
||
}
|
||
outTradeNo := "A_" + l.svcCtx.AlipayService.GenerateOutTradeNo()
|
||
redisKey := fmt.Sprintf(types.AgentVipCacheKey, userID, outTradeNo)
|
||
agentVipCache := types.AgentVipCache{Type: req.Type}
|
||
jsonData, err := json.Marshal(agentVipCache)
|
||
if err != nil {
|
||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "序列化代理VIP缓存失败: %v", err)
|
||
}
|
||
cacheErr := l.svcCtx.Redis.SetexCtx(l.ctx, redisKey, string(jsonData), int(2*time.Hour))
|
||
if cacheErr != nil {
|
||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "设置缓存失败: %v", cacheErr)
|
||
}
|
||
return &types.AgentActivateMembershipResp{
|
||
Id: outTradeNo,
|
||
}, nil
|
||
}
|
||
|
||
// func (l *ActivateAgentMembershipLogic) ActivateAgentMembership(req *types.AgentActivateMembershipReq) (resp *types.AgentActivateMembershipResp, err error) {
|
||
// //userID, err := ctxdata.GetUidFromCtx(l.ctx)
|
||
// //if err != nil {
|
||
// // return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户ID失败: %v", err)
|
||
// //}
|
||
// secretKey := l.svcCtx.Config.Encrypt.SecretKey
|
||
// encryptedMobile, err := crypto.EncryptMobile(req.Mobile, secretKey)
|
||
// if err != nil {
|
||
// return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "加密手机号失败: %v", err)
|
||
// }
|
||
// userModel, err := l.svcCtx.UserModel.FindOneByMobile(l.ctx, encryptedMobile)
|
||
// if err != nil {
|
||
// return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败: %v", err)
|
||
// }
|
||
// // 查询用户代理信息
|
||
// agentModel, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userModel.Id)
|
||
// if err != nil && err != sql.ErrNoRows {
|
||
// return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败: %v", err)
|
||
// }
|
||
|
||
// // 定义等级顺序映射
|
||
// levelOrder := map[string]int{
|
||
// "": 1,
|
||
// model.AgentLeveNameNormal: 1,
|
||
// model.AgentLeveNameVIP: 2,
|
||
// model.AgentLeveNameSVIP: 3,
|
||
// }
|
||
|
||
// // 验证请求等级合法性
|
||
// if _, valid := levelOrder[req.Type]; !valid {
|
||
// return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "无效的代理等级: %s", req.Type)
|
||
// }
|
||
|
||
// // 如果存在代理记录，进行等级验证
|
||
// if agentModel != nil {
|
||
// currentLevel, exists := levelOrder[agentModel.LevelName]
|
||
// if !exists {
|
||
// return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR),
|
||
// "非法的当前代理等级: %s", agentModel.LevelName)
|
||
// }
|
||
|
||
// requestedLevel := levelOrder[req.Type]
|
||
// if requestedLevel < currentLevel {
|
||
// return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR),
|
||
// "禁止降级操作（当前等级：%s，请求等级：%s）", agentModel.LevelName, req.Type)
|
||
// }
|
||
// }
|
||
|
||
// err = l.svcCtx.AgentModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error {
|
||
// agentModel.LevelName = req.Type
|
||
// agentModel.MembershipExpiryTime = RenewMembership(agentModel.MembershipExpiryTime)
|
||
// transErr := l.svcCtx.AgentModel.UpdateWithVersion(transCtx, session, agentModel)
|
||
// if transErr != nil {
|
||
// return transErr
|
||
// }
|
||
// agentMembershipRechargeOrder := model.AgentMembershipRechargeOrder{
|
||
// AgentId: agentModel.Id,
|
||
// UserId: userModel.Id,
|
||
// LevelName: req.Type,
|
||
// Amount: req.Amount,
|
||
// PaymentMethod: req.PaymentMethod,
|
||
// TransactionId: req.TransactionId,
|
||
// }
|
||
// _, transErr = l.svcCtx.AgentMembershipRechargeOrderModel.Insert(transCtx, session, &agentMembershipRechargeOrder)
|
||
// if transErr != nil {
|
||
// return transErr
|
||
// }
|
||
// return nil
|
||
// })
|
||
// if err != nil {
|
||
// return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "升级代理等级失败: %s", req.Type)
|
||
// }
|
||
// return &types.AgentActivateMembershipResp{
|
||
// MembershipType: req.Type,
|
||
// ExpireTime: agentModel.MembershipExpiryTime.Time.Format("2006-01-02 15:04:05"),
|
||
// }, nil
|
||
// }
|