org.mentaqueue.ConcurrentLinkedQueue Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of menta-queue Show documentation
A super fast inter-thread transfer queue.
There is a newer version: 1.3.1
/*
 * MentaQueue => http://mentaqueue.soliveirajr.com
 * Copyright (C) 2012  Sergio Oliveira Jr. ([email protected])
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */
package org.mentaqueue;

import org.mentaqueue.util.Builder;

public class ConcurrentLinkedQueue implements Queue {

	private final static int DEFAULT_CAPACITY = 1026 * 16;

	private final java.util.concurrent.ConcurrentLinkedQueue pool;
	private final java.util.concurrent.ConcurrentLinkedQueue queue;
	private final Builder builder;
	private E returnedElem = null;

	public ConcurrentLinkedQueue(int capacity, Builder builder) {
		pool = new java.util.concurrent.ConcurrentLinkedQueue();
		queue = new java.util.concurrent.ConcurrentLinkedQueue();
		this.builder = builder;
		for (int i = 0; i < capacity; i++) {
			pool.offer(builder.newInstance());
		}
	}

	public ConcurrentLinkedQueue(Builder builder) {
		this(DEFAULT_CAPACITY, builder);
	}
	
	private final static  Builder createBuilder(final Class klass) {
		return new Builder() {
			@Override
            public E newInstance() {
	            try {
	            	return klass.newInstance();
	            } catch(Exception e) {
	            	throw new RuntimeException(e);
	            }
            }
		};
	}
	
	public ConcurrentLinkedQueue(Class klass) {
		this(createBuilder(klass));
	}
	
	public ConcurrentLinkedQueue(int capacity, Class klass) {
		this(capacity, createBuilder(klass));
	}


	@Override
	public E nextToOffer() {
		E e = pool.poll();
		if (e == null) {
			e = builder.newInstance();
		}
		return e;
	}

	@Override
	public void offer(E e) {
		queue.offer(e);
	}

	@Override
	public long available() {
		E e = queue.peek();
		return e != null ? 1 : 0;
	}

	@Override
	public E poll() {
		returnedElem = queue.poll();
		return returnedElem;
	}

	@Override
	public void done() {
		pool.offer(returnedElem);
		returnedElem = null;
	}
}