python限制调用接口速度

复制代码
import time
import requests
import pandas as pd
from concurrent.futures import ThreadPoolExecutor


class RateLimiter:
    def __init__(self, max_calls, period):
        self.max_calls = max_calls
        self.period = period
        self.calls = 0
        self.reset_time = time.time()

    def can_call(self):
        current_time = time.time()
        if current_time - self.reset_time > self.period:
            self.reset_time = current_time
            self.calls = 0
        return self.calls < self.max_calls

    def called(self):
        self.calls += 1

在这里,RateLimiter类通过在指定时间段内限制最大调用次数来实现速率限制。构造函数__init__接受两个参数:max_calls表示最大调用次数,period表示时间段(以秒为单位)。

类中的can_call方法用于判断是否可以进行调用。它首先检查当前时间与上次重置时间之间的时间差是否超过设定的时间段。如果超过了时间段,则重置计数器,并更新重置时间和调用次数。最后,判断调用次数是否小于最大调用次数,如果是则返回True,表示可以进行调用;否则返回False,表示已经达到限制。

called方法用于在每次调用后更新调用次数。当函数或接口被调用时,应该在合适的位置调用called方法,以便在限制内适当增加调用次数。

这种速率限制器通常用于控制对外部API的访问次数,以免超出API提供商的规定限制。它能够平滑地限制请求速率,避免因短时间内大量请求而导致服务崩溃或被封禁。

相关推荐
阮瑭雅6 分钟前
Bash语言的微服务
开发语言·后端·golang
乌旭23 分钟前
量子计算入门:Qiskit实战量子门电路设计
人工智能·pytorch·python·深度学习·transformer·量子计算
霍徵琅31 分钟前
CSS语言的硬件驱动
开发语言·后端·golang
霍珵蕴33 分钟前
Lisp语言的计算机视觉
开发语言·后端·golang
褚翾澜34 分钟前
Lisp语言的无线通信
开发语言·后端·golang
weixin_3077791336 分钟前
判断HiveQL语句为ALTER TABLE语句的识别函数
开发语言·数据仓库·hive·c#
Niuguangshuo43 分钟前
Python设计模式:责任链模式
开发语言·python·责任链模式
甄霓裳1 小时前
APL语言的游戏音效
开发语言·后端·golang
小杨4041 小时前
python入门系列十五(asyncio)
人工智能·python·pycharm
爱学习的capoo1 小时前
对应列表数据的分割和分组
python·pandas