01: /*
02: * Licensed to the Apache Software Foundation (ASF) under one or more
03: * contributor license agreements. See the NOTICE file distributed with
04: * this work for additional information regarding copyright ownership.
05: * The ASF licenses this file to You under the Apache License, Version 2.0
06: * (the "License"); you may not use this file except in compliance with
07: * the License. You may obtain a copy of the License at
08: *
09: * http://www.apache.org/licenses/LICENSE-2.0
10: *
11: * Unless required by applicable law or agreed to in writing, software
12: * distributed under the License is distributed on an "AS IS" BASIS,
13: * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14: * See the License for the specific language governing permissions and
15: * limitations under the License.
16: */
17: package org.apache.catalina.tribes.io;
18:
19: import java.util.concurrent.ConcurrentLinkedQueue;
20: import java.util.concurrent.atomic.AtomicInteger;
21:
22: /**
23: *
24: * @author Filip Hanik
25: * @version 1.0
26: */
27: class BufferPool15Impl implements BufferPool.BufferPoolAPI {
28: protected int maxSize;
29: protected AtomicInteger size = new AtomicInteger(0);
30: protected ConcurrentLinkedQueue queue = new ConcurrentLinkedQueue();
31:
32: public void setMaxSize(int bytes) {
33: this .maxSize = bytes;
34: }
35:
36: public XByteBuffer getBuffer(int minSize, boolean discard) {
37: XByteBuffer buffer = (XByteBuffer) queue.poll();
38: if (buffer != null)
39: size.addAndGet(-buffer.getCapacity());
40: if (buffer == null)
41: buffer = new XByteBuffer(minSize, discard);
42: else if (buffer.getCapacity() <= minSize)
43: buffer.expand(minSize);
44: buffer.setDiscard(discard);
45: buffer.reset();
46: return buffer;
47: }
48:
49: public void returnBuffer(XByteBuffer buffer) {
50: if ((size.get() + buffer.getCapacity()) <= maxSize) {
51: size.addAndGet(buffer.getCapacity());
52: queue.offer(buffer);
53: }
54: }
55:
56: public void clear() {
57: queue.clear();
58: size.set(0);
59: }
60:
61: public int getMaxSize() {
62: return maxSize;
63: }
64:
65: }
|