应急响应新年促销

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):
        now = time.time()
        if self.tokens < tokens:
            return False
        else:
            self.tokens -= tokens
            return True

    def refill(self):
        now = time.time()
        delta = self.fill_rate * (now - self.timestamp)
        self.tokens = min(self.capacity, self.tokens + delta)
        self.timestamp = now

# 使用示例
bucket = TokenBucket(capacity=100, fill_rate=10)

def handle_request():
    if bucket.consume(1):
        print("Request processed")
    else:
        print("Too many requests, please try again later")

# 模拟每秒请求
while True:
    handle_request()
    bucket.refill()
    time.sleep(0.1)