网站安全加速限时秒杀

import time

class TokenBucket:
    def __init__(self, capacity, fill_rate):
        self.capacity = float(capacity)
        self.tokens = float(capacity)
        self.fill_rate = float(fill_rate)
        self.timestamp = time.time()

    def consume(self, tokens):
        if tokens <= self.tokens:
            self.tokens -= tokens
            return True
        else:
            return False

    def get_tokens(self):
        now = time.time()
        if self.tokens < self.capacity:
            delta = self.fill_rate * (now - self.timestamp)
            self.tokens = min(self.capacity, self.tokens + delta)
        self.timestamp = now
        return self.tokens

# 使用示例
bucket = TokenBucket(capacity=100, fill_rate=10)  # 容量为100，每秒填充10个令牌
if bucket.consume(1):  # 尝试消费1个令牌
    print("请求通过")
else:
    print("请求被限流")