1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.eclipse.jetty.io;
20
21 import java.nio.ByteBuffer;
22 import java.util.Queue;
23 import java.util.concurrent.ConcurrentHashMap;
24 import java.util.concurrent.ConcurrentLinkedQueue;
25 import java.util.concurrent.ConcurrentMap;
26
27 import org.eclipse.jetty.util.BufferUtil;
28
29 public class MappedByteBufferPool implements ByteBufferPool
30 {
31 private final ConcurrentMap<Integer, Queue<ByteBuffer>> directBuffers = new ConcurrentHashMap<>();
32 private final ConcurrentMap<Integer, Queue<ByteBuffer>> heapBuffers = new ConcurrentHashMap<>();
33 private final int factor;
34
35 public MappedByteBufferPool()
36 {
37 this(1024);
38 }
39
40 public MappedByteBufferPool(int factor)
41 {
42 this.factor = factor;
43 }
44
45 @Override
46 public ByteBuffer acquire(int size, boolean direct)
47 {
48 int bucket = bucketFor(size);
49 ConcurrentMap<Integer, Queue<ByteBuffer>> buffers = buffersFor(direct);
50
51 ByteBuffer result = null;
52 Queue<ByteBuffer> byteBuffers = buffers.get(bucket);
53 if (byteBuffers != null)
54 result = byteBuffers.poll();
55
56 if (result == null)
57 {
58 int capacity = bucket * factor;
59 result = direct ? BufferUtil.allocateDirect(capacity) : BufferUtil.allocate(capacity);
60 }
61
62 BufferUtil.clear(result);
63 return result;
64 }
65
66 @Override
67 public void release(ByteBuffer buffer)
68 {
69 if (buffer == null)
70 return;
71
72
73 assert((buffer.capacity() % factor) == 0);
74
75 int bucket = bucketFor(buffer.capacity());
76 ConcurrentMap<Integer, Queue<ByteBuffer>> buffers = buffersFor(buffer.isDirect());
77
78
79 Queue<ByteBuffer> byteBuffers = buffers.get(bucket);
80 if (byteBuffers == null)
81 {
82 byteBuffers = new ConcurrentLinkedQueue<>();
83 Queue<ByteBuffer> existing = buffers.putIfAbsent(bucket, byteBuffers);
84 if (existing != null)
85 byteBuffers = existing;
86 }
87
88 BufferUtil.clear(buffer);
89 byteBuffers.offer(buffer);
90 }
91
92 public void clear()
93 {
94 directBuffers.clear();
95 heapBuffers.clear();
96 }
97
98 private int bucketFor(int size)
99 {
100 int bucket = size / factor;
101 if (size % factor > 0)
102 ++bucket;
103 return bucket;
104 }
105
106
107 ConcurrentMap<Integer, Queue<ByteBuffer>> buffersFor(boolean direct)
108 {
109 return direct ? directBuffers : heapBuffers;
110 }
111 }