1. 程式人生 > 實用技巧 >Sentinel滑動視窗演算法

Sentinel滑動視窗演算法

在前面搞清楚了Sentinel的使用後,大致理了一下Sentinel的責任鏈,搞清楚了這個,基本就已經梳理清楚sentinel-core模組的大部分內容,順著這條鏈路可以繼續梳理很多東西。

知其然、知其所以然。而閱讀原始碼就是最好的知其所以然的方式。這一次找了一些空閒時間,捋了一下它的滑動視窗演算法,在這裡做一個記錄。後面會繼續去梳理它的令牌演算法和漏桶演算法。


關於滑動視窗的原理,Sentinel為什麼要使用滑動視窗,Sentinel是怎樣使用的滑動,直接使用下面這兩張圖。一圖勝千言,一張好的圖足以說明問題,在這裡我引用兩張圖。


首先從StatisticSlot類開始,它是Sentinel統計的核心功能槽,先看它的entry[^對這個方法做了一下精簡,只保留了幾行能夠說明問題的程式碼。]方法:

@SpiOrder(-7000)
public class StatisticSlot extends AbstractLinkedProcessorSlot<DefaultNode> {

    @Override
    public void entry(Context context, ResourceWrapper resourceWrapper, DefaultNode node, int count,
                      boolean prioritized, Object... args) throws Throwable {
        try {
            // 先執行後續限流、降級等功能
            fireEntry(context, resourceWrapper, node, count, prioritized, args);

            // 上面執行通過,更新通過請求資料
            node.addPassRequest(count);

        } catch (PriorityWaitException ex) {

        } catch (BlockException e) {
            // 上面執行阻塞,更新阻塞請求資料
            node.increaseBlockQps(count);
        } catch (Throwable e) {
        }
    }
}

通過程式碼可以看出,它是先執行後面的限流、降級等,然後以後面的執行結果為基礎來更新對應資源的通過、阻塞、異常等統計資料。上面的執行通過和異常處理邏輯大體一致。這裡就以執行通過這條線來說明問題,所以對應程式碼就是node.addPassRequest(count);進入到這一行程式碼,經過幾次呼叫轉到了StatisticNode這個類上,根據類名可以知道這個表示一個統計節點,呼叫的方法是addPassRequest:

@Override
public void addPassRequest(int count) {
    rollingCounterInSecond.addPass(count);
    rollingCounterInMinute.addPass(count);
}

由這個方法可以看出,StatisticNode在處理統計資料的時候,分了兩個維度,分別是秒級的和分鐘級的。對應的rollingCounterInSecond和rollingCounterInMinute是它的兩個成員屬性。其定義如下:

public class StatisticNode implements Node {

    /**
     * Holds statistics of the recent {@code INTERVAL} seconds. The {@code INTERVAL} is divided into time spans
     * by given {@code sampleCount}.
     */
    private transient volatile Metric rollingCounterInSecond = new ArrayMetric(SampleCountProperty.SAMPLE_COUNT,
        IntervalProperty.INTERVAL);

    /**
     * Holds statistics of the recent 60 seconds. The windowLengthInMs is deliberately set to 1000 milliseconds,
     * meaning each bucket per second, in this way we can get accurate statistics of each second.
     */
    private transient Metric rollingCounterInMinute = new ArrayMetric(60, 60 * 1000, false);
}

Metric是一個度量單位介面,其具體實現下面需要提一下,在這裡先不展開,只需要知道它儲存的是一些執行資料,如成功數、異常數等。而在上面的StatisticNode.addPassRequest方法中就是分別呼叫兩個維度的統計單位增加請求通過數量。從rollingCounterInSecond.addPass(count)這一句進入,對應的方法在ArrayMetric類中,這一個就是Metric的唯一實現。ArrayMetric.addPass這個方法程式碼如下:

@Override
public void addPass(int count) {
    // 獲取當前時間對應的視窗,返回的是當前視窗的一個包裝類
    WindowWrap<MetricBucket> wrap = data.currentWindow();
    wrap.value().addPass(count);
}

可以看出這裡就已經是滑動視窗演算法的入口了。通過滑動視窗演算法,使用當前時間獲取一個合適的視窗,然後在這個視窗中增加通過的請求數。進入到程式碼裡面,最終落實到了LeapArray的currentWindow方法中了。就LeapArray這個類名來說,非常有我在開篇第二張圖的那味道了。

