双十一API 网关购买

import time
from threading import Lock

class TokenBucket:
    def __init__(self, capacity, refill_rate):
        self.capacity = capacity
        self.tokens = capacity
        self.refill_rate = refill_rate
        self.last_refill_time = time.time()
        self.lock = Lock()

    def refill(self):
        now = time.time()
        tokens_to_add = (now - self.last_refill_time) * self.refill_rate
        with self.lock:
            self.tokens = min(self.capacity, self.tokens + tokens_to_add)
            self.last_refill_time = now

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

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

def handle_request():
    if bucket.consume(1):
        # 处理请求
        print("Request handled successfully")
    else:
        # 请求被限流
        print("Request throttled")

# 模拟高并发请求
for _ in range(150):
    handle_request()
    time.sleep(0.1)