|
|
51CTO旗下网站
|
|
移动端

Redis 多方式实现计数器功能

计数器在很多网站中都进行了广泛的应用,比如文章的点赞数、页面的浏览数、网站的访客数、视频的播放数等等。在这篇文章里,我会使用 Redis 的三种数据类型,来分别实现计数器的功能。

作者:yanglbme来源:掘金|2019-10-16 15:13

【线上直播】11月21日晚8点贝壳技术总监侯圣文《数据安全之数据库安全黄金法则》

计数器在很多网站中都进行了广泛的应用,比如文章的点赞数、页面的浏览数、网站的访客数、视频的播放数等等。在这篇文章里,我会使用 Redis 的三种数据类型,来分别实现计数器的功能。

请跟随我一起来看看吧。

使用字符串键

下面代码演示了如何利用 Redis 中的字符串键来实现计数器功能。其中,incr() 方法用于累加计数,get_cnt() 方法用于获取当前的计数值。

  1. from redis import Redis 
  2.  
  3. class Counter: 
  4.     def __init__(self, client: Redis, key: str): 
  5.         self.client = client 
  6.         self.key = key 
  7.  
  8.     def incr(self, amount=1): 
  9.         """计数累加""" 
  10.         self.client.incr(self.key, amount=amount) 
  11.  
  12.     def decr(self, amount=1): 
  13.         """计数累减""" 
  14.         self.client.decr(self.key, amount=amount) 
  15.  
  16.     def get_cnt(self): 
  17.         """获取当前计数的值""" 
  18.         return self.client.get(self.key
  19.  
  20.  
  21. if __name__ == '__main__'
  22.     client = Redis(decode_responses=True
  23.     counter = Counter(client, 'page_view:12'
  24.     counter.incr() 
  25.     counter.incr() 
  26.     print(counter.get_cnt())  # 2 

假设我们要统计 page_id 为 12 的页面的浏览数,那么我们可以设定 key 为 page_view:12,用户每一次浏览,就调用一次 counter 的 incr() 方法进行计数。

使用哈希键

在上面的代码中,我们需要针对每个统计项,都单独设置一个字符串键。那么,下面我们来看看如何通过 Redis 的哈希键,来对关联的统计项进行统一管理。

  1. from redis import Redis 
  2.  
  3. class Counter: 
  4.     def __init__(self, client: Redis, key: str, counter: str): 
  5.         self.client = client 
  6.         self.key = key 
  7.         self.counter = counter 
  8.  
  9.     def incr(self, amount=1): 
  10.         """计数累加""" 
  11.         self.client.hincrby(self.key, self.counter, amount=amount) 
  12.  
  13.     def decr(self, amount=1): 
  14.         """计数累减""" 
  15.         self.client.hincrby(self.key, self.counter, amount=-amount) 
  16.  
  17.     def get_cnt(self): 
  18.         """获取当前计数的值""" 
  19.         return self.client.hget(self.key, self.counter) 
  20.  
  21.  
  22. if __name__ == '__main__'
  23.     client = Redis(decode_responses=True
  24.     counter = Counter(client, 'page_view''66'
  25.     counter.incr() 
  26.     counter.incr() 
  27.     print(counter.get_cnt())  # 2 

如果采用哈希键,那么,我们对于同一类型的计数,可以使用一个相同的 key 来进行存储。比如,在上面例子中,我们使用 page_view 来统计页面的浏览数,对于 page_id 为 66 的页面,直接添加到 page_view 对应的字段中即可。

使用集合键

在上面两个例子中,当动作被执行时,程序可以调用一次 incr() 累加计数的方法。某些场景下,我们可能需要对特定的动作,仅仅计数一次。什么叫“仅仅计数一次”?就是说,同一个用户/IP,多次访问某个页面,计数器只会将计数值增加 1。来看看以下代码:

  1. from redis import Redis 
  2.  
  3. class Counter: 
  4.     def __init__(self, client: Redis, key: str): 
  5.         self.client = client 
  6.         self.key = key 
  7.  
  8.     def add(self, item: str) -> bool: 
  9.         """计数累加,若计数之前item已存在,放回False;否则返回True""" 
  10.         return self.client.sadd(self.key, item) == 1 
  11.  
  12.     def get_cnt(self): 
  13.         """获取当前计数的值""" 
  14.         return self.client.scard(self.key
  15.  
  16.  
  17. if __name__ == '__main__'
  18.     client = Redis(decode_responses=True
  19.     counter = Counter(client, 'uv'
  20.     counter.add('user1'
  21.     counter.add('user2'
  22.     counter.add('user1')  # 重复放入 
  23.     print(counter.get_cnt())  # 2 

在实际应用中,以上代码需要稍作改动,但基本的思路不变。怎么样,你学会了吗?

【编辑推荐】

  1. 什么?Redis的QPS是MySQL的100倍?
  2. 记一次找因Redis使用不当导致应用卡死Bug的过程
  3. Redis内存满了怎么办……
  4. Redis缓存击穿、缓存穿透、缓存雪崩
【责任编辑:华轩 TEL:(010)68476606】

点赞 0
分享:
大家都在看
猜你喜欢

订阅专栏+更多

CentOS 8 全新学习术

CentOS 8 全新学习术

CentOS 8 正式发布
共16章 | UbuntuServer

230人订阅学习

用Python玩转excel

用Python玩转excel

让重复操作傻瓜化
共3章 | DE8UG

219人订阅学习

AI入门级算法

AI入门级算法

算法常识
共22章 | 周萝卜123

184人订阅学习

读 书 +更多

Scrum敏捷项目管理

本书详细描述如何在复杂技术项目中使用Scrum,并结合真实的Scrum案例及专家洞识,在简明及高度概括的理论之上更侧重于实践,并不断强调Scru...

订阅51CTO邮刊

点击这里查看样刊

订阅51CTO邮刊

51CTO服务号

51CTO官微