雪花算法运用

雪花算法实现
继上篇//www.greatytc.com/p/c83715d7dfaa分析了分布式id问题,说到了雪花算法,本着知其所以然的想法,看一看雪花算法如何实现的,很多大厂也基于雪花算法封装了自己的分布式id生成算法。

核心思想

使用一个 64 bit 的 long 型的数字作为全局唯一 id。在分布式系统中的应用十分广泛,且ID 引入了时间戳,基本上保持自增的。

举个例子

image.png
  • 1 符号位 0表示正数
  • 时间戳 41位可表示69年
  • 机房id 5位最大32
  • 机器id 5位最大32
  • 序列号,同一毫秒内生成的id顺序 有4096个

使用流程

需要分布式id时请求部署雪花算法获得id
注意:

  • 算法在启动时传入机器机房id,配置文件及java启动文件
    需要配置在bootstrap.yml,设置启动参数-Dserver.workId -Dserver.datacenterId
  • 算法需要做成单例模式

算法实现

网上大体都有,我复制了一个可运行的注释版;
思路:
时间戳 | 机房 | 机器 | 序列号
时间戳可以减去一个固定时间值,左移22位
机房机器id实际不需要,可以简单点生成如机房为1机器1-n,左移12位
序列号在同一毫秒一个则为0,多个则递增
最后按位或运算,生成一个19位的long型,且递增姿势,对数据库友好
运行结果:
1229400235483205632
1229400235491594240
1229400235491594241
1229400235491594242
1229400235491594243
1229400235495788544
1229400235495788545
1229400235495788546
1229400235495788547

public class SnowFlake {
        private long workerId; // 这个就是代表了机器id
        private long datacenterId; // 这个就是代表了机房id
        private long sequence; // 这个就是代表了一毫秒内生成的多个id的最新序号
        public SnowFlake(long workerId, long datacenterId, long sequence) {
            // sanity check for workerId
            // 这儿不就检查了一下,要求就是你传递进来的机房id和机器id不能超过32,不能小于0
            if (workerId > maxWorkerId || workerId < 0) {

                throw new IllegalArgumentException(
                        String.format("worker Id can't be greater than %d or less than 0",maxWorkerId));
            }

            if (datacenterId > maxDatacenterId || datacenterId < 0) {

                throw new IllegalArgumentException(
                        String.format("datacenter Id can't be greater than %d or less than 0",maxDatacenterId));
            }
            this.workerId = workerId;
            this.datacenterId = datacenterId;
            this.sequence = sequence;
        }
        private long twepoch = 1288834974657L;
        private long workerIdBits = 5L;
        private long datacenterIdBits = 5L;

        // 这个是二进制运算,就是5 bit最多只能有31个数字,也就是说机器id最多只能是32以内
        private long maxWorkerId = -1L ^ (-1L << workerIdBits);
        // 这个是一个意思,就是5 bit最多只能有31个数字,机房id最多只能是32以内
        private long maxDatacenterId = -1L ^ (-1L << datacenterIdBits);
        private long sequenceBits = 12L;
        private long workerIdShift = sequenceBits;
        private long datacenterIdShift = sequenceBits + workerIdBits;
        private long timestampLeftShift = sequenceBits + workerIdBits + datacenterIdBits;
        private long sequenceMask = -1L ^ (-1L << sequenceBits);
        private long lastTimestamp = -1L;
        public long getWorkerId(){
            return workerId;
        }
        public long getDatacenterId() {
            return datacenterId;
        }
        public long getTimestamp() {
            return System.currentTimeMillis();
        }
        // 这个是核心方法,通过调用nextId()方法,让当前这台机器上的snowflake算法程序生成一个全局唯一的id
        public synchronized long nextId() {
            // 这儿就是获取当前时间戳,单位是毫秒
            long timestamp = timeGen();
            if (timestamp < lastTimestamp) {
                System.err.printf(
                        "clock is moving backwards. Rejecting requests until %d.", lastTimestamp);
                throw new RuntimeException(
                        String.format("Clock moved backwards. Refusing to generate id for %d milliseconds",
                                lastTimestamp - timestamp));
            }

            // 下面是说假设在同一个毫秒内,又发送了一个请求生成一个id
            // 这个时候就得把seqence序号给递增1,最多就是4096
            if (lastTimestamp == timestamp) {

                // 这个意思是说一个毫秒内最多只能有4096个数字,无论你传递多少进来,
                //这个位运算保证始终就是在4096这个范围内,避免你自己传递个sequence超过了4096这个范围
                sequence = (sequence + 1) & sequenceMask;
                if (sequence == 0) {
                    timestamp = tilNextMillis(lastTimestamp);
                }

            } else {
                sequence = 0;
            }
            // 这儿记录一下最近一次生成id的时间戳,单位是毫秒
            lastTimestamp = timestamp;
            // 这儿就是最核心的二进制位运算操作,生成一个64bit的id
            // 先将当前时间戳左移,放到41 bit那儿;将机房id左移放到5 bit那儿;将机器id左移放到5 bit那儿;将序号放最后12 bit
            // 最后拼接起来成一个64 bit的二进制数字,转换成10进制就是个long型
            System.out.println(timestamp);
            System.out.println(datacenterId);
            System.out.println(workerId);
            System.out.println(sequence);
            return ((timestamp - twepoch) << timestampLeftShift) |
                    (datacenterId << datacenterIdShift) |
                    (workerId << workerIdShift) | sequence;
        }
        private long tilNextMillis(long lastTimestamp) {

            long timestamp = timeGen();

            while (timestamp <= lastTimestamp) {
                timestamp = timeGen();
            }
            return timestamp;
        }
        private long timeGen(){
            return System.currentTimeMillis();
        }
        public static void main(String[] args) {

            SnowFlake worker = new SnowFlake(1,1,1);

            for (int i = 0; i < 30; i++) {
                System.out.println(worker.nextId());
            }
        }
    }

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容