程序人生

写优雅的程序，做优雅的人

Redis 使用模式之一：计数器

| Comments

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

常见汇总计数器

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

比如注册用户数，基本操作命令如下：

1
2
3
4

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

按时间汇总的计数器

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

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

1
2
3
4
5
6
7

  # 假定操作 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 设置一个计数器，实现控制，伪代码如下所示：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

  # 每秒最大请求数
  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 数据类型压缩所需资源。

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

1
2

  key: topic:<topic_id>:views
  value: view count (integer)

转换为模式：

1
2
3
4

  key: topic:views
  value: hash
    hash key: <topic_id>
    hash value: view count (integer)

总结：利用 Redis 实现计数器，可以简单高效实现各种计数功能。

Comments