feat: 新增 snowid(雪花ID+GORM钩子模板,吸收go-hua); wsc新增SessionClient会话层(吸收两端internal/cloud); jwtx重写为标准三段式HS256+密钥显式传参
This commit is contained in:
@@ -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
|
||||
}
|
||||
Reference in New Issue
Block a user