python 接口限流

"""

滑动窗口限流
假设有一个接口每秒最多请求100次，那么
定义一个数组，可以存储100次请求，把每次请求的时间存进去，
下次再来一个新请求，从头遍历这个数组，把时间超出1秒的请求剔除，
然后看一下数组里还有多少元素，如果小于100，说明可以请求，否则拒绝请求
"""
import time

MAX_REQUEST = 2  # 最大请求次数
INTERVAL = 1  # 1秒

requests = []

for i in range(100):
    print(f'requests={requests}')
    current_time = time.time()
    while len(requests):  # 每次请求到来，先清理过期的请求
        if current_time - requests[0] > INTERVAL:
            requests.pop(0)
        else:
            break

    if len(requests) < MAX_REQUEST:  # 如果小于请求次数，则允许请求
        requests.append(current_time)
    else:
        print(f'服务器繁忙，请求拒绝')
    time.sleep(0.1)