Redis:限流算法

一、限流算法分类

• 简单限流算法
• 漏斗限流算法(令牌桶算法)
• 分布式漏斗限流算法(分布式令牌桶算法)

二、简单限流算法

1. 问题引入

系统要限定用户的某个行为在指定的时间里只能允许发生 N 次，如何使用 Redis 的数据结构来实现这个限流的功能？

2. 解决方案

这个限流需求中存在一个滑动时间窗口，想想 zset 数据结构的 score 值，是不是可以通过 score 来圈出这个时间窗口来。而且我们只需要保留这个时间窗口，窗口之外的数据都可以砍掉。那这个 zset 的 value 填什么比较合适呢？它只需要保证唯一性即可，用 uuid 会比较浪费空间，那就改用毫秒时间戳吧。

3. 代码实现

# coding: utf8
import time
import redis

client = redis.StrictRedis()
def is_action_allowed(user_id, action_key, period, max_count):
	key = 'hist:%s:%s' % (user_id, action_key)
	now_ts = int(time.time() * 1000) # 毫秒时间戳
	with client.pipeline() as pipe: # client 是 StrictRedis 实例
		# 记录行为
		pipe.zadd(key, now_ts, now_ts) # value 和 score 都使用毫秒时间戳
		# 移