saprkteamDash/pkg/utils/snowflake.go

package utils

import (
	"errors"
	"log"
	"net"

	"sparkteam-dash/pkg/snowflake"
	//"github.com/bwmarrin/snowflake"
)

// 使用推特算法生成的唯一ID
var twitSnowNode *snowflake.Node

func InitTwitEquipmentGuid() {
	var machineID uint16 = 1
	if id, err := lower16BitPrivateIP(); err == nil {
		machineID = id
	}
	machineID &= 0x3ff
	var err error
	twitSnowNode, err = snowflake.NewNode(int64(machineID))
	if err != nil {
		log.Fatalln(err)
	}
}

func GetTwitEquipmentGuid() string {
	// Generate a snowflake ID.
	id := twitSnowNode.Generate()
	return id.String()
}

func privateIPv4() (net.IP, error) {
	as, err := net.InterfaceAddrs()
	if err != nil {
		return nil, err
	}

	for _, a := range as {
		ipnet, ok := a.(*net.IPNet)
		if !ok || ipnet.IP.IsLoopback() {
			continue
		}

		ip := ipnet.IP.To4()
		if isPrivateIPv4(ip) {
			return ip, nil
		}
	}
	return nil, errors.New("no private ip address")
}

func isPrivateIPv4(ip net.IP) bool {
	return ip != nil &&
		(ip[0] == 10 || ip[0] == 172 && (ip[1] >= 16 && ip[1] < 32) || ip[0] == 192 && ip[1] == 168)
}

func lower16BitPrivateIP() (uint16, error) {
	ip, err := privateIPv4()
	if err != nil {
		return 0, err
	}

	return uint16(ip[2])<<8 + uint16(ip[3]), nil
}