Netty 作為一款高性能的 RPC 框架必然涉及到頻繁的內存分配銷毀操作。
圖片來自 Pexels
如果是在堆上分配內存空間將會觸發頻繁的 GC，JDK 在 1.4 之后提供的 NIO 也已經提供了直接直接分配堆外內存空間的能力，但是也僅僅是提供了基本的能力，創建、回收相關的功能和效率都很簡陋。
基于此，在堆外內存使用方面，Netty 自己實現了一套創建、回收堆外內存池的相關功能。基于此我們一起來看一下 Netty 是如何實現內存分配的。
Netty 中的數據容器分類
談到數據保存肯定要說到內存分配，按照存儲空間來劃分，可以分為堆內存和堆外內存；按照內存區域連貫性來劃分可以分為池化內存和非池化內存。這些劃分在 Netty 中的實現接口分別如下。
按照底層存儲空間劃分：
- 堆緩沖區：HeapBuffer
- 直接緩沖區：DirectBuffer
按照是否池化劃分：
- 池化：PooledBuffer
- 非池化：UnPooledBuffer
默認使用 PoolDireBuf 類型的內存，這些內存主要由 PoolArea 管理。另外 Netty 并不是直接對外暴露這些 API，提供了 Unsafe 類作為出口暴露數據分配的相關操作。
什么是池化？
一般申請內存是檢查當前內存哪里有適合當前數據塊大小的空閑內存塊，如果有就將數據保存在當前內存塊中。
那么池化想做的事情是：既然每次來數據都要去找內存地址來存，我就先申請一塊內存地址，這一塊就是我的專用空間，內存分配、回收我全權管理。
池化解決的問題：內存碎片。
內碎片：就是申請的地址空間大于真正數據使用的內存空間。
比如固定申請 1M 的空間作為某個線程的使用內存，但是該線程每次最多只占用 0.5M，那么每次都有 0.5M 的碎片。如果該空間不被有效回收時間一長必然存在內存空洞。
外碎片：是指多個內存空間合并的時候發現不夠分配給待使用的空間大小。
比如有一個 20byte，13byte 的連續內存空間可以被回收，現在有一個 48byte 的數據塊需要存儲，而這兩個加起來也只有 33byte 的空間，必然不會被使用到。
如何實現內存池？
①鏈表維護空閑內存地址
最簡單的就是弄一個鏈表來維護當前空閑的內存空間地址。如果有使用就從鏈表刪除，有釋放就加入鏈表對應位置。
這種方式實現簡單，但是搜索和釋放內存維護的難度還是比較大，不太適合。
②定長內存空間分配
維護兩個列表，一個是未分配內存列表，一個是已分配內存列表。每個內存塊都是一樣大小，分配時如果不夠就將多個塊合并到一起。
這種方式的缺點就是會浪費一定的內存空間，如果有特定的場景還是沒有問題。
③多段定長池分配
在上面的定長分配基礎上，由原來的固定一個長度分配空間變為按照不同對象大小（8，16，32，64，128，256，512，1k…64K），的方式分配多個固定大小的內存池。
每次要申請內存的時候按照當前對象大小去對應的池中查找是否有剩余空間。
linux 本身支持動態內存分配和釋放，對應的命令為：malloc/free。malloc 的全稱是 memory allocation，中文叫動態內存分配，用于申請一塊連續的指定大小的內存塊區域以 void* 類型返回分配的內存區域地址。
malloc/free 的實現過程：
- 空閑存儲空間以空閑鏈表的方式組織（地址遞增），每個塊包含一個長度、一個指向下一塊的指針以及一個指向自身存儲空間的指針。（因為程序中的某些地方可能不通過 malloc 調用申請，因此 malloc 管理的空間不一定連續）
- 當有申請請求時，malloc 會掃描空閑鏈表，直到找到一個足夠大的塊為止。（首次適應）(因此每次調用 malloc 時并不是花費了完全相同的時間）
- 如果該塊恰好與請求的大小相符，則將其從鏈表中移走并返回給用戶。如果該塊太大，則將其分為兩部分，尾部的部分分給用戶，剩下的部分留在空閑鏈表中（更改頭部信息）。因此 malloc 分配的是一塊連續的內存。
- 釋放時首先搜索空閑鏈表，找到可以插入被釋放塊的合適位置。如果與被釋放塊相鄰的任一邊是一個空閑塊，則將這兩個塊合為一個更大的塊，以減少內存碎片。
Netty 中的內存分配
Netty 采用了 jemalloc 的思想，這是 FreeBSD 實現的一種并發 malloc 的算法。
jemalloc 依賴多個 Arena（分配器）來分配內存，運行中的應用都有固定數量的多個 Arena，默認的數量與處理器的個數有關。
系統中有多個 Arena 的原因是由于各個線程進行內存分配時競爭不可避免，這可能會極大的影響內存分配的效率，為了緩解高并發時的線程競爭，Netty 允許使用者創建多個分配器（Arena）來分離鎖，提高內存分配效率。
線程首次分配/回收內存時，首先會為其分配一個固定的 Arena。線程選擇 Arena 時使用 round-robin 的方式，也就是順序輪流選取。
每個線程各種保存 Arena 和緩存池信息，這樣可以減少競爭并提高訪問效率。Arena 將內存分為很多 Chunk 進行管理，Chunk 內部保存 Page，以頁為單位申請。
申請內存分配時，會將分配的規格分為如下四類，分別對應不同的范圍，處理過程也不相同：
- tiny：代表了大小在 0-512B 的內存塊。
- small：代表了大小在 512B-8K 的內存塊。
- normal：代表了大小在 8K-16M 的內存塊。
- huge：代表了大于 16M 的內存塊。
每個塊里面又定義了更細粒度的單位來分配數據：
- Chunk：一個 Chunk 的大小是 16M，Chunk 是 Netty 對操作系統進行內存申請的單位，后續所有的內存分配都是在 Chunk 里面進行操作。
- Page：Chunk 內部以 Page 為單位分配內存，一個 Page 大小為 8K。當我們需要 16K 的空間時，Netty 就會從一個 Chunk 中找到兩個 Page 進行分配。
- Subpage 和 element：element 是比 Page 更小的單位，當我們申請小于 8K 的內存時，Netty 會以 element 為單位進行內存分配。element 沒有固定大小，具體由用戶的需求決定。Netty 通過 Subpage 管理 element，Subpage 是由 Page 轉變過來的。當我們需要 1K 的空間時，Netty 會把一個 Page 變成 Subpage，然后把 Subpage 分成 8 個 1K 的 element 進行分配。
Chunk 中的內存分配
線程分配內存主要從兩個地方：PoolThreadCache 和 Arena。其中 PoolThreadCache 線程獨享，Arena 為幾個線程共享。
初次申請內存的時候，Netty 會從一整塊內存（Chunk）中分出一部分來給用戶使用，這部分工作是由 Arena 來完成。
而當用戶使用完畢釋放內存的時候，這些被分出來的內存會按不同規格大小放在 PoolThreadCache 中緩存起來。當下次要申請內存的時候，就會先從 PoolThreadCache 中找。
Chunk、Page、Subpage 和 element 都是 Arena 中的概念，Arena 的工作就是從一整塊內存中分出合適大小的內存塊。
Arena 中最大的內存單位是 Chunk，這是 Netty 向操作系統申請內存的單位。
而一塊 Chunk（16M） 申請下來之后，內部會被分成 2048 個 Page（8K），當用戶向 Netty 申請超過 8K 內存的時候，Netty 會以 Page 的形式分配內存。
Chunk 內部通過伙伴算法管理 Page，具體實現為一棵完全平衡二叉樹：
二叉樹中所有子節點管理的內存也屬于其父節點。當我們要申請大小為 16K 的內存時，我們會從根節點開始不斷尋找可用的節點，一直到第 10 層。
那么如何判斷一個節點是否可用呢？Netty 會在每個節點內部保存一個值，這個值代表這個節點之下的第幾層還存在未分配的節點。
比如第 9 層的節點的值如果為 9，就代表這個節點本身到下面所有的子節點都未分配。
如果第 9 層的節點的值為 10，代表它本身不可被分配，但第 10 層有子節點可以被分配。
如果第 9 層的節點的值為 12，此時可分配節點的深度大于了總深度，代表這個節點及其下面的所有子節點都不可被分配。
下圖描述了分配的過程：
對于小內存（小于 4096）的分配還會將 Page 細化成更小的單位 Subpage。
Subpage 按大小分有兩大類：
- Tiny：小于 512 的情況，最小空間為 16，對齊大小為 16，區間為[16,512)，所以共有 32 種情況。
- Small：大于等于 512 的情況，總共有四種：512，1024，2048，4096。
PoolSubpage 中直接采用位圖管理空閑空間（因為不存在申請 k 個連續的空間），所以申請釋放非常簡單。
第一次申請小內存空間的時候，需要先申請一個空閑頁，然后將該頁轉成 PoolSubpage，再將該頁設為已被占用，最后再把這個 PoolSubpage 存到 PoolSubpage 池中。
這樣下次就不需要再去申請空閑頁了，直接去池中找就好了。Netty 中有 36 種 PoolSubpage，所以用 36 個 PoolSubpage 鏈表表示 PoolSubpage 池。
因為單個 PoolChunk 只有 16M，這遠遠不夠用，所以會很很多很多 PoolChunk，這些 PoolChunk 組成一個鏈表，然后用 PoolChunkList 持有這個鏈表。
我們先從內存分配器 PoolArena 來分析 Netty 中的內存是如何分配的，Area 的工作就是從一整塊內存中協調如何分配合適大小的內存給當前數據使用。
PoolArena 是 Netty 的內存池實現抽象類，其內部子類為 HeapArena 和 DirectArena。
HeapArena 對應堆內存（heap buffer），DirectArena 對應堆外直接內存(direct buffer)，兩者除了操作的內存（byte[] 和 ByteBuffer）不同外其余完全一致。
從結構上來看，PoolArena 中主要包含三部分子內存池：
- tinySubpagePools
- smallSubpagePools
- 一系列的 PoolChunkList
tinySubpagePools 和 smallSubpagePools 都是 PoolSubpage 的數組，數組長度分別為 32 和 4。
PoolChunkList 是一個容器，其內部可以保存一系列的 PoolChunk 對象，并且，Netty 會根據內存使用率的不同，將 PoolChunkList 分為不同等級的容器。
abstract class PoolArena<T> implements PoolArenaMetric {
enum SizeClass {
Tiny,
Small,
Normal
}
// 該參數指定了tinySubpagePools數組的長度，由于tinySubpagePools每一個元素的內存塊差值為16，
// 因而數組長度是512/16，也即這里的512 >>> 4
static final int numTinySubpagePools = 512 >>> 4;
//表示該PoolArena的allocator
final PooledByteBufAllocator parent;
//表示PoolChunk中由Page節點構成的二叉樹的最大高度，默認11
private final int maxOrder;
//page的大小，默認8K
final int pageSize;
// 指定了葉節點大小8KB是2的多少次冪，默認為13，該字段的主要作用是，在計算目標內存屬于二叉樹的
// 第幾層的時候，可以借助于其內存大小相對于pageShifts的差值，從而快速計算其所在層數
final int pageShifts;
//默認16MB
final int chunkSize;
// 由于PoolSubpage的大小為8KB=8196，因而該字段的值為
// -8192=>=> 1111 1111 1111 1111 1110 0000 0000 0000
// 這樣在判斷目標內存是否小于8KB時，只需要將目標內存與該數字進行與操作，只要操作結果等于0，
// 就說明目標內存是小于8KB的，這樣就可以判斷其是應該首先在tinySubpagePools或smallSubpagePools
// 中進行內存申請
final int subpageOverflowMask;
// 該參數指定了smallSubpagePools數組的長度，默認為4
final int numSmallSubpagePools;
//tinySubpagePools用來分配小于512 byte的Page
private final PoolSubpage<T>[] tinySubpagePools;
//smallSubpagePools用來分配大于等于512 byte且小于pageSize內存的Page
private final PoolSubpage<T>[] smallSubpagePools;
//用來存儲用來分配給大于等于pageSize大小內存的PoolChunk
//存儲內存利用率50-100%的chunk
private final PoolChunkList<T> q050;
//存儲內存利用率25-75%的chunk
private final PoolChunkList<T> q025;
//存儲內存利用率1-50%的chunk
private final PoolChunkList<T> q000;
//存儲內存利用率0-25%的chunk
private final PoolChunkList<T> qInit;
//存儲內存利用率75-100%的chunk
private final PoolChunkList<T> q075;
//存儲內存利用率100%的chunk
private final PoolChunkList<T> q100;
//堆內存(heap buffer)
static final class HeapArena extends PoolArena<byte[]> {
}
//堆外直接內存(direct buffer)
static final class DirectArena extends PoolArena<ByteBuffer> {
}
}
如上所示，PoolArena 是由多個 PoolChunk 組成的大塊內存區域，而每個 PoolChunk 則由多個 Page 組成。
當需要分配的內存小于 Page 的時候，為了節約內存采用 PoolSubpage 實現小于 Page 大小內存的分配。
在 PoolArena 中為了保證 PoolChunk 空間的最大利用化，按照 PoolArena 中各 個 PoolChunk 已使用的空間大小將其劃分為六類：
- qInit：存儲內存利用率 0-25% 的 chunk
- q000：存儲內存利用率 1-50% 的 chunk
- q025：存儲內存利用率 25-75% 的 chunk
- q050：存儲內存利用率 50-100% 的 chunk
- q075：存儲內存利用率 75-100%的 chunk
- q100：存儲內存利用率 100%的 chunk
PoolArena 維護了一個 PoolChunkList 組成的雙向鏈表，每個 PoolChunkList 內部維護了一個 PoolChunk 雙向鏈表。
分配內存時，PoolArena 通過在 PoolChunkList 找到一個合適的 PoolChunk，然后從 PoolChunk 中分配一塊內存。
下面來看 PoolArena 是如何分配內存的：
private void allocate(PoolThreadCache cache, PooledByteBuf<T> buf, final int reqCapacity) {
// 將需要申請的容量格式為 2^N
final int normCapacity = normalizeCapacity(reqCapacity);
// 判斷目標容量是否小于8KB，小于8KB則使用tiny或small的方式申請內存
if (isTinyOrSmall(normCapacity)) { // capacity < pageSize
int tableIdx;
PoolSubpage<T>[] table;
boolean tiny = isTiny(normCapacity);
// 判斷目標容量是否小于512字節，小于512字節的為tiny類型的
if (tiny) { // < 512
// 將分配區域轉移到 tinySubpagePools 中
if (cache.allocateTiny(this, buf, reqCapacity, normCapacity)) {
// was able to allocate out of the cache so move on
return;
}
// 如果無法從當前線程緩存中申請到內存，則嘗試從tinySubpagePools中申請，這里tinyIdx()方法
// 就是計算目標內存是在tinySubpagePools數組中的第幾號元素中的
tableIdx = tinyIdx(normCapacity);
table = tinySubpagePools;
} else {
// 如果目標內存在512byte~8KB之間，則嘗試從smallSubpagePools中申請內存。這里首先從
// 當前線程的緩存中申請small級別的內存，如果申請到了，則直接返回
if (cache.allocateSmall(this, buf, reqCapacity, normCapacity)) {
// was able to allocate out of the cache so move on
return;
}
tableIdx = smallIdx(normCapacity);
table = smallSubpagePools;
}
// 獲取目標元素的頭結點
final PoolSubpage<T> head = table[tableIdx];
// 這里需要注意的是，由于對head進行了加鎖，而在同步代碼塊中判斷了s != head，
// 也就是說PoolSubpage鏈表中是存在未使用的PoolSubpage的，因為如果該節點已經用完了，
// 其是會被移除當前鏈表的。也就是說只要s != head，那么這里的allocate()方法
// 就一定能夠申請到所需要的內存塊
synchronized (head) {
// s != head就證明當前PoolSubpage鏈表中存在可用的PoolSubpage，并且一定能夠申請到內存，
// 因為已經耗盡的PoolSubpage是會從鏈表中移除的
final PoolSubpage<T> s = head.next;
// 如果此時 subpage 已經被分配過內存了執行下文，如果只是初始化過，則跳過該分支
if (s != head) {
// 從PoolSubpage中申請內存
assert s.doNotDestroy && s.elemSize == normCapacity;
// 通過申請的內存對ByteBuf進行初始化
long handle = s.allocate();
assert handle >= 0;
// 初始化 PoolByteBuf 說明其位置被分配到該區域，但此時尚未分配內存
s.chunk.initBufWithSubpage(buf, handle, reqCapacity);
// 對tiny類型的申請數進行更新
if (tiny) {
allocationsTiny.increment();
} else {
allocationsSmall.increment();
}
return;
}
}
// 走到這里，說明目標PoolSubpage鏈表中無法申請到目標內存塊，因而就嘗試從PoolChunk中申請
allocateNormal(buf, reqCapacity, normCapacity);
return;
}
// 走到這里說明目標內存是大于8KB的，那么就判斷目標內存是否大于16M，如果大于16M，
// 則不使用內存池對其進行管理，如果小于16M，則到PoolChunkList中進行內存申請
if (normCapacity <= chunkSize) {
// 小于16M，首先到當前線程的緩存中申請，如果申請到了則直接返回，如果沒有申請到，
// 則到PoolChunkList中進行申請
if (cache.allocateNormal(this, buf, reqCapacity, normCapacity)) {
// was able to allocate out of the cache so move on
return;
}
allocateNormal(buf, reqCapacity, normCapacity);
} else {
// 對于大于16M的內存，Netty不會對其進行維護，而是直接申請，然后返回給用戶使用
allocateHuge(buf, reqCapacity);
}
}
所有內存分配的 size 都會經過 normalizeCapacity() 進行處理，申請的容量總是會被格式為 2^N。
主要規則如下：
- 如果目標容量小于 16 字節，則返回 16。
- 如果目標容量大于 16 字節，小于 512 字節，則以 16 字節為單位，返回大于目標字節數的第一個 16 字節的倍數。比如申請的 100 字節，那么大于 100 的 16 整數倍最低為：16*7=112，因而返回 112。
- 如果目標容量大于 512 字節，則返回大于目標容量的第一個 2 的指數冪。比如申請的 1000 字節，那么返回的將是：2^10 = 1024。
PoolArena 提供了兩種方式進行內存分配：
①PoolSubpage 用于分配小于 8k 的內存
tinySubpagePools：用于分配小于 512 字節的內存，默認長度為 32，因為內存分配最小為 16，每次增加 16，直到 512，區間 [16，512) 一共有 32 個不同值。
smallSubpagePools：用于分配大于等于 512 字節的內存，默認長度為 4。tinySubpagePools 和 smallSubpagePools 中的元素默認都是 subpage。
②poolChunkList 用于分配大于 8k 的內存
上面已經解釋了 q 開頭的幾個變量用于保存大于 8k 的數據。
默認先嘗試從 poolThreadCache 中分配內存，PoolThreadCache 利用 ThreadLocal 的特性，消除了多線程競爭，提高內存分配效率；
首次分配時，poolThreadCache 中并沒有可用內存進行分配，當上一次分配的內存使用完并釋放時，會將其加入到 poolThreadCache 中，提供該線程下次申請時使用。
如果是分配小內存，則嘗試從 tinySubpagePools 或 smallSubpagePools 中分配內存，如果沒有合適 subpage，則采用方法 allocateNormal 分配內存。
如果分配一個 page 以上的內存，直接采用方法 allocateNormal() 分配內存，allocateNormal() 則會將申請動作交由 PoolChunkList 進行。
private synchronized void allocateNormal(PooledByteBuf<T> buf, int reqCapacity, int normCapacity) {
//如果在對應的PoolChunkList能申請到內存，則返回
if (q050.allocate(buf, reqCapacity, normCapacity) || q025.allocate(buf, reqCapacity, normCapacity) ||
q000.allocate(buf, reqCapacity, normCapacity) || qInit.allocate(buf, reqCapacity, normCapacity) ||
q075.allocate(buf, reqCapacity, normCapacity)) {
++allocationsNormal;
return;
}
// Add a new chunk.
PoolChunk<T> c = newChunk(pageSize, maxOrder, pageShifts, chunkSize);
long handle = c.allocate(normCapacity);
++allocationsNormal;
assert handle > 0;
c.initBuf(buf, handle, reqCapacity);
qInit.add(c);
}
首先將申請動作按照 q050→q025→q000→qInit→q075 的順序依次交由各個 PoolChunkList 進行處理，如果在對應的 PoolChunkList 中申請到了內存，則直接返回。
如果申請不到，那么直接創建一個新的 PoolChunk，然后在該 PoolChunk 中申請目標內存，最后將該 PoolChunk 添加到 qInit 中。
上面說過 Chunk 是 Netty 向操作系統申請內存塊的最大單位，每個 Chunk 是 16M。
PoolChunk 內部通過 memoryMap 數組維護了一顆完全平衡二叉樹作為管理底層內存分布及回收的標記位，所有的子節點管理的內存也屬于其父節點。
關于 PoolChunk 內部如何維護完全平衡二叉樹就不在這里展開，大家有興趣可以自行看源碼。
對于內存的釋放，PoolArena 主要是分為兩種情況，即池化和非池化，如果是非池化，則會直接銷毀目標內存塊，如果是池化的，則會將其添加到當前線程的緩存中。
如下是 free() 方法的源碼：
public void free(PoolChunk<T> chunk, ByteBuffer nioBuffer, long handle, int normCapacity,
PoolThreadCache cache) {
// 如果是非池化的，則直接銷毀目標內存塊，并且更新相關的數據
if (chunk.unpooled) {
int size = chunk.chunkSize();
destroyChunk(chunk);
activeBytesHuge.add(-size);
deallocationsHuge.increment();
} else {
// 如果是池化的，首先判斷其是哪種類型的，即tiny，small或者normal，
// 然后將其交由當前線程的緩存進行處理，如果添加成功，則直接返回
SizeClass sizeClass = sizeClass(normCapacity);
if (cache != null && cache.add(this, chunk, nioBuffer, handle,
normCapacity, sizeClass)) {
return;
}
// 如果當前線程的緩存已滿，則將目標內存塊返還給公共內存塊進行處理
freeChunk(chunk, handle, sizeClass, nioBuffer);
}
}
作者：rickiyang
編輯：陶家龍
出處：轉載自微信公眾號 rickiyang
