如何设置Netty的接收Buffer为堆内存模式

Posted 2021-04-30 Netty之家

如何设置Netty的接收Buffer为堆内存模式

Netty为了提升报文的读写性能，默认会采用“零拷贝”模式，即消息读取时使用非堆的DirectBuffer来减少ByteBuffer的内存拷贝，如下图所示：

---

如果需要修改接收Buffer的类型，例如从DirectByteBuf修改为HeapByteBuf，首先需要在初始化Channel的时候对接收缓冲区进行设置，客户端代码示例如下：

---

 b.group(group)

.channel(NiosocketChannel.class)

 .option(ChannelOption.TCP_NODELAY, true)

 .option(ChannelOption.ALLOCATOR, PooledByteBufAllocator.DEFAULT)

 .option(ChannelOption.RCVBUF_ALLOCATOR, AdaptiveRecvByteBufAllocator.DEFAULT)

---

如果是服务端，则需要在链路创建之后，初始化时对Channel的SocketChannelConfig的Allocator属性进行设置，代码如下：

---

.childHandler(new ChannelInitializer<SocketChannel>() {          

 @Override

  public void initChannel(SocketChannel ch) throws Exception {

  ch.config().setAllocator(UnpooledByteBufAllocator.DEFAULT);

---

无论是通过ChannelOption.RCVBUF_ALLOCATOR还是Allocator，都只能设置Allocator的类型，无法直接设置ByteBufAllocator分配的ByteBuf类型。下面我们接着分析消息读取时的ByteBuf分配机制。消息读取时，调用的是NioByteUnsafe的read方法，代码如下：

---

---

首先从SocketChannelConfig中获取ByteBufAllocator，在这里就是UnpooledByteBufAllocator，然后调用它的ioBuffer方法分配内存，它的具体实现如下：

---

 @Override

    public ByteBuf ioBuffer() {

        if (PlatformDependent.hasUnsafe()) {

            return directBuffer(0);

        }

        return heapBuffer(0);

    }

---

如果非Unsafe模式，则会使用堆内存heapBuffer，接着看hasUnsafe方法实现，它最终会调用如下方法：

---

---

设置io.netty.noUnsafe属性为true，则默认会使用Heap堆内存创建ByteBuf，下面我们在启动时设置-Dio.netty.noUnsafe为true进行测试

---

---

设置完成之后，使用Echo程序进行测试，测试结果如下：

---

总结

利用ch.config().setAllocator或Bootstrap.option(ChannelOption.ALLOCATOR, ByteBufAllocator)，结合-Dio.netty.noUnsafe，可以灵活的在如下四种ByteBuf之间进行切换：

• UnpooledHeapByteBuf

• PooledHeapByteBuf

• UnpooledDirectByteBuf

• PooledDirectByteBuf