❃博主首页 ：

「码到三十五」 ，同名公众号 :「码到三十五」

♝博主的话 ：

搬的每块砖，皆为峰峦之基；公众号搜索「码到三十五」关注这个爱发技术干货的coder，一起筑基

Netty，作为一款高性能的网络编程框架，其背后的内存管理机制起着至关重要的作用。其中，PooledByteBufAllocator是Netty内存管理中的一个核心组件，它实现了基于内存池的字节缓冲区（ByteBuf）分配策略。本文将结合源码，深入探讨PooledByteBufAllocator的工作原理、实现细节以及其在Netty中的作用。

一、PooledByteBufAllocator概述

PooledByteBufAllocator是Netty提供的一种ByteBufAllocator实现，它使用内存池来管理ByteBuf的分配和回收。相比不使用内存池的分配器（如UnpooledByteBufAllocator），PooledByteBufAllocator能够显著减少内存的分配和回收开销，提高性能。

二、工作原理

PooledByteBufAllocator的工作原理是围绕内存池化技术展开的，它通过预先分配和重用内存块来减少内存分配和释放的开销，从而提高性能.

1. 内存池初始化
  • PoolArena数组：PooledByteBufAllocator内部维护了一个PoolArena数组，每个PoolArena代表一块连续的内存区域。这些PoolArena可以被多个线程共享，但在高并发场景下，Netty会创建与线程数量相等的PoolArena实例，以减少线程间的竞争。
  • Chunk和Page：每个PoolArena由多个PoolChunkList组成，每个PoolChunkList包含多个PoolChunk。PoolChunk是内存分配的基本单位，默认大小为16MB（这个值可以根据需要调整）。每个PoolChunk又被划分为多个Page，每个Page的大小通常为8KB（这个值也是可配置的）。
2. 内存分配

当需要分配一个新的ByteBuf时，PooledByteBufAllocator会遵循以下步骤：

  • 线程局部缓存（PoolThreadCache）：首先，Netty会尝试从当前线程的PoolThreadCache中分配内存。PoolThreadCache是保存在ThreadLocal上的，因此每个线程都有自己独立的缓存，这有助于减少线程间的竞争。
  • PoolArena分配：如果PoolThreadCache中没有足够的内存可供分配，Netty会转向对应的PoolArena进行内存分配。在PoolArena中，Netty会根据请求的内存大小选择不同的分配策略。
  • Tiny和Small内存分配：对于小于Page大小（通常为8KB）的内存请求，Netty会使用tinySubpagePools或smallSubpagePools进行分配。这些子页池会将Page进一步划分为更小的内存块，以满足小内存请求的需求。
  • Normal内存分配：对于大于等于Page大小但小于Chunk大小的内存请求，Netty会直接在Page上进行分配。
  • Huge内存分配：对于大于Chunk大小的内存请求，Netty会分配一个独立的Chunk给请求者，但这通常不是池化内存的一部分。
  • 内存规格化：在分配内存时，Netty会对请求的内存大小进行规格化，以确保分配的内存块大小是标准的、易于管理的。例如，如果请求的内存大小为20字节，Netty可能会将其规格化为32字节。
3. 内存回收

当ByteBuf不再需要时，Netty会将其释放回内存池。释放过程遵循以下步骤：

  • PoolThreadCache回收：首先，Netty会尝试将释放的ByteBuf缓存到当前线程的PoolThreadCache中，以便后续重用。
  • PoolArena回收：如果PoolThreadCache已满或无法缓存释放的ByteBuf，Netty会将其释放回对应的PoolArena。在PoolArena中，释放的内存块会被标记为空闲，并添加到空闲列表中。
4. 内存清理

为了防止内存泄漏，Netty会定期清理内存池中的无效内存块。这通常包括检查空闲列表中的内存块，移除长时间未被使用的内存块，以及回收那些不再需要的Chunk。

5. 并发优化

在高并发场景下，Netty通过以下方式优化PooledByteBufAllocator的性能：

  • 多个PoolArena实例：为每个线程或每个CPU核心分配一个独立的PoolArena实例，以减少线程间的竞争。
  • 线程绑定：Netty在分配内存时会尽量将内存块分配给请求它的线程绑定的PoolArena，以减少跨线程操作。
  • 缓存策略：通过PoolThreadCache为每个线程提供独立的缓存，减少线程间的内存分配竞争。

综上所述，PooledByteBufAllocator通过内存池化技术、线程局部缓存、规格化内存分配和回收策略以及并发优化等措施，实现了高效的内存管理，为Netty的高性能网络通信提供了有力支持。

三、源码解析

以下是PooledByteBufAllocator的一些关键源码：

public class PooledByteBufAllocator extends ByteBufAllocator {
    // 内存池数组，用于存储不同大小的内存池
    private final PoolChunkList<PooledByteBuf<byte[]>>[] chunkLists;
    // ... 其他字段 ...

    @Override
    protected ByteBuf newDirectBuffer(int initialCapacity, int maxCapacity) {
        PoolThreadCache cache = threadCache.get();
        PoolArena<byte[]> directArena = cache.directArena();

        ByteBuf buf;
        if (directArena != null) {
            buf = directArena.allocate(cache, initialCapacity, maxCapacity);
        } else {
            buf = PlatformDependent.hasUnsafe() ?
                UnsafeByteBufUtil.newUnsafeDirectByteBuf(this, initialCapacity, maxCapacity) :
                new UnpooledDirectByteBuf(this, initialCapacity, maxCapacity);
        }

        return toLeakAwareBuffer(buf);
    }

    // ... 其他方法 ...
}
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.
  • 16.
  • 17.
  • 18.
  • 19.
  • 20.
  • 21.
  • 22.
  • 23.
  • 24.

在上面的代码中，newDirectBuffer方法是PooledByteBufAllocator用于分配新的直接内存ByteBuf的实现。它首先尝试从线程缓存（PoolThreadCache）中获取一个内存池（PoolArena），然后从该内存池中分配一个新的ByteBuf。如果线程缓存中没有可用的内存池，它将使用UnsafeByteBufUtil或UnpooledDirectByteBuf来分配一个新的ByteBuf。

四、内存池管理

PooledByteBufAllocator使用一组内存池来管理ByteBuf的分配和回收。每个内存池由多个内存块组成，每个内存块可以存储多个ByteBuf实例。内存池的管理包括内存块的分配、回收和清理等操作。

当需要分配一个新的ByteBuf时，PooledByteBufAllocator会首先查找一个合适的空闲内存块，并在其中分配一个新的ByteBuf。如果内存块中没有足够的空间，它会创建一个新的内存块，并将其添加到内存池中。

当ByteBuf被释放时，PooledByteBufAllocator会将其标记为空闲，并将其添加到内存块的空闲列表中。如果内存块中的所有ByteBuf都被释放了，该内存块将被添加到内存池的空闲列表中，以便后续重用。

五、性能优势

使用PooledByteBufAllocator相比不使用内存池的分配器具有显著的性能优势。通过内存池化，它可以减少内存的分配和回收开销，降低垃圾回收（GC）的影响，并提高内存的利用率。这使得PooledByteBufAllocator成为Netty中处理大量网络数据时的高效选择。

六、总结

PooledByteBufAllocator是Netty中基于内存池的ByteBuf分配器实现。它通过内存池来管理ByteBuf的分配和回收，显著减少了内存的分配和回收开销，提高了性能。本文结合源码详细介绍了PooledByteBufAllocator的工作原理、实现细节以及其在Netty中的作用，希望对读者深入理解Netty的内存管理机制有所帮助。

关注公众号[码到三十五]获取更多技术干货 !