24 lines
485 B
Go
24 lines
485 B
Go
package utils
|
||
|
||
import (
|
||
"crypto/md5"
|
||
"fmt"
|
||
"math/rand"
|
||
"time"
|
||
)
|
||
|
||
const charset = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
|
||
|
||
func GenerateInviteCode(userID uint) string {
|
||
rand.Seed(time.Now().UnixNano())
|
||
// 拼接用户ID和随机数
|
||
data := fmt.Sprintf("%d%d", userID, rand.Intn(1000000))
|
||
hash := md5.Sum([]byte(data))
|
||
code := ""
|
||
for i := 0; i < 6; i++ {
|
||
// 取哈希的前6位，每位映射到字符集
|
||
code += string(charset[int(hash[i])%len(charset)])
|
||
}
|
||
return code
|
||
}
|