什么时候出个go语言版本的

[daydaystudylab]

No description provided.

[xmapst]

package snowflake

import (
	"encoding/base64"
	"fmt"
	"strconv"
	"sync"
	"time"
)

type ID uint64

const (
	Epoch           = 1730390400 // 元时间 2024-11-01 00:00:00
	timestampBits   = 28         // 时间戳占用位数, 秒级
	nodeBits        = 22         // 节点 ID 占用位数
	sequenceBits    = 13         // 递增序列占用位数
	maxNodeID       = (1 << nodeBits) - 1
	maxSequence     = (1 << sequenceBits) - 1
	timestampShift  = sequenceBits + nodeBits
	nodeShift       = sequenceBits
	maxID           = (1 << (timestampBits + nodeBits + sequenceBits)) - 1
)

// IDGenerator 共享内存和锁的结构
type IDGenerator struct {
	lock      sync.Mutex
	node      uint64
	sequence  uint64
	timestamp uint64
}

// New 初始化雪花 ID 生成器
func New(nodeID int64) (*IDGenerator, error) {

	if nodeID > maxNodeID || nodeID < 0 {
		return nil, fmt.Errorf("%d NodeID out of range", nodeID)
	}
	// 初始化节点 ID（包括数据中心 ID 和节点 ID 的位移拼接）
	g := &IDGenerator{
		node:     uint64(nodeID << nodeShift),
		sequence: 1,
	}
	g.timestamp = g.currentEpoch()
	return g, nil
}

// NextID 获取新的 ID
func (g *IDGenerator) NextID() ID {
	g.lock.Lock()
	defer g.lock.Unlock()

	// 检查序列是否溢出
	if g.sequence > maxSequence {
		// 等待时间前进
		for g.timestamp > g.currentEpoch() {
			time.Sleep(100 * time.Millisecond)
		}
		g.timestamp++
		g.sequence = 1
	} else {
		g.sequence++
	}

	// 通过移位运算拼接生成最终的 ID
	id := (g.timestamp << timestampShift) | g.node | g.sequence

	return ID(id & maxID)
}

func (g *IDGenerator) currentEpoch() uint64 {
	return uint64(time.Now().Unix()) - Epoch
}

func (f ID) String() string {
	return strconv.FormatInt(int64(f), 10)
}

func (f ID) Uint64() uint64 {
	return uint64(f)
}

func (f ID) Int64() int64 {
	return int64(f)
}

func (f ID) Bytes() []byte {
	return []byte(f.String())
}

func (f ID) Base64() string {
	return base64.StdEncoding.EncodeToString(f.Bytes())
}

func (f ID) Time() int64 {
	return (int64(f) >> timestampShift) + Epoch
}

func (f ID) Node() int64 {
	shifted := int64(f) >> nodeShift
	return shifted & maxNodeID
}

func (f ID) Sequence() int64 {
	return int64(f) & maxSequence
}

func ParseInt64(id int64) ID {
	return ID(id)
}

func ParseString(id string) (ID, error) {
	i, err := strconv.ParseInt(id, 10, 64)
	return ID(i), err
}

func ParseBytes(id []byte) (ID, error) {
	i, err := strconv.ParseInt(string(id), 10, 64)
	return ID(i), err
}