雪花算法golang实现

雪花算法是一种分布式唯一ID生成算法，其在分布式系统中广泛应用。它以时间戳、机器ID和序列号的方式，生成全局唯一且有序的ID。雪花算法的Golang实现，可以帮助我们实现高效、可靠的ID生成。

雪花算法原理

雪花算法的核心思想是根据当前时间、机器ID和序列号生成ID。

具体来说，雪花算法的ID由以下几个部分组成：

• 时间戳：占用41位，精确到毫秒级别，可以使用当前时间戳减去一个固定的起始时间戳，得到相对时间戳。
• 机器ID：占用10位，用来标识不同的机器，保证每台机器生成的ID是唯一的。
• 序列号：占用12位，表示在同一毫秒内生成的序列号，从0开始递增，最多可以生成4096个ID。

实现雪花算法

下面是一个基于Golang的雪花算法实现：

package main

import (
	"errors"
	"sync"
	"time"
)

const (
	twepoch            = int64(1609430400000) // 起始时间戳
	workerIDBits       = uint(10)             // 机器ID所占位数
	sequenceBits       = uint(12)             // 序列号所占位数
	workerIDShift      = sequenceBits         // 机器ID左移位数
	timestampShift     = sequenceBits + workerIDBits
	sequenceMask       = int64(-1 ^ (-1 << sequenceBits)) // 最大序列号
	maxWorkerID   uint = 1023                                // 最大机器ID
)

type Snowflake struct {
	mu         sync.Mutex
	workerID   uint
	sequence   int64
	lastMillis int64
}

func NewSnowflake(workerID uint) (*Snowflake, error) {
	if workerID > maxWorkerID {
		return nil, errors.New("Invalid worker ID")
	}
	return &Snowflake{
		workerID: workerID,
	}, nil
}

func (s *Snowflake) NextID() (int64, error) {
	s.mu.Lock()
	defer s.mu.Unlock()

	curMillis := currentMillis()
	if curMillis < s.lastMillis {
		return 0, errors.New("Clock moved backwards")
	}

	if curMillis == s.lastMillis {
		s.sequence = (s.sequence + 1) & sequenceMask
		if s.sequence == 0 {
			for curMillis <= s.lastMillis {
				curMillis = currentMillis()
			}
		}
	} else {
		s.sequence = 0
	}

	s.lastMillis = curMillis

	id := (curMillis-twepoch)<

sf, err := NewSnowflake(1)
if err != nil {
    panic(err)
}

id, err := sf.NextID()
if err != nil {
    panic(err)
}

println(id)