django中实现一个高性能计数器(counter)实例
2016-06-22来源:

计数器(counter)是一个非常常用的功能组件,这篇blog以未读消息数为例,介绍了在 django中实现一个高性能计数器的基本要点。

故事的开始:.count()

假设你有一个notification model类,保存的主要是所有的站内通知:

代码如下:

class notification(models.model):

    一个简化过的notification类,拥有三个字段:

    - `user_id`: 消息所有人的用户id

    - `has_readed`: 表示消息是否已读

    user_id = models.integerfield(db_index=true)

    has_readed = models.booleanfield(default=false)

理所当然的,刚开始你会通过这样的查询来获取某个用户的未读消息数:

代码如下:

# 获取id为3074的用户的未读消息数

notification.objects.filter(user_id=3074, has_readed=false).count()

当你的notification表比较小的时候,这样的方式没有任何的问题,但是慢慢的,随着业务量 的扩大。消息表里面有了 上亿条数据 。很多懒惰的用户的未读消息数都到了上千条。

这时候,你就需要实现一个计数器,让这个计数器来统计每个用户的未读消息数,这样 比起之前的 count() ,我们只需要执行一条简单的主键查询(或者更优)就可以拿到实时的未读消息数了。

更优的方案:建立计数器

首先,让我们得建立一个新表来存储每个用户的未读消息数。

代码如下:

class usernotificationscount(models.model):

    这个model保存着每一个用户的未读消息数目

    user_id = models.integerfield(primary_key=true)

    unread_count = models.integerfield(default=0)

    def __str__(self):

        return '<usernotificationscount %s: %s>' % (self.user_id, self.unread_count)

我们为每一个注册用户提供了一条对应的 usernotificationscount 记录来保存他的未读消息数。 每次获取他的未读消息数的时候,只需要 usernotificationscount.objects.get(pk=user_id).unread_count 就可以了。

接下来,问题的重点来了,我们如何知道什么时候应该更新我们的计数器?django在这方面提供了什么捷径吗?

挑战:实时更新你的计数器

为了让我们的计数器正常的工作,我们必须实时的更新它,这包括:

1.当有新的未读消息过来的时候,为计数器 +1

2.当消息被异常删除时,如果关联的消息为未读,为计数器 -1

3.当阅读完一个新消息的时候,为计数器 -1

让我们一个一个来解决这些情况。

在抛出解决方案之前,我们需要先介绍django中的一个功能: signals ,signals是django提供的一个事件通知机制,它可以让你在监听某些自定义或者 预设的事件,当这些事件发生的时候,调用实现定义好的方法。

比如 django.db.models.signals.pre_save & django.db.models.signals.post_save 表示的是 某个model调用save方法之前和之后会触发的事件,它和database提供的触发器在功能上有一点相似。

2025公考·省考培训课程试听预约报名

  • 报班类型
  • 姓名
  • 手机号
  • 验证码
推荐信息
Baidu
map