緩沖區(qū)分配器

顧明思議就是分配緩沖區(qū)的工具, 在netty中, 緩沖區(qū)分配器的頂級抽象是接口ByteBufAllocator, 里面定義了有關緩沖區(qū)分配的相關api

抽象類AbstractByteBufAllocator實現了ByteBufAllocator接口, 并且實現了其大部分功能

和AbstractByteBuf一樣, AbstractByteBufAllocator也實現了緩沖區(qū)分配的骨架邏輯, 剩余的交給其子類

以其中的分配ByteBuf的方法為例, 對其做簡單的介紹

public ByteBuf buffer() {
    if (directByDefault) {
        return directBuffer();
    }
    return heapBuffer();
}

這里if (directByDefault)會判斷默認創(chuàng)建的ByteBuf是不是一個基于直接內存的ByteBuf, 也就是direct類型的ByteBuf, 如果是, 則通過directBuffer()方法返回direct類型的ByteBuf, 否則, 會通過heapBuffer()返回heap類型的ByteBuf

跟到directBuffer()方法中

public ByteBuf directBuffer() {
    return directBuffer(DEFAULT_INITIAL_CAPACITY, Integer.MAX_VALUE);
}

這里又調用了一個重載directBuffer方法, 其中DEFAULT_INITIAL_CAPACITY代表分配的默認容量, Integer.MAX_VALUE表示分配的ByteBuf可擴容的最大容量, 也就是Integer類型的最大值, 我們再跟進去:

public ByteBuf directBuffer(int initialCapacity, int maxCapacity) {
    if (initialCapacity == 0 && maxCapacity == 0) {
        return emptyBuf;
    }
    validate(initialCapacity, maxCapacity);
    return newDirectBuffer(initialCapacity, maxCapacity);
}

這里判斷如果初始容量和最大容量都為0的話, 則返回一個emptyBuf的成員變量, emptyBuf代表一個空的ByteBuf

然后通過validate方法進行參數驗證

最后newDirectBuffer創(chuàng)建一個Direct類型的ByteBuf, 并將初始容量和最大容量傳入

在AbstractByteBufAllocator中, newDirectBuffer是一個抽象方法, 由其子類實現

protected abstract ByteBuf newDirectBuffer(int initialCapacity, int maxCapacity);

我們回到緩沖區(qū)分配的方法

public ByteBuf buffer() {
    if (directByDefault) {
        return directBuffer();
    }
    return heapBuffer();
}

剛才簡單剖析了directBuffer()的分配, 現在在繼續(xù)跟到heapBuffer()中, 看其分配heap類型的ByteBuf的抽象邏輯:

public ByteBuf heapBuffer() {
    return heapBuffer(DEFAULT_INITIAL_CAPACITY, Integer.MAX_VALUE);
}

這里同樣調用了重載的heapBuffer, 并傳入了初始容量和最大容量

再繼續(xù)跟heapBuffer方法:

public ByteBuf heapBuffer(int initialCapacity, int maxCapacity) {
    if (initialCapacity == 0 && maxCapacity == 0) {
        return emptyBuf;
    }
    validate(initialCapacity, maxCapacity);
    return newHeapBuffer(initialCapacity, maxCapacity);
}

同樣, 這里如果初始容量和最大容量都為空的話, 返回一個代表空的ByteBuf

然后通過validate方法進行參數驗證

最后通過newHeapBuffer方法創(chuàng)建一個新的heap類型的ByteBuf

同樣, newHeapBuffer方法在AbstractByteBufAllocator中也是一個抽象方法, 具體邏輯交給其子類實現

protected abstract ByteBuf newHeapBuffer(int initialCapacity, int maxCapacity);

newDirectBuffer和newHeapBuffer兩個抽象方法中, 在其子類PooledByteBufAllocator和UnpooledByteBufAllocator中都有實現

我們以UnpooledByteBufAllocator的newHeapBuffer方法為例, 看其實現:

protected ByteBuf newHeapBuffer(int initialCapacity, int maxCapacity) {
    return PlatformDependent.hasUnsafe() ? new UnpooledUnsafeHeapByteBuf(this, initialCapacity, maxCapacity)
            : new UnpooledHeapByteBuf(this, initialCapacity, maxCapacity);
}

里實現方式其實很簡單, 首先通過PlatformDependent.hasUnsafe()判斷當前運行環(huán)境是否能創(chuàng)建unsafe對象, 如果能, 則直接通過new UnpooledUnsafeHeapByteBuf(this, initialCapacity, maxCapacity)方式創(chuàng)建一個UnpooledUnsafeHeapByteBuf對象, 也就是一個Unsafe的ByteBuf對象

如果當前環(huán)境不能創(chuàng)建unsafe對象, 則通過new UnpooledHeapByteBuf(this, initialCapacity, maxCapacity)這種方方式創(chuàng)建一個UnpooledHeapByteBuf對象, 也就是非Unsafe的ByteBuf對象

從這里能看出, 其實在創(chuàng)建ByteBuf對象時, 是否創(chuàng)建unsafe類型的對象并不是我們自己控制的, 而是通過程序判斷當前環(huán)境來決定是否創(chuàng)建unsafe類型的ByteBuf對象的

有關ByteBufAllocator的繼承關系如下:

以上就是Netty分布式ByteBuf緩沖區(qū)分配器源碼解析的詳細內容，更多關于Netty分布式ByteBuf緩沖區(qū)分配器的資料請關注腳本之家其它相關文章！

最新評論