com.sun.grizzly.async.AsyncQueue Maven / Gradle / Ivy
/*
*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright 2007-2008 Sun Microsystems, Inc. All rights reserved.
*
* The contents of this file are subject to the terms of either the GNU
* General Public License Version 2 only ("GPL") or the Common Development
* and Distribution License("CDDL") (collectively, the "License"). You
* may not use this file except in compliance with the License. You can obtain
* a copy of the License at https://glassfish.dev.java.net/public/CDDL+GPL.html
* or glassfish/bootstrap/legal/LICENSE.txt. See the License for the specific
* language governing permissions and limitations under the License.
*
* When distributing the software, include this License Header Notice in each
* file and include the License file at glassfish/bootstrap/legal/LICENSE.txt.
* Sun designates this particular file as subject to the "Classpath" exception
* as provided by Sun in the GPL Version 2 section of the License file that
* accompanied this code. If applicable, add the following below the License
* Header, with the fields enclosed by brackets [] replaced by your own
* identifying information: "Portions Copyrighted [year]
* [name of copyright owner]"
*
* Contributor(s):
*
* If you wish your version of this file to be governed by only the CDDL or
* only the GPL Version 2, indicate your decision by adding "[Contributor]
* elects to include this software in this distribution under the [CDDL or GPL
* Version 2] license." If you don't indicate a single choice of license, a
* recipient has the option to distribute your version of this file under
* either the CDDL, the GPL Version 2 or to extend the choice of license to
* its licensees as provided above. However, if you add GPL Version 2 code
* and therefore, elected the GPL Version 2 license, then the option applies
* only if the new code is made subject to such option by the copyright
* holder.
*
*/
package com.sun.grizzly.async;
import com.sun.grizzly.util.LinkedTransferQueue;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicReference;
import java.util.concurrent.locks.ReentrantLock;
/**
* Class implements {@link Map}-like collection, maps keys to values, where
* single key could have queue of correspondent values.
*
* @author Alexey Stashok
*/
public class AsyncQueue {
private Map queueMap =
new ConcurrentHashMap();
/**
* Add data to the {@link AsyncQueue}, corresponding to the given
* E
key
*
* @param key E
* @param queueRecord data unit
*/
public void offer(K key, E queueRecord) {
AsyncQueueEntry entry = obtainAsyncQueueEntry(key);
entry.queue.offer(queueRecord);
}
/**
* Get head element of E
key related queue.
* Element will not be removed from queue.
*
* @param key K
*
* @return E
data unit
*/
public E peek(K key) {
AsyncQueueEntry entry = queueMap.get(key);
if (entry != null) {
return entry.queue.peek();
}
return null;
}
/**
* Get head element of K
key related queue.
* Element will be removed from queue.
*
* @param key K
*
* @return E
data unit
*/
public E poll(K key) {
AsyncQueueEntry entry = queueMap.get(key);
if (entry != null) {
return entry.queue.poll();
}
return null;
}
/**
* Remove head element of K
key related queue.
*
* @param key K
* @return removed entry
*/
public AsyncQueueEntry removeEntry(K key) {
return queueMap.remove(key);
}
/**
* Get the size of K
key related queue.
*
* @param key K
* @return size of K
key related queue.
*/
public int size(K key) {
AsyncQueueEntry entry = queueMap.get(key);
return entry == null ? 0 : entry.queue.size();
}
/**
* Checks if K
key related queue is empty.
*
* @param key K
* @return true, if K
key related queue is empty, false otherwise
*/
public boolean isEmpty(K key) {
AsyncQueueEntry entry = queueMap.get(key);
return entry == null || entry.queue.isEmpty();
}
public void clear() {
queueMap.clear();
}
protected AsyncQueueEntry obtainAsyncQueueEntry(K key) {
AsyncQueueEntry entry = queueMap.get(key);
if (entry == null) {
synchronized(key) {
entry = queueMap.get(key);
if (entry == null) {
entry = new AsyncQueueEntry();
queueMap.put(key, entry);
}
}
}
return entry;
}
protected AsyncQueueEntry getAsyncQueueEntry(K key) {
return queueMap.get(key);
}
/**
* {@link AsyncQueue} data unit
*/
public class AsyncQueueEntry {
public LinkedTransferQueue queue;
public AtomicReference currentElement;
public ReentrantLock queuedActionLock;
// Amound of data, processed by the key
public AtomicInteger processedDataSize;
// Number of queue elements processed
public AtomicInteger processedElementsCount;
// Total number of elements were requested to be processed
public AtomicInteger totalElementsCount;
// Number of elements passed throw the async queue (not processed directly)
public AtomicInteger queuedElementsCount;
protected OperationResult tmpResult;
public AsyncQueueEntry() {
queue = new LinkedTransferQueue();
currentElement = new AtomicReference();
queuedActionLock = new ReentrantLock();
processedDataSize = new AtomicInteger();
processedElementsCount = new AtomicInteger();
totalElementsCount = new AtomicInteger();
queuedElementsCount = new AtomicInteger();
tmpResult = new OperationResult();
}
}
}