Netty源码分析第5章(ByteBuf)—->第3节: 缓冲区分配器

Netty源码分析第五章: ByteBuf

第三节: 缓冲区分配器

缓冲区分配器
, 顾明思议就是分配缓冲区的工具
, 在
netty中
, 缓冲区分配器的顶级抽象是接口
ByteBufAllocator, 里面定义了有关缓冲区分配的相关
api

抽象类
AbstractByteBufAllocator实现了
ByteBufAllocator接口
, 并且实现了其大部分功能

和
AbstractByteBuf一样
, AbstractByteBufAllocator也实现了缓冲区分配的骨架逻辑
, 剩余的交给其子类

以其中的分配
ByteBuf的方法为例
, 对其做简单的介绍
:

1
2
3
4
5
6
7
1public ByteBuf buffer() {

2    if (directByDefault) {

3        return directBuffer();

4    }

5    return heapBuffer();

6}

7

这里
if (directByDefault)会判断默认创建的
ByteBuf是不是一个基于直接内存的
ByteBuf, 也就是
direct类型的
ByteBuf, 如果是
, 则通过
directBuffer()方法返回
direct类型的
ByteBuf, 否则
, 会通过
heapBuffer()返回
heap类型的
ByteBuf

跟到
directBuffer()方法中
:

1
2
3
4
1public ByteBuf directBuffer() {

2    return directBuffer(DEFAULT_INITIAL_CAPACITY, Integer.MAX_VALUE);

3}

4

这里又调用了一个重载
directBuffer方法
, 其中
DEFAULT_INITIAL_CAPACITY代表分配的默认容量
, Integer.MAX_VALUE表示分配的
ByteBuf可扩容的最大容量
, 也就是
Integer类型的最大值
, 我们再跟进去
:

1
2
3
4
5
6
7
8
1public ByteBuf directBuffer(int initialCapacity, int maxCapacity) {

2    if (initialCapacity == 0 && maxCapacity == 0) {

3        return emptyBuf;

4    }

5    validate(initialCapacity, maxCapacity);

6    return newDirectBuffer(initialCapacity, maxCapacity);

7}

8

这里判断如果初始容量和最大容量都为
0的话
, 则返回一个
emptyBuf的成员变量
, emptyBuf代表一个空的
ByteBuf

然后通过
validate方法进行参数验证

最后
newDirectBuffer创建一个
Direct类型的
ByteBuf, 并将初始容量和最大容量传入

在
AbstractByteBufAllocator中
, newDirectBuffer是一个抽象方法
, 由其子类实现

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

2

我们回到缓冲区分配的方法
:

1
2
3
4
5
6
7
1public ByteBuf buffer() {

2    if (directByDefault) {

3        return directBuffer();

4    }

5    return heapBuffer();

6}

7

刚才简单剖析了
directBuffer()的分配
, 现在在继续跟到
heapBuffer()中
, 看其分配
heap类型的
ByteBuf的抽象逻辑
:

1
2
3
4
1public ByteBuf heapBuffer() {

2    return heapBuffer(DEFAULT_INITIAL_CAPACITY, Integer.MAX_VALUE);

3}

4

这里同样调用了重载的
heapBuffer, 并传入了初始容量和最大容量

再继续跟
heapBuffer方法
:

1
2
3
4
5
6
7
8
1public ByteBuf heapBuffer(int initialCapacity, int maxCapacity) {

2    if (initialCapacity == 0 && maxCapacity == 0) {

3        return emptyBuf;

4    }

5    validate(initialCapacity, maxCapacity);

6    return newHeapBuffer(initialCapacity, maxCapacity);

7}

8

同样
, 这里如果初始容量和最大容量都为空的话
, 返回一个代表空的
ByteBuf

然后通过
validate方法进行参数验证

最后通过
newHeapBuffer方法创建一个新的
heap类型的
ByteBuf

同样
, newHeapBuffer方法在
AbstractByteBufAllocator中也是一个抽象方法
, 具体逻辑交给其子类实现

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

2

newDirectBuffer
和
newHeapBuffer两个抽象方法中
, 在其子类
PooledByteBufAllocator和
UnpooledByteBufAllocator中都有实现

我们以
UnpooledByteBufAllocator的
newHeapBuffer方法为例
, 看其实现
:

1
2
3
4
5
1protected ByteBuf newHeapBuffer(int initialCapacity, int maxCapacity) {

2    return PlatformDependent.hasUnsafe() ? new UnpooledUnsafeHeapByteBuf(this, initialCapacity, maxCapacity)

3            : new UnpooledHeapByteBuf(this, initialCapacity, maxCapacity);

4}

5

里实现方式其实很简单
, 首先通过
PlatformDependent.hasUnsafe()判断当前运行环境是否能创建
unsafe对象
, 如果能
, 则直接通过
new UnpooledUnsafeHeapByteBuf(this, initialCapacity, maxCapacity)方式创建一个
UnpooledUnsafeHeapByteBuf对象
, 也就是一个
Unsafe的
ByteBuf对象

如果当前环境不能创建
unsafe对象
, 则通过
new UnpooledHeapByteBuf(this, initialCapacity, maxCapacity)这种方方式创建一个
UnpooledHeapByteBuf对象
, 也就是非
Unsafe的
ByteBuf对象

从这里能看出
, 其实在创建
ByteBuf对象时
, 是否创建
unsafe类型的对象并不是我们自己控制的
, 而是通过程序判断当前环境来决定是否创建
unsafe类型的
ByteBuf对象的

有关
ByteBufAllocator的继承关系如下
:

5-3-1

上一节: ByteBuf的分类

下一节: PooledByteBufAllocator简述