Python雪花代码

"雪花"在编程中通常指的是雪花算法，这是一种用于生成唯一ID的算法。以下是一个简易的Python实现：

import time
import threading

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

    def get_timestamp(self):
        return int(time.time() * 1000)

    def wait_for_next_millis(self, last_timestamp):
        timestamp = self.get_timestamp()
        while timestamp <= last_timestamp:
            timestamp = self.get_timestamp()
        return timestamp

    def generate_id(self):
        timestamp = self.get_timestamp()

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

        if timestamp == self.last_timestamp:
            self.sequence = (self.sequence + 1) & 0xFFF
            if self.sequence == 0:
                timestamp = self.wait_for_next_millis(self.last_timestamp)
        else:
            self.sequence = 0

        self.last_timestamp = timestamp

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

snowflake = SnowFlake()
print(snowflake.generate_id())

 

 

在这段代码中，SnowFlake类用于生成唯一的ID。每次调用 generate_id方法时，它都会生成一个新的、唯一的ID。这个ID由时间戳、数据中心ID、工作机器ID和序列号组成，确保了在不同的时间、不同的数据中心、不同的工作机器上都能生成唯一的ID。