python雪花算法（简易实现版）

简介

雪花算法是一种分布式唯一ID生成算法，其特点在于结合了时间戳、数据中心ID和工作机器ID来生成全局唯一的ID，适合在分布式系统中生成有序且唯一的ID。

代码如下

import time

class Snowflake:
    def __init__(self, data_center_id, worker_id):
        self.data_center_id = data_center_id
        self.worker_id = worker_id
        self.sequence = 0
        self.last_timestamp = -1

    def next_id(self):
        timestamp = int(time.time() * 1000)

        if timestamp < self.last_timestamp:
            raise Exception("Clock moved backwards. Refusing to generate id")

        if timestamp == self.last_timestamp:
            self.sequence = (self.sequence + 1) & 4095
            if self.sequence == 0:
                timestamp = self.wait_next_millis(timestamp)
        else:
            self.sequence = 0

        self.last_timestamp = timestamp

        return ((timestamp - 1288834974657) << 22) | (self.data_center_id << 17) | (self.worker_id << 12) | self.sequence

    def wait_next_millis(self, last_timestamp):
        timestamp = int(time.time() * 1000)
        while timestamp <= last_timestamp:
            timestamp = int(time.time() * 1000)
        return timestamp

# 示例用法
snowflake = Snowflake(data_center_id=0, worker_id=0)

# 生成下一个唯一ID
for i in range(10):
    next_id = snowflake.next_id()
    print(i,":",next_id)

生成结果