43 lines
885 B
Go
43 lines
885 B
Go
package utils
|
||
|
||
import (
|
||
"errors"
|
||
"time"
|
||
|
||
sf "github.com/bwmarrin/snowflake"
|
||
)
|
||
|
||
const (
|
||
_dafaultStartTime = "2023-12-31" // 默认开始时间
|
||
)
|
||
|
||
var node *sf.Node
|
||
|
||
// Init 雪花算法组件初始化,正常应该把雪花算法当成一个独立的服务部署
|
||
// startTime 开始时间
|
||
// machineID 机器id
|
||
func Init(startTime string, machineID int64) (err error) {
|
||
if machineID < 0 {
|
||
return errors.New("snowflake need machineID")
|
||
}
|
||
if len(startTime) == 0 {
|
||
startTime = _dafaultStartTime
|
||
}
|
||
var st time.Time
|
||
st, err = time.Parse("2006-01-02", startTime)
|
||
if err != nil {
|
||
return
|
||
}
|
||
sf.Epoch = st.UnixNano() / 1000000 // 时间戳的开始时间，默认从1970年开始计算
|
||
node, err = sf.NewNode(machineID) // 机器编号，最多1024
|
||
return
|
||
}
|
||
|
||
func GenID() int64 {
|
||
return node.Generate().Int64()
|
||
}
|
||
|
||
func GenIDStr() string {
|
||
return node.Generate().String()
|
||
}
|