|
17 | 17 | */ |
18 | 18 | public class PersistentMappedArrayBuffer { |
19 | 19 | private final ArrayList<PersistentMappedBuffer> buffers = new ArrayList<>(); |
20 | | - protected int byteSize, usingIndex = -1; |
| 20 | + private PersistentMappedBuffer lastUsing = null; |
| 21 | + protected int byteSize, usingIndex = 0; |
21 | 22 | private static final int N = 4; |
22 | | - private boolean trapping = false; |
23 | 23 |
|
24 | 24 | public PersistentMappedArrayBuffer() { |
| 25 | + ensureCapacity(1024 * 1024); |
25 | 26 | } |
26 | 27 |
|
27 | 28 | public void ensureCapacity(int byteSize) { |
28 | | - PersistentMappedBuffer using = usingIndex == -1 ? null : buffers.get(usingIndex); |
29 | 29 | if (byteSize > this.byteSize) { |
30 | | - buffers.forEach(persistentMappedBuffer -> { |
31 | | - if (persistentMappedBuffer != using) { |
32 | | - persistentMappedBuffer.free(); |
| 30 | + if (lastUsing != null) { |
| 31 | + lastUsing.free(); |
| 32 | + } |
| 33 | + if (!buffers.isEmpty()) { |
| 34 | + lastUsing = buffers.get(usingIndex); |
| 35 | + } |
| 36 | + for (var buffer : buffers) { |
| 37 | + if (buffer != lastUsing) { |
| 38 | + buffer.free(); |
33 | 39 | } |
34 | | - }); |
| 40 | + } |
35 | 41 | buffers.clear(); |
36 | 42 | this.byteSize = (int) (byteSize * 1.25); |
37 | 43 | buffers.addAll(IntStream.range(0, N).mapToObj(i -> new PersistentMappedBuffer(this.byteSize)).toList()); |
38 | | - if (using != null) { |
39 | | - buffers.add(using); |
40 | | - trapping = true; |
41 | | - } |
| 44 | + usingIndex = 0; |
42 | 45 | } |
43 | 46 | } |
44 | 47 |
|
45 | | - @SuppressWarnings("SequencedCollectionMethodCanBeUsed") |
46 | 48 | public PersistentMappedBuffer getNext() { |
47 | | - return trapping ? buffers.get(0) : buffers.get((usingIndex + 1) % N); |
| 49 | + return buffers.get((usingIndex + 1) % N); |
48 | 50 | } |
49 | 51 |
|
50 | 52 | public PersistentMappedBuffer getCurrent() { |
51 | | - return trapping ? buffers.get(N) : buffers.get(usingIndex); |
| 53 | + if (lastUsing != null) { |
| 54 | + return lastUsing; |
| 55 | + } |
| 56 | + return buffers.get(usingIndex); |
52 | 57 | } |
53 | 58 |
|
54 | 59 | public void next() { |
55 | | - if (trapping) { |
56 | | - trapping = false; |
57 | | - usingIndex = 0; |
58 | | - return; |
| 60 | + if (lastUsing != null) { |
| 61 | + lastUsing.free(); |
| 62 | + lastUsing = null; |
59 | 63 | } |
60 | 64 | usingIndex = (usingIndex + 1) % N; |
61 | 65 | } |
|
0 commit comments