緩沖區(qū)分配器

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

抽象類AbstractByteBufAllocator實(shí)現(xiàn)了ByteBufAllocator接口, 并且實(shí)現(xiàn)了其大部分功能

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

以其中的分配ByteBuf的方法為例, 對(duì)其做簡(jiǎn)單的介紹

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

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

跟到directBuffer()方法中

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

這里又調(diào)用了一個(gè)重載directBuffer方法, 其中DEFAULT_INITIAL_CAPACITY代表分配的默認(rèn)容量, Integer.MAX_VALUE表示分配的ByteBuf可擴(kuò)容的最大容量, 也就是Integer類型的最大值, 我們?cè)俑M(jìn)去:

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

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

然后通過(guò)validate方法進(jìn)行參數(shù)驗(yàn)證

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

在AbstractByteBufAllocator中, newDirectBuffer是一個(gè)抽象方法, 由其子類實(shí)現(xiàn)

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

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

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

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

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

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

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

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

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

然后通過(guò)validate方法進(jìn)行參數(shù)驗(yàn)證

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

同樣, newHeapBuffer方法在AbstractByteBufAllocator中也是一個(gè)抽象方法, 具體邏輯交給其子類實(shí)現(xiàn)

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

newDirectBuffer和newHeapBuffer兩個(gè)抽象方法中, 在其子類PooledByteBufAllocator和UnpooledByteBufAllocator中都有實(shí)現(xiàn)

我們以UnpooledByteBufAllocator的newHeapBuffer方法為例, 看其實(shí)現(xiàn):

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

里實(shí)現(xiàn)方式其實(shí)很簡(jiǎn)單, 首先通過(guò)PlatformDependent.hasUnsafe()判斷當(dāng)前運(yùn)行環(huán)境是否能創(chuàng)建unsafe對(duì)象, 如果能, 則直接通過(guò)new UnpooledUnsafeHeapByteBuf(this, initialCapacity, maxCapacity)方式創(chuàng)建一個(gè)UnpooledUnsafeHeapByteBuf對(duì)象, 也就是一個(gè)Unsafe的ByteBuf對(duì)象

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

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

有關(guān)ByteBufAllocator的繼承關(guān)系如下:

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