[發明專利]一種流量控制方法及裝置、分布式集群

有效

• 申請號: 201811289371.0
• 申請日: 2018-10-31
• 公開（公告）號: CN109450806B
• 公開（公告）日: 2022-05-06
• 發明（設計）人: 張偉;王國忠;丁厚東;史永飛;李成;高旺
• 申請（專利權）人: 中國聯合網絡通信集團有限公司
• 主分類號: H04L67/1097
• 分類號: H04L67/1097;H04L67/1004;H04L47/10
• 代理公司: 北京中博世達專利商標代理有限公司 11274
• 代理人: 申健

本發明的實施例公開一種流量控制方法及裝置、分布式集群，涉及通信技術領域，能夠將分布式集群與ZooKeeper集群相結合，實現分布式環境下的多維度訪問流量控制。該方法包括：控制裝置獲取初始化信息；并初始化信息發送至ZooKeeper集群；控制裝置將第一結點設備在預定周期內的流量按照指定維度進行累加，獲取累加流量，并通過第一結點設備將累加流量發送至ZooKeeper集群；控制裝置根據至少一個預定周期獲取至少一個維度的第二結點設備的流量；并將至少一個維度的第二結點設備的流量進行累加生成第一結點設備的總流量，將總流量與預設閾值進行對比，確定是否對第一結點設備進行限流操作。本發明實施例應用于通信系統。

技術領域

本發明的實施例涉及通信技術領域，尤其涉及一種流量控制方法及裝置、分布式集群。

背景技術

針對互聯網分布式的并發系統，為防止瞬時高并發場景超過系統的承載能力，保障系統的穩定運營，需要建立訪問流量控制功能，根據不同的維度設置相應的流量閾值，來達到限制訪問的功能。現有的流量控制方案采用的計數器法主要用來限制一定時間內的總并發數，比如數據庫連接池、線程池、秒殺的并發數；計數器限流只要一定時間內的總請求數超過設定的閥值則進行限流。漏桶算法的核心思想是假設一個固定容量的常量速率流出請求，流入請求速率任意，當流入的請求數累積到漏桶容量時，則新流入的請求被拒絕。漏桶可以看作是一個具有固定容量、固定流出速率的隊列，漏桶算法限制的是請求的流出速率。令牌桶法是一個存放固定容量令牌的桶，按照固定速率往桶里添加令牌，填滿了就丟棄令牌，請求是否被處理要看桶中令牌是否足夠，當令牌數減為零時則拒絕新的請求。令牌桶算法允許一定程度突發流量，只要有令牌就可以處理，支持一次拿多個令牌。但是，由于漏桶算法使用的是單點限流，速率固定，無法應對突發流量請求；令牌桶算法容易造成熱點，而計數器法一般用于單結點限流，難以用于分布式集群。

發明內容

本發明的實施例提供一種流量控制方法及裝置、分布式集群，能夠將分布式集群與ZooKeeper集群相結合，實現分布式環境下的多維度訪問流量控制。

為達到上述目的，本發明采用如下技術方案：

第一方面，提供一種流量控制方法，應用于分布式集群，分布式集群包括：至少一個第一結點設備，其中第一結點設備連接ZooKeeper集群中的至少一個維度的第二結點設備；其中ZooKeeper集群包含至少一個維度的第二結點設備，每個維度包含至少一個第二結點設備；分布式集群還包括控制裝置，該方法包括：控制裝置獲取初始化信息；其中，初始化信息包括維度以及預定周期；控制裝置將初始化信息發送至ZooKeeper集群，以便ZooKeeper集群根據初始化信息進行初始化；控制裝置將第一結點設備在預定周期內的流量按照指定維度進行累加，獲取累加流量，并通過第一結點設備將累加流量發送至ZooKeeper集群，以便ZooKeeper集群將累加流量存儲至指定維度的第二結點設備中；其中，在預定周期結束后第一結點設備對累加流量清零；控制裝置根據至少一個預定周期獲取至少一個維度的第二結點設備的流量；控制裝置將至少一個維度的第二結點設備的流量進行累加生成第一結點設備的總流量，將總流量與預設閾值進行對比，確定是否對第一結點設備進行限流操作。

在上述方法中，主要應用于分布式集群。該方法的主要實現方式為：控制裝置首先獲取初始化信息；其中，初始化信息包括維度以及預定周期；并將初始化信息發送至ZooKeeper集群，以便ZooKeeper集群根據初始化信息進行初始化；然后將第一結點設備在預定周期內的流量按照指定維度進行累加，獲取累加流量，并通過第一結點設備將累加流量發送至ZooKeeper集群，以便ZooKeeper集群將累加流量存儲至指定維度的第二結點設備中；另外，在預定周期結束后第一結點設備對累加流量清零；最后，根據至少一個預定周期獲取至少一個維度的第二結點設備的流量；并對其進行累加生成第一結點設備的總流量，將總流量與預設閾值進行對比，確定是否對第一結點設備進行限流操作。本發明的實施例能夠將分布式集群與ZooKeeper集群相結合，實現分布式環境下的多維度訪問流量控制。