You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
31 lines
494 B
31 lines
494 B
package snowflake
|
|
|
|
import (
|
|
"github.com/sony/sonyflake"
|
|
|
|
"go-mysql-transfer/util/logs"
|
|
)
|
|
|
|
// 雪花ID工具
|
|
|
|
var _sf *sonyflake.Sonyflake
|
|
|
|
func InitSnowflake(machineId uint16) {
|
|
if _sf != nil {
|
|
return
|
|
}
|
|
|
|
var st sonyflake.Settings
|
|
st.MachineID = func() (u uint16, e error) {
|
|
return machineId, nil
|
|
}
|
|
_sf = sonyflake.NewSonyflake(st)
|
|
}
|
|
|
|
func NextId() (uint64, error) {
|
|
id, err := _sf.NextID()
|
|
if err != nil {
|
|
logs.Errorf("snowflake NextId ：%s", err.Error())
|
|
}
|
|
return id, nil
|
|
}
|
|
|