001/**
002 * Licensed to the Apache Software Foundation (ASF) under one or more
003 * contributor license agreements.  See the NOTICE file distributed with
004 * this work for additional information regarding copyright ownership.
005 * The ASF licenses this file to You under the Apache License, Version 2.0
006 * (the "License"); you may not use this file except in compliance with
007 * the License.  You may obtain a copy of the License at
008 *
009 *      http://www.apache.org/licenses/LICENSE-2.0
010 *
011 * Unless required by applicable law or agreed to in writing, software
012 * distributed under the License is distributed on an "AS IS" BASIS,
013 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014 * See the License for the specific language governing permissions and
015 * limitations under the License.
016 */
017package org.apache.activemq.transport.udp;
018
019import java.nio.ByteBuffer;
020import java.util.ArrayList;
021import java.util.List;
022
023/**
024 * A default implementation of {@link SimpleBufferPool} which keeps a pool of direct
025 * byte buffers.
026 */
027public class DefaultBufferPool extends SimpleBufferPool implements ByteBufferPool {
028
029    private List<ByteBuffer> buffers = new ArrayList<ByteBuffer>();
030    private Object lock = new Object();
031
032    public DefaultBufferPool() {
033        super(true);
034    }
035
036    public DefaultBufferPool(boolean useDirect) {
037        super(useDirect);
038    }
039
040    @Override
041    public synchronized ByteBuffer borrowBuffer() {
042        synchronized (lock) {
043            int size = buffers.size();
044            if (size > 0) {
045                return buffers.remove(size - 1);
046            }
047        }
048        return createBuffer();
049    }
050
051    @Override
052    public void returnBuffer(ByteBuffer buffer) {
053        synchronized (lock) {
054            buffers.add(buffer);
055        }
056    }
057
058    @Override
059    public void start() throws Exception {
060    }
061
062    @Override
063    public void stop() throws Exception {
064        synchronized (lock) {
065            /*
066             * for (Iterator iter = buffers.iterator(); iter.hasNext();) {
067             * ByteBuffer buffer = (ByteBuffer) iter.next(); }
068             */
069            buffers.clear();
070        }
071    }
072}