add agent vip activateagent

app/user/cmd/api/internal/logic/agent/activateagentmembershiplogic.go

@@ -2,17 +2,17 @@ package agent
 
 import (
 	"context"
-	"database/sql"
-	"qnc-server/app/user/model"
+	"encoding/json"
+	"fmt"
+	"qnc-server/common/ctxdata"
 	"qnc-server/common/xerr"
-	"qnc-server/pkg/lzkit/crypto"
 	"time"
 
 	"github.com/pkg/errors"
-	"github.com/zeromicro/go-zero/core/stores/sqlx"
 
 	"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"
 )
@@ -30,27 +30,16 @@ func NewActivateAgentMembershipLogic(ctx context.Context, svcCtx *svc.ServiceCon
 		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)
-	//}
-	secretKey := l.svcCtx.Config.Encrypt.SecretKey
-	encryptedMobile, err := crypto.EncryptMobile(req.Mobile, secretKey)
+	userID, err := ctxdata.GetUidFromCtx(l.ctx)
 	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)
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户ID失败: %v", err)
 	}
 	// 查询用户代理信息
-	agentModel, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userModel.Id)
-	if err != nil && err != sql.ErrNoRows {
+	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,
@@ -78,51 +67,96 @@ func (l *ActivateAgentMembershipLogic) ActivateAgentMembership(req *types.AgentA
 				"禁止降级操作（当前等级：%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
-	})
+	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), "升级代理等级失败: %s", req.Type)
+		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{
-		MembershipType: req.Type,
-		ExpireTime:     agentModel.MembershipExpiryTime.Time.Format("2006-01-02 15:04:05"),
+		Id: outTradeNo,
 	}, nil
 }
-func RenewMembership(expiry sql.NullTime) sql.NullTime {
-	// 确定基准时间
-	var baseTime time.Time
-	if expiry.Valid {
-		baseTime = expiry.Time
-	} else {
-		baseTime = time.Now()
-	}
 
-	// 增加一年（自动处理闰年）
-	newTime := baseTime.AddDate(1, 0, 0)
+// 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)
+// 	}
 
-	// 返回始终有效的 NullTime
-	return sql.NullTime{
-		Time:  newTime,
-		Valid: true,
-	}
-}
+// 	// 定义等级顺序映射
+// 	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
+//	}