feat: 新增 snowid(雪花ID+GORM钩子模板,吸收go-hua); wsc新增SessionClient会话层(吸收两端internal/cloud); jwtx重写为标准三段式HS256+密钥显式传参

This commit is contained in:
w11
2026-09-19 21:12:44 +08:00
parent 9ef71f51d4
commit 83679e2080
8 changed files with 452 additions and 42 deletions
+69
View File
@@ -0,0 +1,69 @@
// Package snowid 基于雪花算法的全局唯一 ID 生成器(进程内单例,需先 Init)。
//
// 运行时:main 里 Init 后调用 Generate/GenerateString。
// GORM 模型:IDMethod 交给 gorm.io/gen 的 WithMethod 注入,
// 生成的模型自带 BeforeCreate 钩子(Id 为 0 时自动填充 Generate())。
package snowid
import (
"sync"
"github.com/bwmarrin/snowflake"
"gorm.io/gorm"
)
var (
node *snowflake.Node
once sync.Once
initErr error
)
// Init 初始化雪花ID生成器
// nodeID: 节点ID,范围 0-1023
func Init(nodeID int64) error {
once.Do(func() {
node, initErr = snowflake.NewNode(nodeID)
})
return initErr
}
// InitWithStartTime 使用自定义起始时间初始化雪花ID生成器
// nodeID: 节点ID,范围 0-1023
// startTime: 自定义起始时间(Unix毫秒时间戳)
func InitWithStartTime(nodeID int64, startTime int64) error {
once.Do(func() {
snowflake.Epoch = startTime
node, initErr = snowflake.NewNode(nodeID)
})
return initErr
}
// Generate 生成一个新的雪花ID
func Generate() int64 {
if node == nil {
panic("snowid: not initialized, please call Init() first")
}
return node.Generate().Int64()
}
// GenerateString 生成字符串形式的雪花ID
func GenerateString() string {
if node == nil {
panic("snowid: not initialized, please call Init() first")
}
return node.Generate().String()
}
// IDMethod GORM 模型方法模板:交给 gen.WithMethod 注入后,
// 生成的模型在 Create 前自动生成雪花ID主键(字段名须为 Id)。
type IDMethod struct {
Id int64
}
// BeforeCreate 创建前钩子,自动生成雪花ID
func (m *IDMethod) BeforeCreate(tx *gorm.DB) error {
if m.Id == 0 {
m.Id = Generate()
}
return nil
}