Sergey Mashkov | 86f7061 | 2017-07-25 10:59:44 +0300 | [diff] [blame] | 1 | package kotlinx.coroutines.experimental.io.internal |
| 2 | |
Sergey Mashkov | 6d05993 | 2017-10-06 18:46:23 +0300 | [diff] [blame] | 3 | import kotlinx.io.pool.* |
| 4 | import java.nio.* |
Sergey Mashkov | a10b692 | 2017-08-03 13:41:54 +0300 | [diff] [blame] | 5 | |
Roman Elizarov | 8376203 | 2017-08-08 19:58:40 +0300 | [diff] [blame] | 6 | internal val BUFFER_SIZE = getIOIntProperty("BufferSize", 4096) |
| 7 | private val BUFFER_POOL_SIZE = getIOIntProperty("BufferPoolSize", 2048) |
| 8 | private val BUFFER_OBJECT_POOL_SIZE = getIOIntProperty("BufferObjectPoolSize", 1024) |
Sergey Mashkov | a10b692 | 2017-08-03 13:41:54 +0300 | [diff] [blame] | 9 | |
Roman Elizarov | 8376203 | 2017-08-08 19:58:40 +0300 | [diff] [blame] | 10 | // ------------- standard shared pool objects ------------- |
| 11 | |
| 12 | internal val BufferPool: ObjectPool<ByteBuffer> = |
Sergey Mashkov | 6d05993 | 2017-10-06 18:46:23 +0300 | [diff] [blame] | 13 | object : DefaultPool<ByteBuffer>(BUFFER_POOL_SIZE) { |
Roman Elizarov | 8376203 | 2017-08-08 19:58:40 +0300 | [diff] [blame] | 14 | override fun produceInstance(): ByteBuffer = |
| 15 | ByteBuffer.allocateDirect(BUFFER_SIZE) |
| 16 | override fun clearInstance(instance: ByteBuffer): ByteBuffer = |
| 17 | instance.also { it.clear() } |
| 18 | override fun validateInstance(instance: ByteBuffer) { |
| 19 | require(instance.capacity() == BUFFER_SIZE) |
| 20 | } |
Sergey Mashkov | a10b692 | 2017-08-03 13:41:54 +0300 | [diff] [blame] | 21 | } |
Sergey Mashkov | 86f7061 | 2017-07-25 10:59:44 +0300 | [diff] [blame] | 22 | |
Roman Elizarov | 8376203 | 2017-08-08 19:58:40 +0300 | [diff] [blame] | 23 | internal val BufferObjectPool: ObjectPool<ReadWriteBufferState.Initial> = |
Sergey Mashkov | 6d05993 | 2017-10-06 18:46:23 +0300 | [diff] [blame] | 24 | object: DefaultPool<ReadWriteBufferState.Initial>(BUFFER_OBJECT_POOL_SIZE) { |
Roman Elizarov | 8376203 | 2017-08-08 19:58:40 +0300 | [diff] [blame] | 25 | override fun produceInstance() = |
| 26 | ReadWriteBufferState.Initial(BufferPool.borrow()) |
| 27 | override fun disposeInstance(instance: ReadWriteBufferState.Initial) { |
| 28 | BufferPool.recycle(instance.backingBuffer) |
| 29 | } |
| 30 | } |
Sergey Mashkov | 86f7061 | 2017-07-25 10:59:44 +0300 | [diff] [blame] | 31 | |
Roman Elizarov | 8376203 | 2017-08-08 19:58:40 +0300 | [diff] [blame] | 32 | internal val BufferObjectNoPool: ObjectPool<ReadWriteBufferState.Initial> = |
| 33 | object : NoPoolImpl<ReadWriteBufferState.Initial>() { |
| 34 | override fun borrow(): ReadWriteBufferState.Initial = |
| 35 | ReadWriteBufferState.Initial(ByteBuffer.allocateDirect(BUFFER_SIZE)) |
| 36 | } |