Fix memory release failure when "maxNumElems == 1" of PoolSubpage (#10988)

Motivation:

when customer need large of 'byteBuf.capacity' in [7168, 8192], the size of 'chunk.subpages' may be inflated when large of byteBuf be released, not consistent with other 'byteBuf.capacity'

Modification:

when maxNumElems == 1 need consider remove from pool

Result:

Fixes #10896. 

Co-authored-by: zxingy <zxingy@servyou.com.cn>
This commit is contained in:
Zxy 2021-02-05 21:53:58 +08:00 committed by Norman Maurer
parent d7fb0f5c0b
commit 87392634d2
2 changed files with 32 additions and 3 deletions

View File

@ -115,7 +115,13 @@ final class PoolSubpage<T> implements PoolSubpageMetric {
if (numAvail ++ == 0) { if (numAvail ++ == 0) {
addToPool(head); addToPool(head);
return true; /* When maxNumElems == 1, the maximum numAvail is also 1.
* Each of these PoolSubpages will go in here when they do free operation.
* If they return true directly from here, then the rest of the code will be unreachable
* and they will not actually be recycled. So return true only on maxNumElems > 1. */
if (maxNumElems > 1) {
return true;
}
} }
if (numAvail != maxNumElems) { if (numAvail != maxNumElems) {

View File

@ -622,8 +622,8 @@ public class PooledByteBufAllocatorTest extends AbstractByteBufAllocatorTest<Poo
} }
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
private static PooledByteBuf<ByteBuffer> unwrapIfNeeded(ByteBuf buf) { private static <T> PooledByteBuf<T> unwrapIfNeeded(ByteBuf buf) {
return (PooledByteBuf<ByteBuffer>) (buf instanceof PooledByteBuf ? buf : buf.unwrap()); return (PooledByteBuf<T>) (buf instanceof PooledByteBuf ? buf : buf.unwrap());
} }
@Test @Test
@ -651,4 +651,27 @@ public class PooledByteBufAllocatorTest extends AbstractByteBufAllocatorTest<Poo
assertEquals(3, allocator.metric().directArenas().get(0).numNormalAllocations()); assertEquals(3, allocator.metric().directArenas().get(0).numNormalAllocations());
buffer.release(); buffer.release();
} }
@Test
public void testNormalPoolSubpageRelease() {
// 16 < elemSize <= 7168 or 8192 < elemSize <= 28672, 1 < subpage.maxNumElems <= 256
// 7168 <= elemSize <= 8192, subpage.maxNumElems == 1
int elemSize = 8192;
int length = 1024;
ByteBuf[] byteBufs = new ByteBuf[length];
final PooledByteBufAllocator allocator = new PooledByteBufAllocator(false, 32, 32, 8192, 11, 256, 64, false, 0);
for (int i = 0; i < length; i++) {
byteBufs[i] = allocator.heapBuffer(elemSize, elemSize);
}
PoolChunk<Object> chunk = unwrapIfNeeded(byteBufs[0]).chunk;
int beforeFreeBytes = chunk.freeBytes();
for (int i = 0; i < length; i++) {
byteBufs[i].release();
}
int afterFreeBytes = chunk.freeBytes();
assertTrue(beforeFreeBytes < afterFreeBytes);
}
} }