基于 redis 的简易滑动窗口实现

以下是一个使用Redis实现简易滑动窗口的Python示例代码：

import redis
import time
 
# 连接Redis
r = redis.Redis(host='localhost', port=6379, db=0)
 
# 滑动窗口限流的实现
def is_rate_limited(user_id, max_requests, window_size):
    key = f'rate_limit:{user_id}'
    timestamp = time.time()
 
    # 请求数加一
    r.incr(key)
 
    # 定义窗口内最大请求数
    r.expire(key, window_size)
 
    # 获取窗口内的请求数
    requests = r.get(key)
 
    # 如果请求数超过限制，则返回True表示被限流
    if requests and int(requests) > max_requests:
        return True
    else:
        return False
 
# 用户ID和最大请求数、窗口大小（秒）
user_id = 'user123'
max_requests = 5
window_size = 60
 
# 检查是否被限流
if is_rate_limited(user_id, max_requests, window_size):
    print("被限流了")
else:
    print("没有被限流")

这段代码定义了一个is_rate_limited函数，它使用Redis来跟踪给定用户ID的请求数，并且如果窗口内的请求数超过最大请求限制，则返回True表示被限流。每次调用该函数，请求数都会增加，并且窗口会被重置。这个简易的实现没有考虑多线程/进程的竞争条件，但它可以作为一个基本的滑动窗口限流策略的示范。