邹方成
1ab39d2f5a
Some checks failed
Build docker and publish / linux (1.24.5) (push) Failing after 25s
feat(admin): 新增工会管理功能 feat(activity): 添加活动管理相关服务 feat(user): 实现用户道具卡和积分管理 feat(guild): 新增工会成员管理功能 fix: 修复数据库连接配置 fix: 修正jwtoken导入路径 fix: 解决端口冲突问题 style: 统一代码格式和注释风格 style: 更新项目常量命名 docs: 添加项目框架和开发规范文档 docs: 更新接口文档注释 chore: 移除无用代码和文件 chore: 更新Makefile和配置文件 chore: 清理日志文件 test: 添加道具卡测试脚本
23 lines
515 B
Go
23 lines
515 B
Go
package user
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"math/big"
|
|
)
|
|
|
|
func (s *service) generateInviteCode(ctx context.Context) string {
|
|
const letters = "ABCDEFGHJKLMNPQRSTUVWXYZ23456789"
|
|
b := make([]byte, 8)
|
|
for i := 0; i < 8; i++ {
|
|
n, _ := rand.Int(rand.Reader, big.NewInt(int64(len(letters))))
|
|
b[i] = letters[n.Int64()]
|
|
}
|
|
code := string(b)
|
|
exists, _ := s.readDB.Users.WithContext(ctx).Where(s.readDB.Users.InviteCode.Eq(code)).First()
|
|
if exists != nil {
|
|
return s.generateInviteCode(ctx)
|
|
}
|
|
return code
|
|
}
|