com.facebook.drift.transport.netty.buffer

Class TestingPooledByteBufAllocator

java.lang.Object

io.netty.buffer.AbstractByteBufAllocator

io.netty.buffer.PooledByteBufAllocator

com.facebook.drift.transport.netty.buffer.TestingPooledByteBufAllocator

All Implemented Interfaces:

io.netty.buffer.ByteBufAllocator, io.netty.buffer.ByteBufAllocatorMetricProvider, Closeable, AutoCloseable

public class TestingPooledByteBufAllocator
extends io.netty.buffer.PooledByteBufAllocator
implements Closeable

This is a custom ByteBufAllocator that tracks outstanding allocations and throws from the close() method if it detects any leaked buffers.

Never use this class in production, it will cause your server to run out of memory! This is because it holds strong references to all allocated buffers and doesn't release them until close() is called at the end of a unit test.

Field Summary

Fields inherited from class io.netty.buffer.PooledByteBufAllocator

DEFAULT

Constructor Summary

Constructors

Constructor and Description
TestingPooledByteBufAllocator()

Method Summary

Modifier and Type	Method and Description
void	close()
io.netty.buffer.CompositeByteBuf	compositeDirectBuffer(int maxNumComponents)
io.netty.buffer.CompositeByteBuf	compositeHeapBuffer(int maxNumComponents)
protected io.netty.buffer.ByteBuf	newDirectBuffer(int initialCapacity, int maxCapacity)
protected io.netty.buffer.ByteBuf	newHeapBuffer(int initialCapacity, int maxCapacity)

Methods inherited from class io.netty.buffer.PooledByteBufAllocator

chunkSize, defaultMaxOrder, defaultNormalCacheSize, defaultNumDirectArena, defaultNumHeapArena, defaultPageSize, defaultPreferDirect, defaultSmallCacheSize, defaultTinyCacheSize, defaultUseCacheForAllThreads, directArenas, dumpStats, freeThreadLocalCache, hasThreadLocalCache, heapArenas, isDirectBufferPooled, isDirectMemoryCacheAlignmentSupported, metric, normalCacheSize, numDirectArenas, numHeapArenas, numThreadLocalCaches, smallCacheSize, tinyCacheSize

Methods inherited from class io.netty.buffer.AbstractByteBufAllocator

buffer, buffer, buffer, calculateNewCapacity, compositeBuffer, compositeBuffer, compositeDirectBuffer, compositeHeapBuffer, directBuffer, directBuffer, directBuffer, heapBuffer, heapBuffer, heapBuffer, ioBuffer, ioBuffer, ioBuffer, toLeakAwareBuffer, toLeakAwareBuffer, toString

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

TestingPooledByteBufAllocator

public TestingPooledByteBufAllocator()

Method Detail

newHeapBuffer

protected io.netty.buffer.ByteBuf newHeapBuffer(int initialCapacity,
                                                int maxCapacity)

Overrides:

newHeapBuffer in class io.netty.buffer.PooledByteBufAllocator

newDirectBuffer

protected io.netty.buffer.ByteBuf newDirectBuffer(int initialCapacity,
                                                  int maxCapacity)

Overrides:

newDirectBuffer in class io.netty.buffer.PooledByteBufAllocator

compositeHeapBuffer

public io.netty.buffer.CompositeByteBuf compositeHeapBuffer(int maxNumComponents)

Specified by:

compositeHeapBuffer in interface io.netty.buffer.ByteBufAllocator

Overrides:

compositeHeapBuffer in class io.netty.buffer.AbstractByteBufAllocator

compositeDirectBuffer

public io.netty.buffer.CompositeByteBuf compositeDirectBuffer(int maxNumComponents)

Specified by:

compositeDirectBuffer in interface io.netty.buffer.ByteBufAllocator

Overrides:

compositeDirectBuffer in class io.netty.buffer.AbstractByteBufAllocator

close

public void close()

Specified by:

close in interface Closeable

Specified by:

close in interface AutoCloseable