每日一Go-55、分布式 ID 生成(雪花算法 / Segment / Redis / DB)

一、为什么分布式系统一定要"自己造ID"?

javascript 复制代码
单机时代,利用数据库的自增`ID`
AUTO_INCREMENT

但是在微服务/多实例/分库分表的情况下,会出现:

  • ID冲突

  • 数据迁移困难

  • 顺序失控

  • 跨库无法唯一定位

二、分布式ID的核心指标

一个靠谱的ID方案,至少要满足:

|------|---------|
| 指标 | 说明 |
| 全局唯一 | 不能重复 |
| 高性能 | AQS≥10w |
| 有序性 | 越新越大 |
| 高可用 | 不能成为单点 |
| 易扩展 | 节点随时加 |

三、主流4种方案总览

|------------|----|----|-------|-----|---------|
| 方案 | 性能 | 有序 | 依赖 | 复杂度 | 典型 |
| 雪花算法 | 5星 | 4星 | 本地 | 两星 | Twitter |
| Segment号段 | 4星 | 5星 | DB | 3星 | 美团/京东 |
| Redis INCR | 3星 | 5星 | Redis | 1星 | 小系统 |
| DB自增 | 1星 | 5星 | DB | 1星 | 单体 |

四、方案1:DB自增

cs 复制代码
INSERT INTO t VALUES ();
SELECT LAST_INSERT_ID();

这是所有分布式系统大忌,不推荐。

五、方案2:Redis INCR

css 复制代码
// Redis 命令
INCR order:id
// Go调用
id,_:=rdb.Incr(ctx,"order:id").Result()

优点:实现简单;严格递增。

缺点:Redis单点;网络开销;QPS上限。

适合低并发系统/管理后台

六、方案3:Segment号段(美团/京东订单号)

思路:

DB中维护一个号段

一次取一段(例如:1000个)

本地内存自增

表结构:

sql 复制代码
CREATE TABLE id_segment (
  biz_tag VARCHAR(64) PRIMARY KEY,
  max_id BIGINT,
  step INT
);
go 复制代码
type Segment struct {
	cur  int64
	max  int64
}

func (s *Segment) Next() int64 {
if s.cur >= s.max {
		s.reload()
	}
	s.cur++
return s.cur
}

优点:严格递增;ID短

缺点:依赖DB;实现复杂;冷启动慢。

适合订单号、流水号

七、方案4:雪花算法(Snowflake)

  1. ID结构
apache 复制代码
0 | 41bit 时间戳 | 10bit 机器ID | 12bit 序列号

时间戳:毫秒

机器ID:数据中心+worker

序列号:同毫秒并发

趋势递增、完全本地生成、无依赖

  1. go 代码实现
