分布式ID工具(snowflake)

Distributed ID for snowflake implementation 

package com.marcosbarbero.cloud.autoconfigure.zuul.ratelimit.support;

import java.util.concurrent.ThreadLocalRandom;

/**
 * @date 19-3-27
 * @auther jackliang
 * @description TODO
 */
public class SimpleFlakeGenerator {

        private static class SimpleFlake {

            private static SimpleFlake simpleFlake = new SimpleFlake(1, 1, (2 >>> 8));
            private long workerId; // machine id assert < 31 bit
            private long datacenterId;  // application id  assert < 31 bit
            /**
             * The 12-bit count sequence number supports each node to
             * generate 4096 ID numbers per millisecond (same machine, same time cut)
             */
            //Different serial numbers of different services according to business classification
            private long sequence;
            private long twepoch = 1553688313603L; // Start using component time difference 2019-3-27 x:x:x:
            private long workerIdBits = 5L; // 5 bit
            private long datacenterIdBits = 5L; // 5bit
            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;

            private SimpleFlake(long workerId, long datacenterId, long sequence)
            {

                this.workerId = workerId;
                this.datacenterId = datacenterId;
                this.sequence = sequence;
            }

            /**
             * Default flake method
             *
             * @return
             */
            public static SimpleFlake getDefaultFlake()
            {
                return simpleFlake;
            }

            /**
             * Here different instances are cached according to different services
             *
             * @param workerId
             * @param datacenterId
             * @param sequence
             * @return
             */
            public static SimpleFlake getCustomizeFlake(long workerId, long datacenterId, long sequence)
            {
                return new SimpleFlake(workerId, datacenterId, sequence);
            }

            public synchronized long nextId()
            {
                long timestamp = unixCurrentTime();

                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));
                }

                if (lastTimestamp == timestamp) {
                    sequence = (sequence + 1) & sequenceMask;
                    if (sequence == 0) {
                        timestamp = tilNextMillis(lastTimestamp);
                    }
                } else {
                    sequence = ThreadLocalRandom.current().nextInt(0, 6);
                }

                lastTimestamp = timestamp;
                return ((timestamp - twepoch) << timestampLeftShift) |
                        (datacenterId << datacenterIdShift) |
                        (workerId << workerIdShift) |
                        sequence;
            }

            /**
             * Secure the next timestamp peak
             * @param lastTimestamp
             * @return
             */
            private long tilNextMillis(long lastTimestamp)
            {
                long timestamp = unixCurrentTime();

                while (timestamp <= lastTimestamp)
                {
                    timestamp = unixCurrentTime();
                }
                return timestamp;
            }

            /**
             * current time for unix
             * @return
             */
            private long unixCurrentTime()
            {
                return System.currentTimeMillis();
            }

        }

    //---------------test---------------
    public static void main(String[] args) {
//        SimpleFlakeGenerator worker = new SimpleFlakeGenerator(1, 1, 1);
         SimpleFlake simpleFlake = SimpleFlake.getDefaultFlake();

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


}

 

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值