116 lines
3.2 KiB
Go
116 lines
3.2 KiB
Go
|
package agent
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"database/sql"
|
|||
|
|
"time"
|
|||
|
|
"ycc-server/app/main/model"
|
|||
|
|
"ycc-server/common/ctxdata"
|
|||
|
|
"ycc-server/common/tool"
|
|||
|
|
"ycc-server/common/xerr"
|
|||
|
|
|
|||
|
|
"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 GenerateInviteCodeLogic struct {
|
|||
|
|
logx.Logger
|
|||
|
|
ctx context.Context
|
|||
|
|
svcCtx *svc.ServiceContext
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewGenerateInviteCodeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GenerateInviteCodeLogic {
|
|||
|
|
return &GenerateInviteCodeLogic{
|
|||
|
|
Logger: logx.WithContext(ctx),
|
|||
|
|
ctx: ctx,
|
|||
|
|
svcCtx: svcCtx,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (l *GenerateInviteCodeLogic) GenerateInviteCode(req *types.GenerateInviteCodeReq) (resp *types.GenerateInviteCodeResp, err error) {
|
|||
|
|
// 1. 获取当前代理信息
|
|||
|
|
userID, err := ctxdata.GetUidFromCtx(l.ctx)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
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)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 2. 验证生成数量
|
|||
|
|
if req.Count <= 0 || req.Count > 100 {
|
|||
|
|
return nil, errors.Wrapf(xerr.NewErrMsg("生成数量必须在1-100之间"), "")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 3. 生成邀请码
|
|||
|
|
codes := make([]string, 0, req.Count)
|
|||
|
|
var expireTime sql.NullTime
|
|||
|
|
if req.ExpireDays > 0 {
|
|||
|
|
expireTime = sql.NullTime{
|
|||
|
|
Time: time.Now().AddDate(0, 0, int(req.ExpireDays)),
|
|||
|
|
Valid: true,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
err = l.svcCtx.AgentInviteCodeModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error {
|
|||
|
|
for i := int64(0); i < req.Count; i++ {
|
|||
|
|
// 生成8位随机邀请码（大小写字母+数字）
|
|||
|
|
var code string
|
|||
|
|
maxRetries := 10 // 最大重试次数
|
|||
|
|
for retry := 0; retry < maxRetries; retry++ {
|
|||
|
|
code = tool.Krand(8, tool.KC_RAND_KIND_ALL)
|
|||
|
|
// 检查邀请码是否已存在
|
|||
|
|
_, err := l.svcCtx.AgentInviteCodeModel.FindOneByCode(transCtx, code)
|
|||
|
|
if err != nil {
|
|||
|
|
if errors.Is(err, model.ErrNotFound) {
|
|||
|
|
// 邀请码不存在，可以使用
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "检查邀请码失败, %v", err)
|
|||
|
|
}
|
|||
|
|
// 邀请码已存在，继续生成
|
|||
|
|
if retry == maxRetries-1 {
|
|||
|
|
return errors.Wrapf(xerr.NewErrMsg("生成邀请码失败，请重试"), "")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建邀请码记录
|
|||
|
|
inviteCode := &model.AgentInviteCode{
|
|||
|
|
Code: code,
|
|||
|
|
AgentId: sql.NullInt64{Int64: agent.Id, Valid: true},
|
|||
|
|
TargetLevel: 1, // 代理发放的邀请码，目标等级为普通代理
|
|||
|
|
Status: 0, // 未使用
|
|||
|
|
ExpireTime: expireTime,
|
|||
|
|
Remark: sql.NullString{String: req.Remark, Valid: req.Remark != ""},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
_, err := l.svcCtx.AgentInviteCodeModel.Insert(transCtx, session, inviteCode)
|
|||
|
|
if err != nil {
|
|||
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建邀请码失败, %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
codes = append(codes, code)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return nil
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &types.GenerateInviteCodeResp{
|
|||
|
|
Codes: codes,
|
|||
|
|
}, nil
|
|||
|
|
}
|