go 复制代码
package snowflake
import (
    "errors"
    "strconv"
    "sync"
    "time"
)
// 常量定义
const (
    workerBits   = 10                      // 工作节点位数
    seqBits      = 12                      // 序列号位数
    workerMax    = -1 ^ (-1 << workerBits) // 工作节点最大ID
    seqMask      = -1 ^ (-1 << seqBits)    // 序列号掩码
    timeShift    = workerBits + seqBits    // 时间戳左移位数
    workerShift  = seqBits                 // 工作节点左移位数
    defaultEpoch = int64(1672531200000)    // 默认起始时间戳 (2023-01-01)
)
// ID 自定义类型,用于区分雪花ID和普通int64
type ID int64
// Snowflake 雪花算法生成器
type Snowflake struct {
    mu       sync.Mutex
    lastTime int64
    workerID int64
    sequence int64
    epoch    int64
}
// New 创建雪花算法生成器
// workerID: 工作节点ID,范围 0~1023
// 返回错误如果workerID超出范围
func New(workerID int64) (*Snowflake, error) {
    return NewWithEpoch(workerID, defaultEpoch)
}
// NewWithEpoch 创建带自定义起始时间的雪花算法生成器
// workerID: 工作节点ID,范围 0~1023
// epoch: 自定义起始时间戳(毫秒)
// 返回错误如果workerID超出范围
func NewWithEpoch(workerID int64, epoch int64) (*Snowflake, error) {
    if workerID < 0 || workerID > workerMax {
        return nil, errors.New("worker ID out of range [0, 1023]")
    }
    return &Snowflake{
        workerID: workerID,
        epoch:    epoch,
    }, nil
}
// NextID 生成下一个雪花ID
// 返回ID类型的雪花ID和可能的错误
func (s *Snowflake) NextID() (ID, error) {
    s.mu.Lock()
    defer s.mu.Unlock()
    now := time.Now().UnixMilli()
    // 处理时间回拨
    if now < s.lastTime {
        return 0, errors.New("time is back, ID generation failed")
    }
    if now == s.lastTime {
        // 同一毫秒内,递增序列号
        s.sequence = (s.sequence + 1) & seqMask
        // 序列号耗尽,等待下一个毫秒
        if s.sequence == 0 {
            // 使用短暂休眠代替自旋等待,减少CPU占用
            time.Sleep(time.Millisecond)
            now = time.Now().UnixMilli()
            // 处理时间回拨(再次检查)
            if now < s.lastTime {
                return 0, errors.New("time is back, ID generation failed")
            }
            s.lastTime = now
            s.sequence = 0
        }
    } else {
        // 新的毫秒,重置序列号
        s.lastTime = now
        s.sequence = 0
    }
    // 生成ID
    id := ((now - s.epoch) << timeShift) |
        (s.workerID << workerShift) |
        s.sequence
    return ID(id), nil
}
// ParseID 解析雪花ID
// 返回ID的各组成部分:时间戳、工作节点ID、序列号
func ParseID(id ID, epoch int64) (time.Time, int64, int64) {
    idInt := int64(id)
    timestamp := (idInt >> timeShift) + epoch
    workerID := (idInt >> workerShift) & ((1 << workerBits) - 1)
    sequence := idInt & seqMask
    return time.UnixMilli(timestamp), workerID, sequence
}
// String 将ID转换为字符串
func (id ID) String() string {
    return strconv.FormatInt(int64(id), 10)
}
// Int64 将ID转换为int64
func (id ID) Int64() int64 {
    return int64(id)
}
go 复制代码
//如何使用
sf := snowflake.New(1)
id := sf.NextID()
  1. 雪花算法工程注意点:
  • 时间回拨问题

  • NTP同步导致时间倒退

    解决方案:

  • 禁止自动回拨

  • 检测回拨直接panic/等待

  • 使用逻辑时间

八、如何选?

|----------|---------|
| 场景 | 推荐 |
| 单体 | DB |
| 小系统/快速上线 | Redis |
| 订单/财务流水 | Segment |
| 微服务/高并发 | 雪花算法 |

友情链接:加班费计算器(vx小程序搜索"加班计")

*源码地址*

私给


如果您喜欢这篇文章,请您(点赞、分享、亮爱心),万分感谢!

相关推荐
Hazenix21 分钟前
Go 指南:一篇文章速通 Golang
开发语言·后端·golang
灯澜忆梦24 分钟前
GO_复合类型---指针
开发语言·后端·golang
wuqingshun3141591 小时前
MYSQL默认的事务隔离级别是什么?为什么选择这个级别?
数据库·mysql
空中湖1 小时前
电池简史(五):极限之梦与终章——锂硫、锂空气、量子电池,以及你该知道的电池投资与生活指南
数据库·程序人生
名字还没想好☜2 小时前
Go slice 的 append 陷阱:共享底层数组导致的数据串改
开发语言·后端·golang·go·slice
蓝胖子酱2 小时前
在 Electron 中用 electron-store 实现加密存储与自定义格式文件
前端·javascript·数据库
李昊哲小课2 小时前
ubuntu26.04搭建kafka集群
分布式·kafka
灯澜忆梦3 小时前
GO_面向对象_方法
开发语言·golang
布朗克1683 小时前
Go 入门到精通-16-字符串深入
开发语言·后端·golang·字符串
北冥you鱼3 小时前
Go flag 包详解:从命令行解析到实战应用
开发语言·后端·golang