Python 实现 API 速率限制（Rate Limiting）

最新推荐文章于 2025-05-17 17:07:22 发布

日落邮局ka

最新推荐文章于 2025-05-17 17:07:22 发布

阅读量378

点赞数 3

文章标签： python 开发语言

本文链接：https://blog.youkuaiyun.com/2501_91228273/article/details/146608135

版权

```html Python 实现 API 速率限制（Rate Limiting）

Python 实现 API 速率限制（Rate Limiting）

API 速率限制（Rate Limiting）是一种常见的机制，用于控制客户端对服务器的请求频率。它能够防止滥用 API 资源，保护服务器免受过多请求的影响，并确保公平地分配资源。在本文中，我们将介绍如何使用 Python 实现 API 的速率限制功能。

什么是速率限制？

速率限制是一种限制客户端在特定时间窗口内可以发送的请求数量的技术。例如，一个 API 可能规定每个用户每分钟只能发送 60 次请求。如果超过这个限制，服务器可能会返回错误信息或拒绝服务。

为什么需要速率限制？

API 速率限制的主要目的是：

保护服务器免受过多请求的冲击。
确保公平性，避免某些客户端占用过多资源。
减少不必要的网络流量。
提高系统的稳定性和可靠性。

实现速率限制的常用方法

实现速率限制的方法有很多，常见的有以下几种：

基于时间窗口的计数器：记录每个时间窗口内的请求数量。
令牌桶算法：模拟一个桶，根据固定速率填充令牌，客户端每次请求消耗一个令牌。
漏桶算法：将请求放入一个桶中，以固定速率处理请求。

基于时间窗口的计数器实现

我们首先来看一种简单的实现方式——基于时间窗口的计数器。这种方法通过记录每个时间窗口内的请求数量来限制请求频率。

代码实现


from time import time

class RateLimiter:
    def __init__(self, max_requests, time_window):
        self.max_requests = max_requests  # 最大请求数
        self.time_window = time_window    # 时间窗口（秒）
        self.requests = []                # 存储请求时间戳

    def allow_request(self):
        current_time = time()
        # 清理过期的请求
        self.requests = [t for t in self.requests if t > current_time - self.time_window]
        
        if len(self.requests) < self.max_requests:
            self.requests.append(current_time)
            return True
        else:
            return False

# 示例用法
limiter = RateLimiter(max_requests=5, time_window=60)  # 每分钟最多允许 5 次请求

for i in range(7):
    if limiter.allow_request():
        print(f"请求 {i + 1} 成功")
    else:
        print(f"请求 {i + 1} 被拒绝")