java snowflake算法workId

原创

mob64ca12e7b5cf 2024-09-13 04:55:08 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12e7b5cf的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java Snowflake算法中的workId解析

在分布式系统中，生成唯一标识符（ID）是一个常见的需求。Twitter开发的Snowflake算法是一种高效且可扩展的ID生成策略。它能生成唯一且有序的64位ID，广泛应用于大规模应用中。本文将重点讨论Java实现中的workId的概念，并提供相关的代码示例。

什么是WorkId？

在Snowflake算法中，生成的ID由以下几部分组成：

时间戳（41位）：表示从某个基准时间开始的毫秒数。
机器标识符（10位）：用来标识不同的机器。这部分通常由数据中心ID和机器ID组合而成。
序列号（12位）：同一毫秒内生成的ID序列，确保在同一时间戳下仍然能生成唯一的ID。

workId主要用于区分不同的工作节点（或机器），确保在同一时间生成的ID是独特的。

Snowflake算法的实现

下面是一个简单的Java实现Snowflake算法的示例，包含workId的设置：

public class SnowflakeIdWorker {
    // 开始时间戳
    private final long epoch = 1483228800000L; // 2017-01-01 00:00:00
    // 数据中心ID部分的位数
    private final long datacenterIdBits = 5L; 
    // 机器ID部分的位数
    private final long workerIdBits = 5L; 
    // 支持的最大数据中心ID，结果是31
    private final long maxDatacenterId = -1L ^ (-1L << datacenterIdBits);
    // 支持的最大机器ID，结果是31
    private final long maxWorkerId = -1L ^ (-1L << workerIdBits);
    // 序列号部分的位数
    private final long sequenceBits = 12L; 
    // 数据中心ID向左移位
    private final long datacenterIdShift = sequenceBits + workerIdBits; 
    // 机器ID向左移位
    private final long workerIdShift = sequenceBits; 
    // 时间戳向左移位
    private final long timestampLeftShift = sequenceBits + workerIdBits + datacenterIdBits; 

    private long datacenterId;
    private long workerId;
    private long sequence = 0L; 
    private long lastTimestamp = -1L; 

    public SnowflakeIdWorker(long workerId, long datacenterId) {
        if (workerId > maxWorkerId || workerId < 0) {
            throw new IllegalArgumentException("Worker ID can't be greater than 31 or less than 0");
        }
        if (datacenterId > maxDatacenterId || datacenterId < 0) {
            throw new IllegalArgumentException("Datacenter ID can't be greater than 31 or less than 0");
        }
        this.workerId = workerId;
        this.datacenterId = datacenterId;
    }

    public synchronized long nextId() {
        long timestamp = System.currentTimeMillis();
        
        // 如果当前时间小于上一次生成ID的时间
        if (timestamp < lastTimestamp) {
            throw new RuntimeException("Clock moved backwards. Refusing to generate id");
        }

        // 如果是同一毫秒，序列号加一
        if (lastTimestamp == timestamp) {
            sequence = (sequence + 1) & maxWorkerId;
        } else {
            sequence = 0;
        }

        lastTimestamp = timestamp;

        // 返回生成的ID
        return ((timestamp - epoch) << timestampLeftShift) |
               (datacenterId << datacenterIdShift) |
               (workerId << workerIdShift) |
               sequence;
    }
}

工作流程

通过以下的旅程图，我们可以理解Snowflake算法的工作流程：

journey
    title Snowflake ID Generation Journey
    section ID Generation
      Application requests unique ID                 :start
      Get current timestamp                            :point
      Check for machine ID and datacenter ID        :point
      Generate ID based on timestamp, machine ID     :point
      Return unique ID                                :end