計數器(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提供的觸發器在功能上有一點相似。