計(jì)數(shù)器(counter)是一個(gè)非常常用的功能組件，這篇blog以未讀消息數(shù)為例，介紹了在 django中實(shí)現(xiàn)一個(gè)高性能計(jì)數(shù)器的基本要點(diǎn)。

故事的開(kāi)始：.count()

假設(shè)你有一個(gè)notification model類，保存的主要是所有的站內(nèi)通知：

代碼如下:

class notification(models.model):

    一個(gè)簡(jiǎn)化過(guò)的notification類，擁有三個(gè)字段：

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

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

    user_id = models.integerfield(db_index=true)

    has_readed = models.booleanfield(default=false)

理所當(dāng)然的，剛開(kāi)始你會(huì)通過(guò)這樣的查詢來(lái)獲取某個(gè)用戶的未讀消息數(shù)：

代碼如下:

# 獲取id為3074的用戶的未讀消息數(shù)

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

當(dāng)你的notification表比較小的時(shí)候，這樣的方式?jīng)]有任何的問(wèn)題，但是慢慢的，隨著業(yè)務(wù)量 的擴(kuò)大。消息表里面有了 上億條數(shù)據(jù) 。很多懶惰的用戶的未讀消息數(shù)都到了上千條。

這時(shí)候，你就需要實(shí)現(xiàn)一個(gè)計(jì)數(shù)器，讓這個(gè)計(jì)數(shù)器來(lái)統(tǒng)計(jì)每個(gè)用戶的未讀消息數(shù)，這樣 比起之前的 count() ，我們只需要執(zhí)行一條簡(jiǎn)單的主鍵查詢（或者更優(yōu)）就可以拿到實(shí)時(shí)的未讀消息數(shù)了。

更優(yōu)的方案：建立計(jì)數(shù)器

首先，讓我們得建立一個(gè)新表來(lái)存儲(chǔ)每個(gè)用戶的未讀消息數(shù)。

代碼如下:

class usernotificationscount(models.model):

    這個(gè)model保存著每一個(gè)用戶的未讀消息數(shù)目

    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)

我們?yōu)槊恳粋€(gè)注冊(cè)用戶提供了一條對(duì)應(yīng)的 usernotificationscount 記錄來(lái)保存他的未讀消息數(shù)。 每次獲取他的未讀消息數(shù)的時(shí)候，只需要 usernotificationscount.objects.get(pk=user_id).unread_count 就可以了。

接下來(lái)，問(wèn)題的重點(diǎn)來(lái)了，我們?nèi)绾沃朗裁磿r(shí)候應(yīng)該更新我們的計(jì)數(shù)器？django在這方面提供了什么捷徑嗎？

挑戰(zhàn)：實(shí)時(shí)更新你的計(jì)數(shù)器

為了讓我們的計(jì)數(shù)器正常的工作，我們必須實(shí)時(shí)的更新它，這包括：

1.當(dāng)有新的未讀消息過(guò)來(lái)的時(shí)候，為計(jì)數(shù)器 +1

2.當(dāng)消息被異常刪除時(shí)，如果關(guān)聯(lián)的消息為未讀，為計(jì)數(shù)器 -1

3.當(dāng)閱讀完一個(gè)新消息的時(shí)候，為計(jì)數(shù)器 -1

讓我們一個(gè)一個(gè)來(lái)解決這些情況。

在拋出解決方案之前，我們需要先介紹django中的一個(gè)功能： signals ，signals是django提供的一個(gè)事件通知機(jī)制，它可以讓你在監(jiān)聽(tīng)某些自定義或者 預(yù)設(shè)的事件，當(dāng)這些事件發(fā)生的時(shí)候，調(diào)用實(shí)現(xiàn)定義好的方法。

比如 django.db.models.signals.pre_save & django.db.models.signals.post_save 表示的是 某個(gè)model調(diào)用save方法之前和之后會(huì)觸發(fā)的事件，它和database提供的觸發(fā)器在功能上有一點(diǎn)相似。