在看LeapArray.currentWindow這個方法之前,先來看一個短小簡單但是足夠核心的一個方法LeapArray.calculateTimeIdx,整個方法只有兩行程式碼,如下:

private int calculateTimeIdx(long timeMillis) {
    // 將傳入的當前時間按照視窗時長進行分段,拿到當前時間對應的分段ID
    long timeId = timeMillis / windowLengthInMs;
    // 將當前時間的分段段ID對應到視窗陣列的下標ID上
    return (int)(timeId % array.length());
}

上面的array定義如下:

protected final AtomicReferenceArray<WindowWrap<T>> array;

其賦值在構造方法中,如下語句:

this.array = new AtomicReferenceArray<>(sampleCount);

通過上面這個方法,我們就能夠得到當前時間對應的視窗在視窗陣列中的位置了,接下來我們要做的事情就是根據這個位置取出對應的視窗返回去給對應的統計邏輯使用。

直接看LeapArray.currentWindow[^為了方便閱讀,精簡了它的註釋]方法定義:

public WindowWrap<T> currentWindow(long timeMillis) {
    if (timeMillis < 0) {
        return null;
    }
	// 獲取當前時間在視窗陣列中對映的下標
    int idx = calculateTimeIdx(timeMillis);
	// 計算當前時間對應的視窗的開始時間,具體方法見下面
    long windowStart = calculateWindowStart(timeMillis);

    /*
         * Get bucket item at given time from the array.
         *
         * (1) Bucket is absent, then just create a new bucket and CAS update to circular array.
         * (2) Bucket is up-to-date, then just return the bucket.
         * (3) Bucket is deprecated, then reset current bucket and clean all deprecated buckets.
         */
    while (true) {
        WindowWrap<T> old = array.get(idx);
        if (old == null) {
            // 第一次進入,新建視窗,並使用cas的方式設定,如果出現爭搶導致設定失敗,暫時讓出執行權待其它執行緒成功設定
            WindowWrap<T> window = new WindowWrap<T>(windowLengthInMs, windowStart, newEmptyBucket(timeMillis));
            if (array.compareAndSet(idx, null, window)) {
                return window;
            } else {
                Thread.yield();
            }
        } else if (windowStart == old.windowStart()) {
            // 當前時間對應的視窗開始時間等於獲取到的視窗開始時間,那麼當前獲取到的視窗就是我們需要的
            return old;
        } else if (windowStart > old.windowStart()) {
            // 當前時間對應的視窗開始時間大於獲取到的視窗開始時間,那麼當前獲取到的視窗為已過期視窗,加鎖重置
            if (updateLock.tryLock()) {
                try {
                    return resetWindowTo(old, windowStart);
                } finally {
                    updateLock.unlock();
                }
            } else {
                Thread.yield();
            }
        } else if (windowStart < old.windowStart()) {
            // Should not go through here, as the provided time is already behind.
            return new WindowWrap<T>(windowLengthInMs, windowStart, newEmptyBucket(timeMillis));
        }
    }
}

LeapArray.calculateWindowStart方法:

protected long calculateWindowStart(long timeMillis) {
    return timeMillis - timeMillis % windowLengthInMs;
}

總結上面的程式碼就是:先將當前時間按照統計時長分段,得到當前時間對應的分段ID。因為視窗陣列是固定的,所以隨著時間線向前發展,會不斷的順序迴圈使用陣列中的視窗。所以使用當前時間對應的分段ID與視窗陣列的長度求餘得到當前時間對應的視窗在視窗陣列中的下標,拿到這個下標後,接著就是在迴圈中獲取這個下標對應的視窗了。

在獲取指定下標對應的視窗時,要分情況進行處理:

  • 如果對應下標視窗為null,那麼就是第一次進入,建立新視窗並使用cas設定。如果非空走下面的邏輯。
  • 如果獲取到的視窗開始時間等於當前時間計算出來的對應視窗開始時間,那麼就拿到了當前時間需要的視窗,直接返回。
  • 如果獲取到的視窗開始時間小於當前時間計算出來的對應視窗開始時間,那麼就說明這個視窗已經過期了,所以加鎖重置,然後重複使用。
  • 當前時間小於舊的視窗的開始時間,理論上來說是不應該出現這種情況的,如果存在這種情況,那麼返回一個無效的空視窗。

整個Sentinel滑動視窗演算法的使用就上面這些程式碼,看完後第一感覺是程式碼如此簡介,但是功能卻如此高效強大。