雪花算法，在分布式环境下实现高效的ID生成

ahardstone

已于 2023-08-01 19:33:41 修改

阅读量1.7k

点赞数 2

分类专栏：算法文章标签：算法分布式

于 2023-07-29 17:08:44 首次发布

本文链接：https://blog.youkuaiyun.com/gfyy_bkj/article/details/131997075

版权

算法专栏收录该内容

21 篇文章

订阅专栏

文章介绍了雪花算法，一种用于生成唯一ID的策略。它包括41位时间戳、10位机器码和12位序列号，支持毫秒级时间单位和多台机器的并行生成。在Java中，需要注意Long类型的位数以防止溢出。代码示例展示了如何在Java中实现雪花算法，包括处理时间回滚和序列号重置的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

其实雪花算法比较简单，可能称不上什么算法，就是一种构造UID的方法。
点1：UID是一个long类型的41位时间戳，10位存储机器码，12位存储序列号。
点2：时间戳的单位是毫秒，可以同时链接1024台机器，每台机器每毫秒可以使用4096个序列号，我们会给生成id上一个同步锁，阻塞住其他线程的访问。
点3：利用掩码我们可以检测序列是否溢出，如果溢出的话，就强制等待到下一毫秒。
点4：对了看文章的时候发现，前端js的long类型是2的53次方，而java是2的63次方，要注意溢出哦。


/**
 * @author hardstone
 * @since 29 July 2023(1690603385473)
 */
public class SnowFlakes {
    //开始的时间戳
    private final  long start = 1690603385473L;
    //机器标识长度5位
    private final long machineIdBits = 5L;
    //机器集群标识长度5位
    private final long centerIdBits = 5L;
    //序列标识所占位数12位
    private final long sequenceBits = 12L;
    //机器标识最大值
    private final long maxMachineId = -1L ^ (-1L << machineIdBits);
    //机器集群标识最大值
    private final long maxCenterId = -1L ^ (-1L << centerIdBits);
    //序列标识的最大值
    private final long sequenceMask = -1L ^ (-1L << sequenceBits);
    //机器标识左移长度
    private final long machineIdShift = sequenceBits;
    //机器集群标识左移长度
    private final long centerIdShift = sequenceBits + machineIdBits;
    //时间戳左移长度
    private final long timeStampIdShift = sequenceBits + machineIdBits + centerIdBits;
    //序列Id
    private long sequence = 0L;

    //机器Id
    private long machineId;

    //机器集群Id
    private long centerId;

    //时间戳
    private long lastTimeStamp = -1L;

    public SnowFlakes(long machineId, long centerId) {
        if (machineId > maxMachineId || machineId < 0) {
            throw new IllegalArgumentException(String.format("WorkerId should be between 0 and 31"));
        }
        if (centerId > maxCenterId || centerId < 0) {
            throw new IllegalArgumentException(String.format("CenterId should be between 0 and 31"));
        }
    }
    public synchronized long nextId() {
        long timeStamp = System.currentTimeMillis();
        //时间回滚现象
        if (timeStamp < lastTimeStamp) {
            throw new RuntimeException(
                    String.format("Time gone backwards!")
            );
        }
        if (lastTimeStamp == timeStamp) {
            sequence = (sequence + 1) & sequenceMask;
            //如果序列分配完了
            if (sequence == 0) {
                timeStamp = getNextMillis(lastTimeStamp);
            }
        } else {
            sequence = 0L;
        }
        lastTimeStamp = timeStamp;
        return ((timeStamp - start) << timeStampIdShift)
                | (centerId << centerIdShift)
                | (machineId << machineIdShift)
                | sequence;
    }
    protected long getNextMillis(long lastTimeStamp) {
        long timeStamp = System.currentTimeMillis();
        while (timeStamp <= lastTimeStamp) {
            timeStamp = System.currentTimeMillis();
        }
        return timeStamp;
    }
    public static void main(String[] args) {
        System.out.println(new SnowFlakes(0, 0).nextId());
    }
}