首页 > 后端开发 > Golang > 正文

一文详解使用snowflake算法生成唯一id

藏色散人
发布: 2023-04-19 16:16:22
转载
1906人浏览过

go 语言中,可以使用 snowflake 算法生成唯一 id。snowflake 算法是 twitter 公司开源的一种分布式 id 生成算法,它可以在分布式系统中生成唯一的 id,保证 id 的全局唯一性,而且生成的 id 是按照时间递增的。

以下是使用 Go 语言实现 Snowflake 算法生成唯一 id 的示例代码,在线运行:

package main
import (
    "fmt"
    "time"
)
const (
    workerBits uint8 = 10
    seqBits    uint8 = 12
    maxWorkerNum int64 = -1 ^ (-1 << workerBits)
    maxSeqNum    int64 = -1 ^ (-1 << seqBits)
    timeShift   uint8 = workerBits + seqBits
    workerShift uint8 = seqBits
)
type snowflake struct {
    lastTimestamp int64
    workerId      int64
    sequence      int64
}
func newSnowflake(workerId int64) *snowflake {
    if workerId < 0 || workerId > maxWorkerNum {
        panic("workerId out of range")
    }
    return &snowflake{
        lastTimestamp: 0,
        workerId:      workerId,
        sequence:      0,
    }
}
func (sf *snowflake) NextId() int64 {
    timestamp := time.Now().UnixNano() / 1000000
    if timestamp < sf.lastTimestamp {
        panic("clock is moving backwards")
    }
    if timestamp == sf.lastTimestamp {
        sf.sequence = (sf.sequence + 1) & maxSeqNum
        if sf.sequence == 0 {
            for timestamp <= sf.lastTimestamp {
                timestamp = time.Now().UnixNano() / 1000000
            }
        }
    } else {
        sf.sequence = 0
    }
    sf.lastTimestamp = timestamp
    return (timestamp << timeShift) | (sf.workerId << workerShift) | sf.sequence
}
func main() {
    sf := newSnowflake(1)
    fmt.Println(sf.NextId())
}
登录后复制

在上述代码中,我们定义了一个 snowflake 结构体,其中包含了上一次生成的时间戳、workerId 和 sequence 三个属性。在 newSnowflake 函数中,我们检查传入的 workerId 是否在合理范围内,然后初始化 snowflake 结构体的属性。在 NextId 函数中,我们首先获取当前时间戳,如果当前时间戳小于上一次生成的时间戳,说明时钟发生了回拨,此时需要抛出异常。如果当前时间戳等于上一次生成的时间戳,说明在同一毫秒内生成了多个 ID,此时需要自增 sequence,如果 sequence 达到了最大值,则需要等到下一个毫秒。如果当前时间戳大于上一次生成的时间戳,说明进入了下一个毫秒,此时需要将 sequence 重置为 0,并更新 lastTimestamp。最后,根据时间戳、workerId 和 sequence 生成唯一的 ID。

在本示例中,我们设置 workerId=1 生成了一个唯一的 ID,你可以根据实际情况来使用不同的 workerId。

推荐学习:《go视频教程

以上就是一文详解使用snowflake算法生成唯一id的详细内容,更多请关注php中文网其它相关文章!

最佳 Windows 性能的顶级免费优化软件
最佳 Windows 性能的顶级免费优化软件

每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。

下载
来源:learnku网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习
PHP中文网抖音号
发现有趣的

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号