Go语言实现雪花算法生成分布式唯一ID

雪花算法原理：生成全局唯一ID的分布式解决方案

在分布式系统中，如何生成全局唯一ID一直是一个经典技术难题。雪花算法（Snowflake Algorithm）正是为解决这一需求而诞生——它由Twitter团队提出，巧妙地将一个64位的长整型ID拆分成多个独立部分，每个部分携带特定信息，从而在分布式环境下保证ID的唯一性。我们先来看看它的结构组成。

ID结构详解

1. 符号位（1位）：固定为0，确保生成的ID始终为正数。
2. 时间戳（41位）：记录生成ID时的毫秒级时间戳，支持约69年的跨度。
3. 工作节点ID（10位）：标识生成ID的机器节点，避免不同节点间的ID冲突。
4. 序列号（12位）：同一毫秒内生成的多个ID按此顺序递增，最多支持4096个ID。

生成步骤流程

1. 获取当前毫秒级时间戳。
2. 如果当前时间小于上次生成ID的时间，或者同一毫秒内生成的ID数量已达上限（4096个），则等待下一毫秒再继续生成。
3. 如果当前时间等于上次生成ID的时间，序列号自增1。
4. 如果当前时间大于上次生成ID的时间，序列号重置为0。
5. 将各部分按位组合，最终得到64位唯一ID。

Go实现雪花算法：构建高并发分布式ID生成器

package main

import (
	"fmt"
	"sync"
	"time"
)

const (
	workerBits     = 10
	sequenceBits   = 12
	workerMax      = -1 ^ (-1 << workerBits)
	sequenceMask   = -1 ^ (-1 << sequenceBits)
	timeShift      = workerBits + sequenceBits
	workerShift    = sequenceBits
	epoch          = 1609459200000
)

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

func NewSnowflake(workerID int64) *Snowflake {
	if workerID < 0 || workerID > workerMax {
		panic(fmt.Sprintf("worker ID must be between 0 and %d", workerMax))
	}
	return &Snowflake{
		lastTime: time.Now().UnixNano() / 1e6,
		workerID: workerID,
		sequence: 0,
	}
}

func (sf *Snowflake) NextID() int64 {
	sf.mu.Lock()
	defer sf.mu.Unlock()

	currentTime := time.Now().UnixNano() / 1e6

	if currentTime < sf.lastTime {
		panic(fmt.Sprintf("clock moved backwards, refusing to generate ID for %d milliseconds", sf.lastTime-currentTime))
	}

	if currentTime == sf.lastTime {
		sf.sequence = (sf.sequence + 1) & sequenceMask
		if sf.sequence == 0 {
			for currentTime <= sf.lastTime {
				currentTime = time.Now().UnixNano() / 1e6
			}
		}
	} else {
		sf.sequence = 0
	}

	sf.lastTime = currentTime

	id := (currentTime-epoch)<
高并发场景下的唯一性与递增性保障机制
讨论到这里，大家最关心的通常是：在高并发环境下，雪花算法究竟如何保证ID唯一且有序递增？核心原理其实可以归结为以下两点。

唯一性保障：每个工作节点ID都是全局唯一的，不同节点生成的ID天然不会发生冲突。在同一毫秒内，序列号通过自增操作和位掩码运算来确保不重复。
递增性保障：同一毫秒内生成的多个ID按照序列号依次递增。即使极端情况下一毫秒内生成的ID数量达到最大值（4096个），算法会主动等待下一毫秒再重新开始计数——这个等待机制确保了ID在时间维度上仍然保持严格的递增趋势。

总的来说，雪花算法在分布式高并发环境下是一个非常可靠的唯一ID生成方案。它的性能表现优异，碰撞概率极低，这也是为什么它被广泛应用于各类微服务、消息队列、数据库分片等系统中。