573 lines
22 KiB
Go
573 lines
22 KiB
Go
package agent
|
||
|
||
import (
|
||
"context"
|
||
"database/sql"
|
||
"fmt"
|
||
"os"
|
||
"strconv"
|
||
"time"
|
||
"ycc-server/app/main/model"
|
||
"ycc-server/common/ctxdata"
|
||
"ycc-server/common/globalkey"
|
||
"ycc-server/common/xerr"
|
||
"ycc-server/pkg/lzkit/crypto"
|
||
|
||
"github.com/google/uuid"
|
||
"github.com/pkg/errors"
|
||
"github.com/zeromicro/go-zero/core/stores/redis"
|
||
"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 RegisterByInviteCodeLogic struct {
|
||
logx.Logger
|
||
ctx context.Context
|
||
svcCtx *svc.ServiceContext
|
||
}
|
||
|
||
func NewRegisterByInviteCodeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RegisterByInviteCodeLogic {
|
||
return &RegisterByInviteCodeLogic{
|
||
Logger: logx.WithContext(ctx),
|
||
ctx: ctx,
|
||
svcCtx: svcCtx,
|
||
}
|
||
}
|
||
|
||
func (l *RegisterByInviteCodeLogic) RegisterByInviteCode(req *types.RegisterByInviteCodeReq) (resp *types.RegisterByInviteCodeResp, err error) {
|
||
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)
|
||
}
|
||
|
||
// 校验验证码（开发环境下跳过验证码校验）
|
||
if os.Getenv("ENV") != "development" {
|
||
redisKey := fmt.Sprintf("%s:%s", "agentApply", encryptedMobile)
|
||
cacheCode, err := l.svcCtx.Redis.Get(redisKey)
|
||
if err != nil {
|
||
if errors.Is(err, redis.Nil) {
|
||
return nil, errors.Wrapf(xerr.NewErrMsg("验证码已过期"), "")
|
||
}
|
||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "读取验证码失败, %v", err)
|
||
}
|
||
if cacheCode != req.Code {
|
||
return nil, errors.Wrapf(xerr.NewErrMsg("验证码不正确"), "")
|
||
}
|
||
}
|
||
|
||
// 获取当前登录态（可能为空）
|
||
claims, err := ctxdata.GetClaimsFromCtx(l.ctx)
|
||
if err != nil && !errors.Is(err, ctxdata.ErrNoInCtx) {
|
||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败: %v", err)
|
||
}
|
||
|
||
// 前置检查：如果当前用户是正式用户（有手机号），进行拦截检查
|
||
if claims != nil {
|
||
currentUser, err := l.svcCtx.UserModel.FindOne(l.ctx, claims.UserId)
|
||
if err != nil && !errors.Is(err, model.ErrNotFound) {
|
||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询用户失败: %v", err)
|
||
}
|
||
if currentUser != nil && currentUser.Mobile.Valid && currentUser.Mobile.String != "" {
|
||
// 当前用户是正式用户，检查是否已是代理
|
||
agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, claims.UserId)
|
||
if err != nil && !errors.Is(err, model.ErrNotFound) {
|
||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理失败: %v", err)
|
||
}
|
||
if agent != nil {
|
||
return nil, errors.Wrapf(xerr.NewErrMsg("您已经是代理，不能重复注册"), "")
|
||
}
|
||
// 正式用户手机号必须匹配
|
||
if currentUser.Mobile.String != encryptedMobile {
|
||
return nil, errors.Wrapf(xerr.NewErrMsg("请输入当前账号的手机号码"), "")
|
||
}
|
||
}
|
||
}
|
||
|
||
// 验证邀请码是否有效
|
||
var inviteCodeModel *model.AgentInviteCode
|
||
inviteCodeModel, err = l.svcCtx.AgentInviteCodeModel.FindOneByCode(l.ctx, req.Referrer)
|
||
if err != nil && !errors.Is(err, model.ErrNotFound) {
|
||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询邀请码失败, %v", err)
|
||
}
|
||
if inviteCodeModel != nil {
|
||
if inviteCodeModel.Status != 0 {
|
||
if inviteCodeModel.Status == 1 {
|
||
return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已使用"), "")
|
||
}
|
||
return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已失效"), "")
|
||
}
|
||
if inviteCodeModel.ExpireTime.Valid && inviteCodeModel.ExpireTime.Time.Before(time.Now()) {
|
||
return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已过期"), "")
|
||
}
|
||
}
|
||
|
||
// 使用事务处理注册
|
||
var userID string
|
||
var agentID string
|
||
var agentLevel int64
|
||
|
||
err = l.svcCtx.AgentInviteCodeModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error {
|
||
// 1. 查找目标用户（通过手机号）
|
||
targetUser, findUserErr := l.svcCtx.UserModel.FindOneByMobile(transCtx, sql.NullString{String: encryptedMobile, Valid: true})
|
||
if findUserErr != nil && !errors.Is(findUserErr, model.ErrNotFound) {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询用户失败, %v", findUserErr)
|
||
}
|
||
|
||
// 2. 获取当前登录态信息
|
||
var currentUserID string
|
||
var currentAuthType string
|
||
var currentAuthKey string
|
||
if claims != nil {
|
||
currentUserID = claims.UserId
|
||
currentAuthType = claims.AuthType
|
||
currentAuthKey = claims.AuthKey
|
||
}
|
||
|
||
// 3. 根据目标用户是否存在，处理用户和认证
|
||
if targetUser == nil {
|
||
// 场景1: 手机号不存在
|
||
userID, err = l.handleMobileNotExists(transCtx, session, encryptedMobile, currentUserID)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
} else {
|
||
// 场景2: 手机号已存在
|
||
userID, err = l.handleMobileExists(transCtx, session, targetUser, currentUserID, currentAuthType, currentAuthKey)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
// 4. 处理邀请码和上级关系
|
||
var targetLevel int64
|
||
var parentAgentId string
|
||
if inviteCodeModel != nil {
|
||
targetLevel = inviteCodeModel.TargetLevel
|
||
if inviteCodeModel.AgentId.Valid {
|
||
parentAgentId = inviteCodeModel.AgentId.String
|
||
}
|
||
} else {
|
||
if codeVal, parseErr := strconv.ParseInt(req.Referrer, 10, 64); parseErr == nil && codeVal > 0 {
|
||
parentAgent, err := l.findAgentByCode(transCtx, codeVal)
|
||
if err != nil {
|
||
return errors.Wrapf(err, "")
|
||
}
|
||
parentAgentId = parentAgent.Id
|
||
targetLevel = 1
|
||
} else {
|
||
encRefMobile, _ := crypto.EncryptMobile(req.Referrer, l.svcCtx.Config.Encrypt.SecretKey)
|
||
agents, findErr := l.svcCtx.AgentModel.FindAll(transCtx, l.svcCtx.AgentModel.SelectBuilder().Where("mobile = ? AND del_state = ?", encRefMobile, globalkey.DelStateNo).Limit(1), "")
|
||
if findErr != nil {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询上级代理失败, %v", findErr)
|
||
}
|
||
if len(agents) == 0 {
|
||
return errors.Wrapf(xerr.NewErrMsg("邀请信息无效"), "")
|
||
}
|
||
parentAgentId = agents[0].Id
|
||
targetLevel = 1
|
||
}
|
||
}
|
||
|
||
// 5. 创建代理记录
|
||
|
||
newAgent := &model.Agent{Id: uuid.NewString(), UserId: userID, Level: targetLevel, Mobile: encryptedMobile}
|
||
if req.Region != "" {
|
||
newAgent.Region = sql.NullString{String: req.Region, Valid: true}
|
||
}
|
||
if req.WechatId != "" {
|
||
newAgent.WechatId = sql.NullString{String: req.WechatId, Valid: true}
|
||
}
|
||
|
||
// 6. 处理上级关系
|
||
if parentAgentId != "" {
|
||
// 查找上级代理
|
||
parentAgent, err := l.svcCtx.AgentModel.FindOne(transCtx, parentAgentId)
|
||
if err != nil {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询上级代理失败, %v", err)
|
||
}
|
||
|
||
// 验证关系是否允许（下级不能比上级等级高）
|
||
if newAgent.Level > parentAgent.Level {
|
||
return errors.Wrapf(xerr.NewErrMsg("代理等级不能高于上级代理"), "")
|
||
}
|
||
|
||
// 查找团队首领（钻石代理）
|
||
teamLeaderId, err := l.findTeamLeader(transCtx, parentAgent.Id)
|
||
if err != nil {
|
||
return errors.Wrapf(err, "查找团队首领失败")
|
||
}
|
||
if teamLeaderId != "" {
|
||
newAgent.TeamLeaderId = sql.NullString{String: teamLeaderId, Valid: true}
|
||
}
|
||
|
||
newAgent.AgentCode = 0
|
||
_, err = l.svcCtx.AgentModel.Insert(transCtx, session, newAgent)
|
||
if err != nil {
|
||
return errors.Wrapf(err, "创建代理记录失败")
|
||
}
|
||
agentID = newAgent.Id
|
||
|
||
// 建立关系
|
||
relation := &model.AgentRelation{Id: uuid.NewString(), ParentId: parentAgent.Id, ChildId: agentID, RelationType: 1}
|
||
if _, err := l.svcCtx.AgentRelationModel.Insert(transCtx, session, relation); err != nil {
|
||
return errors.Wrapf(err, "建立代理关系失败")
|
||
}
|
||
} else {
|
||
// 平台发放的钻石邀请码，独立成团队
|
||
if targetLevel == 3 {
|
||
newAgent.AgentCode = 0
|
||
_, err = l.svcCtx.AgentModel.Insert(transCtx, session, newAgent)
|
||
if err != nil {
|
||
return errors.Wrapf(err, "创建代理记录失败")
|
||
}
|
||
agentID = newAgent.Id
|
||
|
||
// 设置自己为团队首领
|
||
newAgent.TeamLeaderId = sql.NullString{String: agentID, Valid: true}
|
||
if err := l.svcCtx.AgentModel.UpdateInTransaction(transCtx, session, newAgent); err != nil {
|
||
return errors.Wrapf(err, "更新团队首领失败")
|
||
}
|
||
} else {
|
||
// 普通/黄金代理，但没有上级（异常情况）
|
||
newAgent.AgentCode = 0
|
||
_, err = l.svcCtx.AgentModel.Insert(transCtx, session, newAgent)
|
||
if err != nil {
|
||
return errors.Wrapf(err, "创建代理记录失败")
|
||
}
|
||
agentID = newAgent.Id
|
||
}
|
||
}
|
||
|
||
// 7. 初始化钱包
|
||
wallet := &model.AgentWallet{Id: uuid.NewString(), AgentId: agentID}
|
||
if _, err := l.svcCtx.AgentWalletModel.Insert(transCtx, session, wallet); err != nil {
|
||
return errors.Wrapf(err, "初始化钱包失败")
|
||
}
|
||
|
||
// 8. 更新邀请码状态
|
||
// 钻石级别的邀请码只能使用一次，使用后立即失效
|
||
// 普通级别的邀请码可以无限使用，不更新状态
|
||
if targetLevel == 3 {
|
||
// 钻石邀请码：使用后失效
|
||
inviteCodeModel.Status = 1 // 已使用（使用后立即失效）
|
||
}
|
||
if inviteCodeModel != nil {
|
||
inviteCodeModel.UsedUserId = sql.NullString{String: userID, Valid: true}
|
||
inviteCodeModel.UsedAgentId = sql.NullString{String: agentID, Valid: true}
|
||
inviteCodeModel.UsedTime = sql.NullTime{Time: time.Now(), Valid: true}
|
||
if err := l.svcCtx.AgentInviteCodeModel.UpdateWithVersion(transCtx, session, inviteCodeModel); err != nil {
|
||
return errors.Wrapf(err, "更新邀请码状态失败")
|
||
}
|
||
}
|
||
|
||
// 9. 记录邀请码使用历史（用于统计和查询）
|
||
if inviteCodeModel != nil {
|
||
usage := &model.AgentInviteCodeUsage{Id: uuid.NewString(), InviteCodeId: inviteCodeModel.Id, Code: inviteCodeModel.Code, UserId: userID, AgentId: agentID, AgentLevel: targetLevel, UsedTime: time.Now()}
|
||
if _, err := l.svcCtx.AgentInviteCodeUsageModel.Insert(transCtx, session, usage); err != nil {
|
||
return errors.Wrapf(err, "记录邀请码使用历史失败")
|
||
}
|
||
}
|
||
|
||
agentLevel = targetLevel
|
||
return nil
|
||
})
|
||
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 10. 生成并返回token
|
||
token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID)
|
||
if err != nil {
|
||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成token失败: %v", err)
|
||
}
|
||
|
||
now := time.Now().Unix()
|
||
// 获取等级名称
|
||
levelName := ""
|
||
switch agentLevel {
|
||
case 1:
|
||
levelName = "普通"
|
||
case 2:
|
||
levelName = "黄金"
|
||
case 3:
|
||
levelName = "钻石"
|
||
}
|
||
agent, _ := l.svcCtx.AgentModel.FindOne(l.ctx, agentID)
|
||
return &types.RegisterByInviteCodeResp{
|
||
AccessToken: token,
|
||
AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire,
|
||
RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter,
|
||
AgentId: agentID,
|
||
Level: agentLevel,
|
||
LevelName: levelName,
|
||
AgentCode: func() int64 {
|
||
if agent != nil {
|
||
return agent.AgentCode
|
||
}
|
||
return 0
|
||
}(),
|
||
}, nil
|
||
}
|
||
|
||
// findTeamLeader 查找团队首领（钻石代理）
|
||
func (l *RegisterByInviteCodeLogic) findTeamLeader(ctx context.Context, agentId string) (string, error) {
|
||
currentId := agentId
|
||
maxDepth := 100
|
||
depth := 0
|
||
|
||
for depth < maxDepth {
|
||
builder := l.svcCtx.AgentRelationModel.SelectBuilder().
|
||
Where("child_id = ? AND relation_type = ? AND del_state = ?", currentId, 1, 0)
|
||
relations, err := l.svcCtx.AgentRelationModel.FindAll(ctx, builder, "")
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
if len(relations) == 0 {
|
||
agent, err := l.svcCtx.AgentModel.FindOne(ctx, currentId)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
if agent.Level == 3 {
|
||
return agent.Id, nil
|
||
}
|
||
return "", nil
|
||
}
|
||
|
||
parentAgent, err := l.svcCtx.AgentModel.FindOne(ctx, relations[0].ParentId)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
|
||
if parentAgent.Level == 3 {
|
||
return parentAgent.Id, nil
|
||
}
|
||
|
||
currentId = parentAgent.Id
|
||
depth++
|
||
}
|
||
|
||
return "", nil
|
||
}
|
||
|
||
func (l *RegisterByInviteCodeLogic) findAgentByCode(ctx context.Context, code int64) (*model.Agent, error) {
|
||
builder := l.svcCtx.AgentModel.SelectBuilder().Where("agent_code = ? AND del_state = ?", code, globalkey.DelStateNo).Limit(1)
|
||
agents, err := l.svcCtx.AgentModel.FindAll(ctx, builder, "")
|
||
if err != nil {
|
||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询上级代理失败, %v", err)
|
||
}
|
||
if len(agents) == 0 {
|
||
return nil, errors.Wrapf(xerr.NewErrMsg("上级邀请码不存在"), "")
|
||
}
|
||
return agents[0], nil
|
||
}
|
||
|
||
func (l *RegisterByInviteCodeLogic) allocateAgentCode(ctx context.Context, session sqlx.Session) (int64, error) {
|
||
builder := l.svcCtx.AgentModel.SelectBuilder().OrderBy("agent_code DESC").Limit(1)
|
||
rows, err := l.svcCtx.AgentModel.FindAll(ctx, builder, "")
|
||
if err != nil {
|
||
return 0, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理编码失败, %v", err)
|
||
}
|
||
var next int64 = 16800
|
||
if len(rows) > 0 && rows[0].AgentCode > 0 {
|
||
next = rows[0].AgentCode + 1
|
||
}
|
||
return next, nil
|
||
}
|
||
|
||
// handleMobileNotExists 处理手机号不存在的情况
|
||
func (l *RegisterByInviteCodeLogic) handleMobileNotExists(ctx context.Context, session sqlx.Session, encryptedMobile string, currentUserID string) (string, error) {
|
||
if currentUserID == "" {
|
||
// 场景1.1: 未登录 + 手机号不存在 -> 创建新用户
|
||
newUser := &model.User{Id: uuid.NewString(), Mobile: sql.NullString{String: encryptedMobile, Valid: true}}
|
||
if _, err := l.svcCtx.UserModel.Insert(ctx, session, newUser); err != nil {
|
||
return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建用户失败: %v", err)
|
||
}
|
||
// 创建 mobile 认证
|
||
if _, err := l.svcCtx.UserAuthModel.Insert(ctx, session, &model.UserAuth{
|
||
Id: uuid.NewString(),
|
||
UserId: newUser.Id,
|
||
AuthType: model.UserAuthTypeMobile,
|
||
AuthKey: encryptedMobile,
|
||
}); err != nil {
|
||
return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建手机号认证失败: %v", err)
|
||
}
|
||
return newUser.Id, nil
|
||
} else {
|
||
// 场景1.2: 已登录临时用户 + 手机号不存在 -> 升级为正式用户
|
||
// 前置检查已保证不是正式用户，所以这里一定是临时用户
|
||
currentUser, err := l.svcCtx.UserModel.FindOne(ctx, currentUserID)
|
||
if err != nil {
|
||
return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询当前用户失败: %v", err)
|
||
}
|
||
// 升级为正式用户
|
||
currentUser.Mobile = sql.NullString{String: encryptedMobile, Valid: true}
|
||
if _, err := l.svcCtx.UserModel.Update(ctx, session, currentUser); err != nil {
|
||
return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新手机号失败: %v", err)
|
||
}
|
||
// 创建 mobile 认证
|
||
if _, err := l.svcCtx.UserAuthModel.Insert(ctx, session, &model.UserAuth{
|
||
Id: uuid.NewString(),
|
||
UserId: currentUserID,
|
||
AuthType: model.UserAuthTypeMobile,
|
||
AuthKey: encryptedMobile,
|
||
}); err != nil {
|
||
return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建手机号认证失败: %v", err)
|
||
}
|
||
return currentUserID, nil
|
||
}
|
||
}
|
||
|
||
// handleMobileExists 处理手机号已存在的情况
|
||
func (l *RegisterByInviteCodeLogic) handleMobileExists(ctx context.Context, session sqlx.Session, targetUser *model.User, currentUserID string, currentAuthType string, currentAuthKey string) (string, error) {
|
||
userID := targetUser.Id
|
||
|
||
// 检查目标用户是否已是代理
|
||
existingAgent, err := l.svcCtx.AgentModel.FindOneByUserId(ctx, userID)
|
||
if err != nil && !errors.Is(err, model.ErrNotFound) {
|
||
return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败: %v", err)
|
||
}
|
||
if existingAgent != nil {
|
||
return "", errors.Wrapf(xerr.NewErrMsg("该手机号已经是代理，不能重复注册"), "")
|
||
}
|
||
|
||
if currentUserID == "" {
|
||
// 场景2.1: 未登录 + 手机号存在 -> 直接使用目标用户（验证码已确认身份）
|
||
return userID, nil
|
||
} else if currentUserID == userID {
|
||
// 场景2.2: 已登录正式用户 + 手机号匹配 -> 直接使用
|
||
// 前置检查已保证手机号匹配且不是代理
|
||
return userID, nil
|
||
} else {
|
||
// 场景2.3: 已登录临时用户 + 手机号存在 -> 需要合并账号
|
||
// 前置检查已保证是临时用户（不是正式用户）
|
||
return l.mergeTempUserToTarget(ctx, session, currentUserID, userID, currentAuthType, currentAuthKey)
|
||
}
|
||
}
|
||
|
||
// mergeTempUserToTarget 合并临时用户到目标用户
|
||
func (l *RegisterByInviteCodeLogic) mergeTempUserToTarget(ctx context.Context, session sqlx.Session, sourceUserID string, targetUserID string, currentAuthType string, currentAuthKey string) (string, error) {
|
||
// 检查目标用户是否已有该认证（除了UUID）
|
||
if currentAuthType != model.UserAuthTypeUUID {
|
||
targetAuth, err := l.svcCtx.UserAuthModel.FindOneByUserIdAuthType(ctx, targetUserID, currentAuthType)
|
||
if err != nil && !errors.Is(err, model.ErrNotFound) {
|
||
return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找认证信息失败: %v", err)
|
||
}
|
||
if targetAuth != nil && targetAuth.AuthKey != currentAuthKey {
|
||
// 目标用户已有该类型的其他认证，证明手机号绑定过其他微信等
|
||
if currentAuthType == model.UserAuthTypeWxh5OpenID {
|
||
return "", errors.Wrapf(xerr.NewErrMsg("该手机号已绑定其他微信号"), "")
|
||
}
|
||
if currentAuthType == model.UserAuthTypeWxMiniOpenID {
|
||
return "", errors.Wrapf(xerr.NewErrMsg("该手机号已绑定其他微信号"), "")
|
||
}
|
||
return "", errors.Wrapf(xerr.NewErrMsg("该手机号已绑定其他终端"), "")
|
||
}
|
||
}
|
||
|
||
// 查找当前认证
|
||
existingAuth, err := l.svcCtx.UserAuthModel.FindOneByAuthTypeAuthKey(ctx, currentAuthType, currentAuthKey)
|
||
if err != nil && !errors.Is(err, model.ErrNotFound) {
|
||
return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找认证信息失败: %v", err)
|
||
}
|
||
|
||
// 执行账号合并
|
||
if err := l.mergeUserAccounts(ctx, session, sourceUserID, targetUserID, currentAuthType, currentAuthKey, existingAuth); err != nil {
|
||
return "", err
|
||
}
|
||
|
||
return targetUserID, nil
|
||
}
|
||
|
||
// mergeUserAccounts 合并账号：迁移认证、业务数据，删除临时用户
|
||
func (l *RegisterByInviteCodeLogic) mergeUserAccounts(ctx context.Context, session sqlx.Session, sourceUserID string, targetUserID string, currentAuthType string, currentAuthKey string, existingAuth *model.UserAuth) error {
|
||
// 1) 认证绑定处理
|
||
if existingAuth != nil && existingAuth.UserId != targetUserID {
|
||
// 认证存在但不属于目标用户，迁移到目标用户
|
||
if currentAuthType == model.UserAuthTypeUUID {
|
||
// UUID替换策略：如果目标用户已有UUID认证，替换UUID；否则迁移认证
|
||
targetUUIDAuth, _ := l.svcCtx.UserAuthModel.FindOneByUserIdAuthType(ctx, targetUserID, model.UserAuthTypeUUID)
|
||
if targetUUIDAuth != nil {
|
||
// 目标用户已有UUID认证，删除源认证并更新目标UUID
|
||
if err := l.svcCtx.UserAuthModel.Delete(ctx, session, existingAuth.Id); err != nil {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除旧UUID认证失败: %v", err)
|
||
}
|
||
if targetUUIDAuth.AuthKey != currentAuthKey {
|
||
targetUUIDAuth.AuthKey = currentAuthKey
|
||
if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, targetUUIDAuth); err != nil {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新目标UUID认证失败: %v", err)
|
||
}
|
||
}
|
||
} else {
|
||
// 目标用户没有UUID认证，迁移源认证
|
||
existingAuth.UserId = targetUserID
|
||
if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, existingAuth); err != nil {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "迁移UUID认证失败: %v", err)
|
||
}
|
||
}
|
||
} else {
|
||
// 其他认证类型，直接迁移
|
||
existingAuth.UserId = targetUserID
|
||
if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, existingAuth); err != nil {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "迁移认证失败: %v", err)
|
||
}
|
||
}
|
||
} else if existingAuth == nil {
|
||
// 认证不存在，创建新认证
|
||
if currentAuthType == model.UserAuthTypeUUID {
|
||
// UUID特殊处理：如果目标用户已有UUID认证，更新UUID；否则创建新认证
|
||
targetUUIDAuth, _ := l.svcCtx.UserAuthModel.FindOneByUserIdAuthType(ctx, targetUserID, model.UserAuthTypeUUID)
|
||
if targetUUIDAuth != nil {
|
||
if targetUUIDAuth.AuthKey != currentAuthKey {
|
||
targetUUIDAuth.AuthKey = currentAuthKey
|
||
if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, targetUUIDAuth); err != nil {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新目标UUID认证失败: %v", err)
|
||
}
|
||
}
|
||
} else {
|
||
if _, err := l.svcCtx.UserAuthModel.Insert(ctx, session, &model.UserAuth{
|
||
Id: uuid.NewString(),
|
||
UserId: targetUserID,
|
||
AuthType: currentAuthType,
|
||
AuthKey: currentAuthKey,
|
||
}); err != nil {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建UUID认证失败: %v", err)
|
||
}
|
||
}
|
||
} else {
|
||
// 其他认证类型，创建新认证
|
||
if _, err := l.svcCtx.UserAuthModel.Insert(ctx, session, &model.UserAuth{
|
||
Id: uuid.NewString(),
|
||
UserId: targetUserID,
|
||
AuthType: currentAuthType,
|
||
AuthKey: currentAuthKey,
|
||
}); err != nil {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建认证失败: %v", err)
|
||
}
|
||
}
|
||
}
|
||
|
||
// 2) 业务数据迁移：迁移订单和报告到目标用户
|
||
if err := l.svcCtx.OrderModel.UpdateUserIDWithSession(ctx, session, sourceUserID, targetUserID); err != nil {
|
||
return errors.Wrapf(err, "迁移订单失败")
|
||
}
|
||
if err := l.svcCtx.QueryModel.UpdateUserIDWithSession(ctx, session, sourceUserID, targetUserID); err != nil {
|
||
return errors.Wrapf(err, "迁移报告失败")
|
||
}
|
||
|
||
// 3) 删除临时用户
|
||
sourceUser, err := l.svcCtx.UserModel.FindOne(ctx, sourceUserID)
|
||
if err != nil {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找源用户失败: %v", err)
|
||
}
|
||
if err := l.svcCtx.UserModel.Delete(ctx, session, sourceUser.Id); err != nil {
|
||
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除源用户失败: %v", err)
|
||
}
|
||
|
||
return nil
|
||
}
|