debug

zhangqian

2023-12-08 84fb8e390b83dc9482524c12d7af6c93405c3fc1

 pkg/snowflake/snowflake.go

@@ -1,10 +1,85 @@
package snowflake

import (
   "apsClient/pkg/logx"
   "errors"
   "fmt"
   "hash/fnv"
   "net"
   "strconv"
   "sync"
   "time"
)

var idGenerater *IdWorker

func init() {
   // 使用 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))
   }
   idGenerater, err = NewIdWorker(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 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 之间
}

const (
   CEpoch         = 1474802888000
@@ -96,9 +171,7 @@
   return
}

var idGenerater, _ = NewIdWorker(0)

func GenerateId() int64 {
func GenerateID() int64 {
start:
   id, err := idGenerater.NextId()
   if err != nil {
@@ -106,3 +179,12 @@
   }
   return id
}

func GenerateIdStr() string {
start:
   id, err := idGenerater.NextId()
   if err != nil {
      goto start
   }
   return strconv.FormatInt(id, 10)
}