redis 实现滑动窗口限流

基于Redis实现滑动窗口限流的方法如下：

import time
import redis

class SlidingWindowRateLimiter:
    def __init__(self, redis_conn, window_size, limit):
        self.redis_conn = redis_conn
        self.window_size = window_size
        self.limit = limit

    def is_allowed(self, key):
        current_time = int(time.time())
        key_name = f"{key}:{current_time // self.window_size}"
        count = self.redis_conn.get(key_name)
        if count is None:
            count = 0
        else:
            count = int(count)
        if count < self.limit:
            self.redis_conn.incr(key_name)
            return True
        return False

# 使用示例
redis_conn = redis.Redis(host='localhost', port=6379, db=0)
limiter = SlidingWindowRateLimiter(redis_conn, window_size=60, limit=100)
if limiter.is_allowed("user1"):
    print("Allow the request")
else:
    print("Reject the request")

上述代码中，我们定义了一个`SlidingWindowRateLimiter`类，它接受Redis连接、窗口大小和限流次数作为参数。`is_allowed`方法用于判断是否允许请求通过限流。

在`is_allowed`方法中，我们首先获取当前时间，并根据窗口大小计算出当前窗口的名称。然后，我们从Redis中获取该窗口的请求数量。如果请求数量小于限流次数，我们将请求数量加1，并返回允许请求通过。否则，我们返回拒绝请求。

请注意，上述代码仅为示例，实际使用时需要根据具体情况进行适当的修改和优化。