我們想要更好地向使用者展示 Reddit 的規模。為了這一點,投票和評論數是一個帖子最重要的指標。然而,在 Reddit 上有相當多的使用者只瀏覽內容,既不投票也不評論。所以我們想要建立一個能夠計算一個帖子瀏覽數的系統。這一數字會被展示給帖子的創作者和版主,以便他們更好的瞭解某個帖子的活躍程度。
在這篇部落格中,我們將討論我們是如何實現超大資料量的計數。
計數機制
對於計數系統我們主要有四種需求:
- 帖子瀏覽數必須是實時或者近實時的,而不是每天或者每小時彙總。
- 同一使用者在短時間內多次訪問帖子,只算一個瀏覽量
- 顯示的瀏覽量與真實瀏覽量間允許有小百分之幾的誤差
- Reddit 是全球訪問量第八的網站,系統要能在生產環境的規模上正常執行,僅允許幾秒的延遲
要全部滿足以上四個需求的困難遠遠比聽上去大的多。為了實時精準計數,我們需要知道某個使用者是否曾經訪問過這篇帖子。想要知道這個資訊,我們就要為每篇帖子維護一個訪問使用者的集合,然後在每次計算瀏覽量時檢查集合。一個 naive 的實現方式就是將訪問使用者的集合儲存在記憶體的 hashMap 中,以帖子 Id 為 key。
這種實現方式對於訪問量低的帖子是可行的,但一旦一個帖子變得流行,訪問量劇增時就很難控制了。甚至有的帖子有超過 100 萬的獨立訪客! 對於這樣的帖子,儲存獨立訪客的 ID 並且頻繁查詢某個使用者是否之前曾訪問過會給記憶體和 CPU 造成很大的負擔。
因為我們不能提供準確的計數,我們檢視了幾種不同的基數估計演算法。有兩個符合我們需求的選擇:
- 一是線性概率計數法,很準確,但當計數集合變大時所需記憶體會線性變大。
- 二是基於 HyperLogLog (以下簡稱 HLL )的計數法。 HLL 空間複雜度較低,但是精確度不如線性計數。
下面看下 HLL 會節省多少記憶體。如果我們需要儲存 100 萬個獨立訪客的 ID, 每個使用者 ID 8 位元組長,那麼為了儲存一篇帖子的獨立訪客我們就需要 8 M的記憶體。反之,如果採用 HLL 會顯著減少記憶體佔用。不同的 HLL 實現方式消耗的記憶體不同。如果採用這篇文章的實現方法,那麼儲存 100 萬個 ID 僅需 12 KB,是原來的 0.15%!!
Big Data Counting: How to count a billion distinct objects using only 1.5KB of Memory – High Scalability –這篇文章很好的總結了上面的演算法。
許多 HLL 的實現都是結合了上面兩種演算法。在集合小的時候採用線性計數,當集合大小到達一定的閾值後切換到 HLL。前者通常被成為 ”稀疏“(sparse) HLL,後者被稱為”稠密“(dense) HLL。這種結合了兩種演算法的實現有很大的好處,因為它對於小集合和大集合都能夠保證精確度,同時保證了適度的記憶體增長。可以在 google 的這篇論文中瞭解這種實現的詳細內容。
現在我們已經確定要採用 HLL 演算法了,不過在選擇具體的實現時,我們考慮了以下三種不同的實現。因為我們的資料工程團隊使用 Java 和 Scala,所以我們只考慮 Java 和 Scala 的實現。
- Twitter 提供的 Algebird,採用 Scala 實現。Algebird 有很好的文件,但他們對於 sparse 和 dense HLL 的實現細節不是很容易理解。
- stream-lib中提供的 HyperLogLog++, 採用 Java 實現。stream-lib 中的程式碼文件齊全,但有些難理解如何合適的使用並且改造的符合我們的需求。
- Redis HLL 實現,這是我們最終選擇的。我們認為 Redis 中 HLLs 的實現文件齊全、容易配置,提供的相關 API 也很容易整合。還有一個好處是,我們可以用一臺專門的伺服器部署,從而減輕效能上的壓力。
Reddit 的資料管道依賴於 Kafka。當一個使用者訪問了一篇部落格,會觸發一個事件,事件會被髮送到事件收集伺服器,並被持久化在 Kafka 中。
之後,計數系統會依次順序執行兩個元件。在我們的計數系統架構中,第一部分是一個 Kafka 的消費者,我們稱之為 Nazar。Nazar 會從 Kafka 中讀取每個事件,並將它通過一系列配置的規則來判斷該事件是否需要被計數。我們取這個名字僅僅是因為 Nazar 是一個眼睛形狀的護身符,而 ”Nazar“ 系統就像眼睛一樣使我們的計數系統遠離不懷好意者的破壞。其中一個我們不將一個事件計算在內的原因就是同一個使用者在很短時間內重複訪問。Nazar 會修改事件,加上個標明是否應該被計數的布林標識,並將事件重新放入 Kafka。
下面就到了系統的第二個部分。我們將第二個 Kafka 的消費者稱作 Abacus,用來進行真正瀏覽量的計算,並且將計算結果顯示在網站或客戶端。Abacus 從 Kafka 中讀取經過 Nazar 處理過的事件,並根據 Nazar 的處理結果決定是跳過這個事件還是將其加入計數。如果 Nazar 中的處理結果是可以加入計數,那麼 Abacus 首先會檢查這個事件所關聯的帖子在 Redis 中是否已經存在了一個 HLL 計數器。如果已經存在,Abacus 會給 Redis 傳送個 PFADD 的請求。如果不存在,那麼 Abacus 會給 Cassandra 叢集傳送個請求(Cassandra 用來持久化 HLL 計數器和 計數值的),然後向 Redis 傳送 SET 請求。這通常會發生在網友訪問較老帖子的時候,這時該帖子的計數器很可能已經在 Redis 中過期了。
為了儲存存在 Redis 中的計數器過期的老帖子的瀏覽量。Abacus 會週期性的將 Redis 中全部的 HLL 和 每篇帖子的瀏覽量寫入到 Cassandra 叢集中。為了避免叢集過載,我們以 10 秒為週期批量寫入。
下圖是事件流的大致流程:
總結
我們希望瀏覽量可以讓發帖者瞭解帖子全部的訪問量,也幫助版主快速定位自己社群中高訪問量的帖子。在未來,我們計劃利用我們資料管道在實時方面的潛力來為 Reddit 的使用者提供更多的有用的反饋。