package snowflake
|
|
import (
|
"apsClient/pkg/logx"
|
"fmt"
|
"github.com/bwmarrin/snowflake"
|
"hash/fnv"
|
"net"
|
)
|
|
var Node *snowflake.Node
|
|
func Init(nodeId string) error {
|
// Create a new Node with a Node number of 1
|
var err error
|
nodeIdInt, err := StringToNumber(nodeId)
|
if err != nil {
|
logx.Errorf("snowflake Init error:%v", err)
|
return err
|
}
|
Node, err = snowflake.NewNode(nodeIdInt)
|
if err != nil {
|
logx.Errorf("snowflake NewNode error:%v", err)
|
return err
|
}
|
return nil
|
}
|
|
func StringToNumber(input string) (int64, error) {
|
hash := fnv.New32a()
|
_, err := hash.Write([]byte(input))
|
if err != nil {
|
return 0, err
|
}
|
hashValue := int64(hash.Sum32())
|
return hashValue, nil
|
}
|
|
func GenerateID() int64 {
|
if Node == nil {
|
// 使用 LookupIP 获取主机的 IP 地址列表
|
addrs, err := net.LookupIP("localhost")
|
if err != nil {
|
logx.Infof("snowflake can not generate, init error, get ip error:%v", err)
|
panic(fmt.Sprintf("snowflake can not generate, init error, get ip error:%v", err))
|
}
|
var ip string
|
// 遍历 IP 地址列表
|
for _, addr := range addrs {
|
// 判断 IP 地址的版本是 IPv4 还是 IPv6
|
if ipNet := addr.To4(); ipNet != nil {
|
ip = ipNet.String()
|
fmt.Printf("IPv4 Address: %s\n", ipNet.String())
|
}
|
}
|
err = Init(ip)
|
if err != nil {
|
panic(fmt.Sprintf("snowflake can not generate, init error,:%v", err))
|
}
|
}
|
return int64(Node.Generate())
|
}
|
