雪花算法实现java

原创

mob649e8160b585 2024-09-12 04:04:40 ©著作权

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

雪花算法实现Java

引言

在现代分布式系统中，生成唯一标识符（ID）是一个非常重要的需求。传统的自增ID方式虽然简单快捷，但在分布式环境下，容易产生冲突。为了解决这个问题，雪花算法（Snowflake Algorithm）应运而生。它可以生成唯一且有序的64位ID，同时具备高性能和高可扩展性。本文将详细介绍雪花算法的原理，并通过Java代码示例来实现这一算法。

雪花算法的原理

雪花算法由Twitter开发，其生成的ID有效减少了由冲突引起的访问数据库的成本。每个ID由以下几个部分组成：

符号位（1 bit）：始终为0
时间戳（41 bits）：毫秒级时间戳，支持69年的时间范围
数据中心ID（5 bits）：可支持最多32个数据中心
机器ID（5 bits）：可支持最多32台机器
序列号（12 bits）：同一毫秒内生成的ID序列，最多支持4096个ID

ID的组成结构

|  0  |  timestamp (41 bits)  |  datacenterId (5 bits)  |  workerId (5 bits)  |  sequence (12 bits)  |

流程图

下面是生成ID的基本流程：

flowchart TD
    A[开始] --> B{获取当前时间}
    B --> |当前时间大于上次时间| C[更新时间戳]
    B --> |当前时间小于上次时间| D[等待下个毫秒]
    C --> E[生成序列号]
    E --> F[拼接所有部分生成ID]
    F --> G[返回ID]
    D --> B

Java实现

接下来，我们来实现雪花算法的Java代码。代码将涵盖ID生成器的各个方面，包括时间戳、数据中心ID、机器ID以及序列号的逻辑处理。

ID生成器代码示例

public class SnowflakeIdGenerator {
    // 机器ID部分
    private final long workerId;
    private final long datacenterId;
    
    // 机器ID最大值
    private final static long MAX_WORKER_ID = 31;
    // 数据中心ID最大值
    private final static long MAX_DATACENTER_ID = 31;

    // 时间戳偏移量
    private final static long TIMESTAMP_LEFT_SHIFT = 22;
    // 数据中心ID偏移量
    private final static long DATACENTER_ID_SHIFT = 17;
    // 机器ID偏移量
    private final static long WORKER_ID_SHIFT = 12;
    
    // 序列号的最大值
    private final static long SEQUENCE_MASK = 4095;
    
    private long sequence = 0L;
    private long lastTimestamp = -1L;

    public SnowflakeIdGenerator(long workerId, long datacenterId) {
        if (workerId > MAX_WORKER_ID || workerId < 0) {
            throw new IllegalArgumentException("Invalid worker ID");
        }
        if (datacenterId > MAX_DATACENTER_ID || datacenterId < 0) {
            throw new IllegalArgumentException("Invalid datacenter ID");
        }
        this.workerId = workerId;
        this.datacenterId = datacenterId;
    }

    public synchronized long nextId() {
        long timestamp = System.currentTimeMillis();

        if (timestamp < lastTimestamp) {
            throw new RuntimeException("Clock moved backwards");
        }
        
        if (lastTimestamp == timestamp) {
            sequence = (sequence + 1) & SEQUENCE_MASK;
            if (sequence == 0) {
                timestamp = waitNextMillis(lastTimestamp);
            }
        } else {
            sequence = 0L;
        }

        lastTimestamp = timestamp;

        return ((timestamp << TIMESTAMP_LEFT_SHIFT) |
                (datacenterId << DATACENTER_ID_SHIFT) |
                (workerId << WORKER_ID_SHIFT) |
                sequence);
    }

    private long waitNextMillis(long lastTimestamp) {
        long timestamp = System.currentTimeMillis();
        while (timestamp <= lastTimestamp) {
            timestamp = System.currentTimeMillis();
        }
        return timestamp;
    }
}

代码解析

构造函数：接收机器ID和数据中心ID，并检查其有效性。
nextId() 方法：生成唯一ID的核心逻辑。
- 获取当前时间戳并与上次时间戳进行比较。
- 如果当前时间与上次时间相同，则生成序列号并处理溢出情况。
- 拼接各部分生成最终ID。
waitNextMillis() 方法：当序列号达到最大值时，等待下一个毫秒。