python令牌桶_python 令牌桶

在Python中，令牌表示为`Token`。`Token`可以用来控制对某个资源的访问速度或并发数，以防止资源被过度使用，例如控制API接口的最大并发量。在Python中，可以通过实现一个令牌桶算法来实现令牌控制。

以下是一个简单的Python示例代码，演示如何实现基于令牌桶算法的令牌控制：```python

import time
 
class TokenBucket:
    def __init__(self, capacity, refill_rate):
        self.capacity = float(capacity)
        self.tokens = self.capacity
        self.refill_rate = float(refill_rate)
        self.last_refill_time = time.time()
 
    def try_acquire(self, tokens=1):
        if tokens <= self.tokens:
            self.tokens -= tokens
            return True
        else:
            return False
 
    def refill(self):
        now = time.time()
        delta_time = now - self.last_refill_time
        new_tokens = delta_time * self.refill_rate
        self.tokens += new_tokens
        self.tokens = min(self.tokens, self.capacity)
        self.last_refill_time = now
 
# 使用 TokenBucket 控制访问速度
bucket = TokenBucket(10, 1)  # 桶容量是 10 个 token，每秒钟恢复 1 个 token
 
while True:
    if bucket.try_acquire():
        print("get resource")
    else:
        print("rate limited")
    
    bucket.refill()  # 更新桶状态，恢复 token
    
    time.sleep(0.1)  # 每 0.1 秒尝试获取资源

```

在上面的代码中，`TokenBucket`类实现了令牌桶算法。`try_acquire()`方法用来尝试获取令牌，如果成功则返回True，否则返回False。`refill()`方法用来更新令牌桶的状态，在这里我们使用了固定速率的令牌补充策略。最后的while循环中，我们不断地尝试获取资源并更新桶状态。

容量（capacity）和刷新速率（refill_rate）。同时也可以根据自己的需要修改程序中 while 循环内部的逻辑，以适应不同场景下的需求。