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

故事的開始：.count()

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

代碼如下:

class notification(models.model):

    一個簡化過的notification類，擁有三個字段：

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

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

    user_id = models.integerfield(db_index=true)

    has_readed = models.booleanfield(default=false)

理所當(dāng)然的，剛開始你會通過這樣的查詢來獲取某個用戶的未讀消息數(shù)：

代碼如下:

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

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

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

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

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

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

代碼如下:

class usernotificationscount(models.model):

    這個model保存著每一個用戶的未讀消息數(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)槊恳粋€注冊用戶提供了一條對應(yīng)的 usernotificationscount 記錄來保存他的未讀消息數(shù)。 每次獲取他的未讀消息數(shù)的時候，只需要 usernotificationscount.objects.get(pk=user_id).unread_count 就可以了。

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

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

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

1.當(dāng)有新的未讀消息過來的時候，為計數(shù)器 +1

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

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

讓我們一個一個來解決這些情況。

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

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