利用Redis简单高效实现各种计数功能

Redis 是目前 NoSQL 领域的当红炸子鸡,它象一把瑞士军刀,小巧、锋利、实用,特别适合解决一些使用传统关系数据库难以解决的问题。打算写一系列 Redis 使用模式的文章,深入总结介绍 Redis 常见的使用模式,以供大家参考。

常见汇总计数器

汇总计数是系统常见功能,比如网站通常需要统计注册用户数,网站总浏览次数等等。 使用 Redis 提供的基本数据类型就能实现汇总计数器,通过 incr 命令实现增加操作。

比如注册用户数,基本操作命令如下:


# 获取注册用户数
get total_users
# 注册用户数增加一位
incr total_users

按时间汇总的计数器

通常计数还要按时间统计,比如注册用户数需要按日统计,处理方法比较简单,把日期带入计数器 key 就可以。

还是注册用户计数的例子,基本操作命令如下:


# 假定操作 2014-07-06 数据
# 获取注册用户数
get total_users:2014-07-06
# 2014-07-06 注册用户数增加一位
incr total_users:2014-07-06
# 设置 48 小时过期时间 172800 = 48 * 60 * 60
expire total_users:2014-07-06 172800

为计数器设置一个 48 小时的过期时间是为了节省计数器占用空间,毕竟 redis 是内存数据库,可以在过期前执行一个任务把计数器存入关系数据库。

速度控制

速度控制也是 Redis 一种常见的计数用途,比如有一个 API 服务,希望控制每一个 IP 每秒请求数不超过 10 次,可以用 IP 和 时间秒作为 key 设置一个计数器,实现控制,伪代码如下所示:


# 每秒最大请求数
MAX_REQUESTS_PER_SECOND = 10

# 检查 ip 请求限制
# @param ip
# @raise 超过限制,抛出 RuntimeError 异常

def check_request_limitation_for_ip(ip)
time_tick = Time.now.to_i
key = "#{ip}:#{time_tick}"
num = $redis.get(key).to_i
if num > MAX_REQUEST_PER_SECOND
raise 'too many requests'
else
$redis.incr(key)
$redis.expire(key, 10)
end
end

使用 Hash 数据类型维护大量计数器

有时候需要维护大量计数器,比如每一个论坛主题的查看数,比如每一个用户访问页面次数,因为论坛主题和用户基数可能很大,直接基于论坛主题或用户 ID 生成计数器的话,占用 Redis 资源还是相当可观的,这时可以用 Hash 数据类型压缩所需资源。

比如,对应论坛主题查看计数,可以由模式


key: topic::views
value: view count (integer)

转换为模式:


key: topic:views
value: hash
hash key:
hash value: view count (integer)

以上就是利用Redis简单高效实现各种计数功能的全部内容,感谢大家的阅读,更多内容请关注中国E盟技术频道网站。