使用redis进行限流

在使用Redis进行限流时，一种常见的方法是利用Redis的原子操作INCR和EXPIRE结合使用。以下是一个使用Python和redis-py库实现的简单示例：

import redis
 
def is_rate_limited(redis_conn, key, max_requests, time_window):
    # 检查key是否已存在
    if not redis_conn.exists(key):
        # 如果不存在，设置过期时间
        redis_conn.expire(key, time_window)
        # 初始化计数器
        redis_conn.incr(key)
        return False
    # 检查计数器是否超过限制
    if redis_conn.get(key) > max_requests:
        return True
    # 如果没有超过限制，增加计数器
    redis_conn.incr(key)
    return False
 
# 配置Redis连接
redis_conn = redis.StrictRedis(host='localhost', port=6379, db=0)
 
# 调用函数进行限流
key = 'user_id:1234'  # 用户ID或其他标识符
max_requests = 100    # 时间窗口内允许的最大请求数
time_window = 60      # 时间窗口，单位为秒
 
if is_rate_limited(redis_conn, key, max_requests, time_window):
    print("You are being rate limited.")
else:
    print("Request is allowed.")

在这个例子中，is_rate_limited 函数检查给定的key是否存在。如果不存在，它会设置key的过期时间并将计数器初始化为1。如果key已存在，它会检查计数器的值是否超过了最大请求数；如果超过了，它会返回True表示请求应该被限流。如果没有超过最大请求数，它会将计数器增加1，并返回False表示请求被允许。