在go语言生成唯一ID之SnowFlake算法
Go / 管理员 发布于 2年前 559
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())
}
ps:
在示例中,以1为 workerId 生成了一个唯一的ID,你可以根据实际情况来使用不同的workerId。
总结:
在上述代码中,我们定义了一个 snowflake 结构体,其中包含了上一次生成的时间戳、workerId 和 sequence 三个属性。
在 newSnowflake 函数中,我们检查传入的 workerId 是否在合理范围内,然后初始化 snowflake 结构体的属性。
在 NextId 函数中,我们首先获取当前时间戳,如果当前时间戳小于上一次生成的时间戳,说明时钟发生了回拨,此时需要抛出异常。
如果当前时间戳等于上一次生成的时间戳,说明在同一毫秒内生成了多个 ID,此时需要自增 sequence,如果 sequence 达到了最大值,则需要等到下一个毫秒。
如果当前时间戳大于上一次生成的时间戳,说明进入了下一个毫秒,此时需要将 sequence 重置为 0,并更新 lastTimestamp。
最后,根据时间戳、workerId 和 sequence 生成唯一的 ID。
122 在
学历:一种延缓就业设计,生活需求下的权衡之选中评论 工作几年后,报名考研了,到现在还没认真学习备考,迷茫中。作为一名北漂互联网打工人..123 在
Clash for Windows作者删库跑路了,github已404中评论 按理说只要你在国内,所有的流量进出都在监控范围内,不管你怎么隐藏也没用,想搞你分..原梓番博客 在
在Laravel框架中使用模型Model分表最简单的方法中评论 好久好久都没看友情链接申请了,今天刚看,已经添加。..博主 在
佛跳墙vpn软件不会用?上不了网?佛跳墙vpn常见问题以及解决办法中评论 @1111老铁这个不行了,可以看看近期评论的其他文章..1111 在
佛跳墙vpn软件不会用?上不了网?佛跳墙vpn常见问题以及解决办法中评论 网站不能打开,博主百忙中能否发个APP下载链接,佛跳墙或极光..
Copyright·© 2019 侯体宗版权所有·
粤ICP备20027696号