zhangqian
2023-12-08 288d78f66ecd1b628e8d1df7a3da99fc5e6880ec
pkg/snowflake/snowflake.go
@@ -10,20 +10,70 @@
var Node *snowflake.Node
func Init(nodeId string) error {
   // Create a new Node with a Node number of 1
func Init(nodeId int64) error {
   // Create a new Node with a Node number of 0~1023
   var err error
   nodeIdInt, err := StringToNumber(nodeId)
   if err != nil {
      logx.Errorf("snowflake Init error:%v", err)
      return err
   }
   Node, err = snowflake.NewNode(nodeIdInt)
   Node, err = snowflake.NewNode(nodeId)
   if err != nil {
      logx.Errorf("snowflake NewNode error:%v", err)
      return err
   }
   return nil
}
func InitWithIP() {
   // 使用 LookupIP 获取主机的 IP 地址列表
   // 获取本机所有网络接口
   interfaces, err := net.Interfaces()
   if err != nil {
      logx.Errorf("snowflake InitWithIP error:%v", err)
      return
   }
   var ip string
   // 遍历所有网络接口
   for _, iface := range interfaces {
      // 排除一些特殊的接口,例如 loopback 接口
      if iface.Flags&net.FlagUp != 0 && iface.Flags&net.FlagLoopback == 0 {
         // 获取接口的所有地址
         addrs, err := iface.Addrs()
         if err != nil {
            logx.Errorf("snowflake InitWithIP error:%v", err)
            continue
         }
         // 遍历接口的所有地址
         for _, addr := range addrs {
            // 检查地址类型是否是 IP 地址
            if ipNet, ok := addr.(*net.IPNet); ok && !ipNet.IP.IsLoopback() {
               // 判断 IP 地址的版本是 IPv4 还是 IPv6
               if ipNet.IP.To4() != nil {
                  fmt.Printf("IPv4 Address: %s\n", ipNet.IP.String())
                  if ipNet.IP.String() != "127.0.0.1" {
                     ip = ipNet.IP.String()
                     goto getIpOK
                  }
               } else {
                  fmt.Printf("IPv6 Address: %s\n", ipNet.IP.String())
               }
            }
         }
      }
   }
getIpOK:
   if ip == "" {
      logx.Errorf("snowflake InitWithIP can not find Ip")
      return
   }
   ipNumber, err := ipToNumber(ip)
   if err != nil {
      logx.Errorf("snowflake can not generate, init error, ip to number error :%v", err)
      panic(fmt.Sprintf("snowflake can not generate, init error, ip to number error :%v", err))
   }
   err = Init(ipNumber)
   if err != nil {
      logx.Errorf("snowflake can not generate, init error :%v", err)
      panic(fmt.Sprintf("snowflake can not generate, init error :%v", err))
   }
}
func StringToNumber(input string) (int64, error) {
@@ -32,31 +82,25 @@
   if err != nil {
      return 0, err
   }
   hashValue := int64(hash.Sum32())
   return hashValue, nil
   return int64(hash.Sum32() % 1024), nil // 取余数确保结果在 1 到 1023 之间
}
func ipToNumber(ip string) (int64, error) {
   // 将 IP 地址字符串解析为 net.IP 类型
   parsedIP := net.ParseIP(ip)
   if parsedIP == nil {
      return 0, fmt.Errorf("invalid IP address:%v", ip)
   }
   // 使用 FNV-1a 散列算法计算哈希值
   hash := fnv.New32a()
   hash.Write(parsedIP)
   return int64(hash.Sum32() % 1024), nil // 取余数确保结果在 1 到 1023 之间
}
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))
      }
      InitWithIP()
   }
   return int64(Node.Generate())
}