lan
29f0bad2bc
- Add YggdrasilErrorResponse struct and standard error codes for protocol compliance - Change UUID storage from varchar(36) to varchar(32) for unsigned format - Add utility functions: GenerateUUID, FormatUUIDToNoDash, RandomHex - Support unsigned query parameter in GetProfileByUUID endpoint - Improve refresh token response with available profiles list - Fix key pair retrieval to use correct database column (rsa_private_key) - Update UUID validator to accept both 32-char and 36-char formats - Add SignStringWithProfileRSA method for profile-specific signing - Fix profile assignment validation in refresh token flow
98 lines
2.9 KiB
Go
98 lines
2.9 KiB
Go
package utils
|
||
|
||
import (
|
||
"crypto/rand"
|
||
"encoding/hex"
|
||
"strings"
|
||
|
||
"github.com/google/uuid"
|
||
"go.uber.org/zap"
|
||
)
|
||
|
||
// FormatUUID 将UUID格式化为带连字符的标准格式
|
||
// 如果输入已经是标准格式，直接返回
|
||
// 如果输入是32位十六进制字符串，添加连字符
|
||
// 如果输入格式无效，返回错误
|
||
func FormatUUID(uuid string) string {
|
||
// 如果为空，直接返回
|
||
if uuid == "" {
|
||
return uuid
|
||
}
|
||
|
||
// 如果已经是标准格式（8-4-4-4-12），直接返回
|
||
if len(uuid) == 36 && uuid[8] == '-' && uuid[13] == '-' && uuid[18] == '-' && uuid[23] == '-' {
|
||
return uuid
|
||
}
|
||
|
||
// 如果是32位十六进制字符串，添加连字符
|
||
if len(uuid) == 32 {
|
||
// 预分配容量以提高性能
|
||
var b strings.Builder
|
||
b.Grow(36) // 最终长度为36(32个字符 + 4个连字符)
|
||
|
||
// 使用WriteString和WriteByte优化性能
|
||
b.WriteString(uuid[0:8])
|
||
b.WriteByte('-')
|
||
b.WriteString(uuid[8:12])
|
||
b.WriteByte('-')
|
||
b.WriteString(uuid[12:16])
|
||
b.WriteByte('-')
|
||
b.WriteString(uuid[16:20])
|
||
b.WriteByte('-')
|
||
b.WriteString(uuid[20:32])
|
||
|
||
return b.String()
|
||
}
|
||
|
||
// 如果长度不是32或36，说明格式无效，直接返回原值
|
||
var logger *zap.Logger
|
||
logger.Warn("[WARN] UUID格式无效: ", zap.String("uuid:", uuid))
|
||
return uuid
|
||
}
|
||
|
||
// GenerateUUID 生成无符号UUID（32位十六进制字符串，不带连字符）
|
||
// 使用github.com/google/uuid库生成标准UUID，然后移除连字符
|
||
// 返回格式示例: "123e4567e89b12d3a456426614174000"
|
||
func GenerateUUID() string {
|
||
return strings.ReplaceAll(uuid.New().String(), "-", "")
|
||
}
|
||
|
||
// FormatUUIDToNoDash 将带连字符的UUID转换为无符号格式（移除连字符）
|
||
// 输入: "123e4567-e89b-12d3-a456-426614174000"
|
||
// 输出: "123e4567e89b12d3a456426614174000"
|
||
// 如果输入已经是32位格式，直接返回
|
||
// 如果输入格式无效，返回原值并记录警告
|
||
func FormatUUIDToNoDash(uuid string) string {
|
||
// 如果为空，直接返回
|
||
if uuid == "" {
|
||
return uuid
|
||
}
|
||
|
||
// 如果已经是32位格式（无连字符），直接返回
|
||
if len(uuid) == 32 {
|
||
return uuid
|
||
}
|
||
|
||
// 如果是36位标准格式，移除连字符
|
||
if len(uuid) == 36 && uuid[8] == '-' && uuid[13] == '-' && uuid[18] == '-' && uuid[23] == '-' {
|
||
return strings.ReplaceAll(uuid, "-", "")
|
||
}
|
||
|
||
// 如果格式无效，记录警告并返回原值
|
||
var logger *zap.Logger
|
||
logger.Warn("[WARN] UUID格式无效，无法转换为无符号格式: ", zap.String("uuid:", uuid))
|
||
return uuid
|
||
}
|
||
|
||
// RandomHex 生成指定长度的随机十六进制字符串
|
||
// 参数 n: 需要生成的十六进制字符数量（每个字节生成2个十六进制字符）
|
||
// 返回: 长度为 2*n 的十六进制字符串
|
||
// 示例: RandomHex(16) 返回 "a1b2c3d4e5f67890abcdef1234567890" (32字符)
|
||
func RandomHex(n uint) (string, error) {
|
||
bytes := make([]byte, n)
|
||
if _, err := rand.Read(bytes); err != nil {
|
||
return "", err
|
||
}
|
||
return hex.EncodeToString(bytes), nil
|
||
}
|