Redis 多方式实现计数器功能（附代码）

本文由 yanglbme 原创，首发于掘金，禁止未受权转载。

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

请跟随我一块儿来看看吧。

使用字符串键

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

from redis import Redis

class Counter:
    def __init__(self, client: Redis, key: str):
        self.client = client
        self.key = key

    def incr(self, amount=1):
        """计数累加"""
        self.client.incr(self.key, amount=amount)

    def decr(self, amount=1):
        """计数累减"""
        self.client.decr(self.key, amount=amount)

    def get_cnt(self):
        """获取当前计数的值"""
        return self.client.get(self.key)


if __name__ == '__main__':
    client = Redis(decode_responses=True)
    counter = Counter(client, 'page_view:12')
    counter.incr()
    counter.incr()
    print(counter.get_cnt())  # 2

复制代码

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

使用哈希键

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

from redis import Redis

class Counter:
    def __init__(self, client: Redis, key: str, counter: str):
        self.client = client
        self.key = key
        self.counter = counter

    def incr(self, amount=1):
        """计数累加"""
        self.client.hincrby(self.key, self.counter, amount=amount)

    def decr(self, amount=1):
        """计数累减"""
        self.client.hincrby(self.key, self.counter, amount=-amount)

    def get_cnt(self):
        """获取当前计数的值"""
        return self.client.hget(self.key, self.counter)


if __name__ == '__main__':
    client = Redis(decode_responses=True)
    counter = Counter(client, 'page_view', '66')
    counter.incr()
    counter.incr()
    print(counter.get_cnt())  # 2

复制代码

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

使用集合键

在上面两个例子中，当动做被执行时，程序能够调用一次 incr() 累加计数的方法。某些场景下，咱们可能须要对特定的动做，仅仅计数一次。什么叫“仅仅计数一次”？就是说，同一个用户/IP，屡次访问某个页面，计数器只会将计数值增长 1。来看看如下代码：

from redis import Redis

class Counter:
    def __init__(self, client: Redis, key: str):
        self.client = client
        self.key = key

    def add(self, item: str) -> bool:
        """计数累加，若计数以前item已存在，放回False；不然返回True"""
        return self.client.sadd(self.key, item) == 1

    def get_cnt(self):
        """获取当前计数的值"""
        return self.client.scard(self.key)


if __name__ == '__main__':
    client = Redis(decode_responses=True)
    counter = Counter(client, 'uv')
    counter.add('user1')
    counter.add('user2')
    counter.add('user1')  # 重复放入
    print(counter.get_cnt())  # 2

复制代码

在实际应用中，以上代码须要稍做改动，但基本的思路不变。怎么样，你学会了吗？

欢迎扫码关注个人公众号“Doocs开源社区”。