Rename the old LinkedTransferQueue to LegacyLinkedTransferQueue and fork

the "current" implementation of LinkedTransferQueue (from Doug Lea).
Introduce a QueueFactory which will load the right implementation
depending on the JVM version. This will make sure that the one with the
best performance is choosen and the code also works with java 5. See
#102
This commit is contained in:
norman 2011-12-07 12:08:39 +01:00
parent 51f69877a6
commit 766525431d
19 changed files with 3046 additions and 1408 deletions

View File

@ -32,7 +32,7 @@ import org.jboss.netty.channel.ChannelPipeline;
import org.jboss.netty.channel.ChannelSink; import org.jboss.netty.channel.ChannelSink;
import org.jboss.netty.channel.DefaultChannelConfig; import org.jboss.netty.channel.DefaultChannelConfig;
import org.jboss.netty.channel.MessageEvent; import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.util.internal.LinkedTransferQueue; import org.jboss.netty.util.internal.QueueFactory;
import org.jboss.netty.util.internal.ThreadLocalBoolean; import org.jboss.netty.util.internal.ThreadLocalBoolean;
/** /**
@ -52,7 +52,7 @@ final class DefaultLocalChannel extends AbstractChannel implements LocalChannel
private final ChannelConfig config; private final ChannelConfig config;
private final ThreadLocalBoolean delivering = new ThreadLocalBoolean(); private final ThreadLocalBoolean delivering = new ThreadLocalBoolean();
final Queue<MessageEvent> writeBuffer = new LinkedTransferQueue<MessageEvent>(); final Queue<MessageEvent> writeBuffer = QueueFactory.createQueue(MessageEvent.class);
volatile DefaultLocalChannel pairedChannel; volatile DefaultLocalChannel pairedChannel;
volatile LocalAddress localAddress; volatile LocalAddress localAddress;

View File

@ -0,0 +1,158 @@
/*
* Copyright 2011 Red Hat, Inc.
*
* Red Hat licenses this file to you under the Apache License, version 2.0
* (the "License"); you may not use this file except in compliance with the
* License. You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*/
package org.jboss.netty.channel.socket.nio;
import java.util.Collection;
import java.util.Iterator;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.TimeUnit;
import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.util.internal.QueueFactory;
/**
*
*
*
*
* @author <a href="http://www.jboss.org/netty/">The Netty Project</a>
* @author <a href="http://www.murkycloud.com/">Norman Maurer</a>
*
*/
abstract class AbstractWriteRequestQueue implements BlockingQueue<MessageEvent>{
protected final BlockingQueue<MessageEvent> queue;
public AbstractWriteRequestQueue() {
this.queue = QueueFactory.createQueue(MessageEvent.class);
}
@Override
public MessageEvent remove() {
return queue.remove();
}
@Override
public MessageEvent element() {
return queue.element();
}
@Override
public MessageEvent peek() {
return queue.peek();
}
@Override
public int size() {
return queue.size();
}
@Override
public boolean isEmpty() {
return queue.isEmpty();
}
@Override
public Iterator<MessageEvent> iterator() {
return queue.iterator();
}
@Override
public Object[] toArray() {
return queue.toArray();
}
@Override
public <T> T[] toArray(T[] a) {
return queue.toArray(a);
}
@Override
public boolean containsAll(Collection<?> c) {
return queue.containsAll(c);
}
@Override
public boolean addAll(Collection<? extends MessageEvent> c) {
return queue.addAll(c);
}
@Override
public boolean removeAll(Collection<?> c) {
return queue.removeAll(c);
}
@Override
public boolean retainAll(Collection<?> c) {
return queue.retainAll(c);
}
@Override
public void clear() {
queue.clear();
}
@Override
public boolean add(MessageEvent e) {
return queue.add(e);
}
@Override
public void put(MessageEvent e) throws InterruptedException {
queue.put(e);
}
@Override
public boolean offer(MessageEvent e, long timeout, TimeUnit unit) throws InterruptedException {
return queue.offer(e, timeout, unit);
}
@Override
public MessageEvent take() throws InterruptedException {
return queue.take();
}
@Override
public MessageEvent poll(long timeout, TimeUnit unit) throws InterruptedException {
return queue.poll(timeout, unit);
}
@Override
public int remainingCapacity() {
return queue.remainingCapacity();
}
@Override
public boolean remove(Object o) {
return queue.remove(o);
}
@Override
public boolean contains(Object o) {
return queue.contains(o);
}
@Override
public int drainTo(Collection<? super MessageEvent> c) {
return queue.drainTo(c);
}
@Override
public int drainTo(Collection<? super MessageEvent> c, int maxElements) {
return queue.drainTo(c, maxElements);
}
}

View File

@ -43,7 +43,7 @@ import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.logging.InternalLogger; import org.jboss.netty.logging.InternalLogger;
import org.jboss.netty.logging.InternalLoggerFactory; import org.jboss.netty.logging.InternalLoggerFactory;
import org.jboss.netty.util.internal.DeadLockProofWorker; import org.jboss.netty.util.internal.DeadLockProofWorker;
import org.jboss.netty.util.internal.LinkedTransferQueue; import org.jboss.netty.util.internal.QueueFactory;
/** /**
* *
@ -183,7 +183,7 @@ class NioClientSocketPipelineSink extends AbstractChannelSink {
private boolean started; private boolean started;
private final AtomicBoolean wakenUp = new AtomicBoolean(); private final AtomicBoolean wakenUp = new AtomicBoolean();
private final Object startStopLock = new Object(); private final Object startStopLock = new Object();
private final Queue<Runnable> registerTaskQueue = new LinkedTransferQueue<Runnable>(); private final Queue<Runnable> registerTaskQueue = QueueFactory.createQueue(Runnable.class);
Boss() { Boss() {
} }

View File

@ -38,7 +38,7 @@ import org.jboss.netty.channel.ChannelSink;
import org.jboss.netty.channel.MessageEvent; import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.channel.socket.DatagramChannelConfig; import org.jboss.netty.channel.socket.DatagramChannelConfig;
import org.jboss.netty.channel.socket.nio.SocketSendBufferPool.SendBuffer; import org.jboss.netty.channel.socket.nio.SocketSendBufferPool.SendBuffer;
import org.jboss.netty.util.internal.LinkedTransferQueue; import org.jboss.netty.util.internal.LegacyLinkedTransferQueue;
import org.jboss.netty.util.internal.ThreadLocalBoolean; import org.jboss.netty.util.internal.ThreadLocalBoolean;
/** /**
@ -247,26 +247,22 @@ class NioDatagramChannel extends AbstractChannel
} }
/** /**
* {@link WriteRequestQueue} is an extension of {@link LinkedTransferQueue} * {@link WriteRequestQueue} is an extension of {@link AbstractWriteRequestQueue}
* that adds support for highWaterMark checking of the write buffer size. * that adds support for highWaterMark checking of the write buffer size.
*/ */
private final class WriteRequestQueue extends private final class WriteRequestQueue extends
LinkedTransferQueue<MessageEvent> { AbstractWriteRequestQueue {
private static final long serialVersionUID = 5057413071460766376L;
private final ThreadLocalBoolean notifying = new ThreadLocalBoolean(); private final ThreadLocalBoolean notifying = new ThreadLocalBoolean();
WriteRequestQueue() {
}
/** /**
* This method first delegates to {@link LinkedTransferQueue#offer(Object)} and * This method first delegates to {@link LegacyLinkedTransferQueue#offer(Object)} and
* adds support for keeping track of the size of the this write buffer. * adds support for keeping track of the size of the this write buffer.
*/ */
@Override @Override
public boolean offer(MessageEvent e) { public boolean offer(MessageEvent e) {
boolean success = super.offer(e); boolean success = queue.offer(e);
assert success; assert success;
int messageSize = getMessageSize(e); int messageSize = getMessageSize(e);
@ -287,12 +283,12 @@ class NioDatagramChannel extends AbstractChannel
} }
/** /**
* This method first delegates to {@link LinkedTransferQueue#poll()} and * This method first delegates to {@link LegacyLinkedTransferQueue#poll()} and
* adds support for keeping track of the size of the this writebuffers queue. * adds support for keeping track of the size of the this writebuffers queue.
*/ */
@Override @Override
public MessageEvent poll() { public MessageEvent poll() {
MessageEvent e = super.poll(); MessageEvent e = queue.poll();
if (e != null) { if (e != null) {
int messageSize = getMessageSize(e); int messageSize = getMessageSize(e);
int newWriteBufferSize = writeBufferSize.addAndGet(-messageSize); int newWriteBufferSize = writeBufferSize.addAndGet(-messageSize);

View File

@ -45,7 +45,7 @@ import org.jboss.netty.channel.ReceiveBufferSizePredictor;
import org.jboss.netty.channel.socket.nio.SocketSendBufferPool.SendBuffer; import org.jboss.netty.channel.socket.nio.SocketSendBufferPool.SendBuffer;
import org.jboss.netty.logging.InternalLogger; import org.jboss.netty.logging.InternalLogger;
import org.jboss.netty.logging.InternalLoggerFactory; import org.jboss.netty.logging.InternalLoggerFactory;
import org.jboss.netty.util.internal.LinkedTransferQueue; import org.jboss.netty.util.internal.QueueFactory;
/** /**
* A class responsible for registering channels with {@link Selector}. * A class responsible for registering channels with {@link Selector}.
@ -105,12 +105,12 @@ class NioDatagramWorker implements Runnable {
/** /**
* Queue of {@link ChannelRegistionTask}s * Queue of {@link ChannelRegistionTask}s
*/ */
private final Queue<Runnable> registerTaskQueue = new LinkedTransferQueue<Runnable>(); private final Queue<Runnable> registerTaskQueue = QueueFactory.createQueue(Runnable.class);
/** /**
* Queue of WriteTasks * Queue of WriteTasks
*/ */
private final Queue<Runnable> writeTaskQueue = new LinkedTransferQueue<Runnable>(); private final Queue<Runnable> writeTaskQueue = QueueFactory.createQueue(Runnable.class);
private volatile int cancelledKeys; // should use AtomicInteger but we just need approximation private volatile int cancelledKeys; // should use AtomicInteger but we just need approximation

View File

@ -33,7 +33,6 @@ import org.jboss.netty.channel.ChannelPipeline;
import org.jboss.netty.channel.ChannelSink; import org.jboss.netty.channel.ChannelSink;
import org.jboss.netty.channel.MessageEvent; import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.channel.socket.nio.SocketSendBufferPool.SendBuffer; import org.jboss.netty.channel.socket.nio.SocketSendBufferPool.SendBuffer;
import org.jboss.netty.util.internal.LinkedTransferQueue;
import org.jboss.netty.util.internal.ThreadLocalBoolean; import org.jboss.netty.util.internal.ThreadLocalBoolean;
/** /**
@ -196,9 +195,7 @@ class NioSocketChannel extends AbstractChannel
} }
} }
private final class WriteRequestQueue extends LinkedTransferQueue<MessageEvent> { private final class WriteRequestQueue extends AbstractWriteRequestQueue {
private static final long serialVersionUID = -246694024103520626L;
private final ThreadLocalBoolean notifying = new ThreadLocalBoolean(); private final ThreadLocalBoolean notifying = new ThreadLocalBoolean();
@ -207,7 +204,7 @@ class NioSocketChannel extends AbstractChannel
@Override @Override
public boolean offer(MessageEvent e) { public boolean offer(MessageEvent e) {
boolean success = super.offer(e); boolean success = queue.offer(e);
assert success; assert success;
int messageSize = getMessageSize(e); int messageSize = getMessageSize(e);
@ -229,7 +226,7 @@ class NioSocketChannel extends AbstractChannel
@Override @Override
public MessageEvent poll() { public MessageEvent poll() {
MessageEvent e = super.poll(); MessageEvent e = queue.poll();
if (e != null) { if (e != null) {
int messageSize = getMessageSize(e); int messageSize = getMessageSize(e);
int newWriteBufferSize = writeBufferSize.addAndGet(-messageSize); int newWriteBufferSize = writeBufferSize.addAndGet(-messageSize);

View File

@ -47,7 +47,7 @@ import org.jboss.netty.channel.socket.nio.SocketSendBufferPool.SendBuffer;
import org.jboss.netty.logging.InternalLogger; import org.jboss.netty.logging.InternalLogger;
import org.jboss.netty.logging.InternalLoggerFactory; import org.jboss.netty.logging.InternalLoggerFactory;
import org.jboss.netty.util.internal.DeadLockProofWorker; import org.jboss.netty.util.internal.DeadLockProofWorker;
import org.jboss.netty.util.internal.LinkedTransferQueue; import org.jboss.netty.util.internal.QueueFactory;
/** /**
* *
@ -70,8 +70,8 @@ class NioWorker implements Runnable {
private final AtomicBoolean wakenUp = new AtomicBoolean(); private final AtomicBoolean wakenUp = new AtomicBoolean();
private final ReadWriteLock selectorGuard = new ReentrantReadWriteLock(); private final ReadWriteLock selectorGuard = new ReentrantReadWriteLock();
private final Object startStopLock = new Object(); private final Object startStopLock = new Object();
private final Queue<Runnable> registerTaskQueue = new LinkedTransferQueue<Runnable>(); private final Queue<Runnable> registerTaskQueue = QueueFactory.createQueue(Runnable.class);
private final Queue<Runnable> writeTaskQueue = new LinkedTransferQueue<Runnable>(); private final Queue<Runnable> writeTaskQueue = QueueFactory.createQueue(Runnable.class);
private volatile int cancelledKeys; // should use AtomicInteger but we just need approximation private volatile int cancelledKeys; // should use AtomicInteger but we just need approximation
private final SocketReceiveBufferPool recvBufferPool = new SocketReceiveBufferPool(); private final SocketReceiveBufferPool recvBufferPool = new SocketReceiveBufferPool();

View File

@ -23,7 +23,7 @@ import org.jboss.netty.channel.ChannelDownstreamHandler;
import org.jboss.netty.channel.ChannelEvent; import org.jboss.netty.channel.ChannelEvent;
import org.jboss.netty.channel.ChannelHandlerContext; import org.jboss.netty.channel.ChannelHandlerContext;
import org.jboss.netty.channel.ChannelUpstreamHandler; import org.jboss.netty.channel.ChannelUpstreamHandler;
import org.jboss.netty.util.internal.LinkedTransferQueue; import org.jboss.netty.util.internal.QueueFactory;
/** /**
* A combination of {@link HttpRequestEncoder} and {@link HttpResponseDecoder} * A combination of {@link HttpRequestEncoder} and {@link HttpResponseDecoder}
@ -46,7 +46,7 @@ public class HttpClientCodec implements ChannelUpstreamHandler,
ChannelDownstreamHandler { ChannelDownstreamHandler {
/** A queue that is used for correlating a request and a response. */ /** A queue that is used for correlating a request and a response. */
final Queue<HttpMethod> queue = new LinkedTransferQueue<HttpMethod>(); final Queue<HttpMethod> queue = QueueFactory.createQueue(HttpMethod.class);
/** If true, decoding stops (i.e. pass-through) */ /** If true, decoding stops (i.e. pass-through) */
volatile boolean done; volatile boolean done;

View File

@ -24,7 +24,7 @@ import org.jboss.netty.channel.Channels;
import org.jboss.netty.channel.MessageEvent; import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.channel.SimpleChannelHandler; import org.jboss.netty.channel.SimpleChannelHandler;
import org.jboss.netty.handler.codec.embedder.EncoderEmbedder; import org.jboss.netty.handler.codec.embedder.EncoderEmbedder;
import org.jboss.netty.util.internal.LinkedTransferQueue; import org.jboss.netty.util.internal.QueueFactory;
/** /**
* Encodes the content of the outbound {@link HttpResponse} and {@link HttpChunk}. * Encodes the content of the outbound {@link HttpResponse} and {@link HttpChunk}.
@ -53,7 +53,7 @@ import org.jboss.netty.util.internal.LinkedTransferQueue;
*/ */
public abstract class HttpContentEncoder extends SimpleChannelHandler { public abstract class HttpContentEncoder extends SimpleChannelHandler {
private final Queue<String> acceptEncodingQueue = new LinkedTransferQueue<String>(); private final Queue<String> acceptEncodingQueue = QueueFactory.createQueue(String.class);
private volatile EncoderEmbedder<ChannelBuffer> encoder; private volatile EncoderEmbedder<ChannelBuffer> encoder;
/** /**

View File

@ -36,7 +36,7 @@ import org.jboss.netty.channel.WriteCompletionEvent;
import org.jboss.netty.util.DefaultObjectSizeEstimator; import org.jboss.netty.util.DefaultObjectSizeEstimator;
import org.jboss.netty.util.ObjectSizeEstimator; import org.jboss.netty.util.ObjectSizeEstimator;
import org.jboss.netty.util.internal.ConcurrentIdentityHashMap; import org.jboss.netty.util.internal.ConcurrentIdentityHashMap;
import org.jboss.netty.util.internal.LinkedTransferQueue; import org.jboss.netty.util.internal.QueueFactory;
import org.jboss.netty.util.internal.SharedResourceMisuseDetector; import org.jboss.netty.util.internal.SharedResourceMisuseDetector;
/** /**
@ -212,7 +212,7 @@ public class MemoryAwareThreadPoolExecutor extends ThreadPoolExecutor {
ThreadFactory threadFactory) { ThreadFactory threadFactory) {
super(corePoolSize, corePoolSize, keepAliveTime, unit, super(corePoolSize, corePoolSize, keepAliveTime, unit,
new LinkedTransferQueue<Runnable>(), threadFactory, new NewThreadRunsPolicy()); QueueFactory.createQueue(Runnable.class), threadFactory, new NewThreadRunsPolicy());
if (objectSizeEstimator == null) { if (objectSizeEstimator == null) {
throw new NullPointerException("objectSizeEstimator"); throw new NullPointerException("objectSizeEstimator");

View File

@ -31,7 +31,7 @@ import org.jboss.netty.channel.ChannelState;
import org.jboss.netty.channel.ChannelStateEvent; import org.jboss.netty.channel.ChannelStateEvent;
import org.jboss.netty.util.ObjectSizeEstimator; import org.jboss.netty.util.ObjectSizeEstimator;
import org.jboss.netty.util.internal.ConcurrentIdentityWeakKeyHashMap; import org.jboss.netty.util.internal.ConcurrentIdentityWeakKeyHashMap;
import org.jboss.netty.util.internal.LinkedTransferQueue; import org.jboss.netty.util.internal.QueueFactory;
/** /**
* A {@link MemoryAwareThreadPoolExecutor} which makes sure the events from the * A {@link MemoryAwareThreadPoolExecutor} which makes sure the events from the
@ -284,7 +284,7 @@ public class OrderedMemoryAwareThreadPoolExecutor extends
} }
private final class ChildExecutor implements Executor, Runnable { private final class ChildExecutor implements Executor, Runnable {
private final Queue<Runnable> tasks = new LinkedTransferQueue<Runnable>(); private final Queue<Runnable> tasks = QueueFactory.createQueue(Runnable.class);
private final AtomicBoolean isRunning = new AtomicBoolean(false); private final AtomicBoolean isRunning = new AtomicBoolean(false);
ChildExecutor() { ChildExecutor() {

View File

@ -29,7 +29,7 @@ import org.jboss.netty.channel.ExceptionEvent;
import org.jboss.netty.channel.MessageEvent; import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.channel.SimpleChannelUpstreamHandler; import org.jboss.netty.channel.SimpleChannelUpstreamHandler;
import org.jboss.netty.util.internal.DeadLockProofWorker; import org.jboss.netty.util.internal.DeadLockProofWorker;
import org.jboss.netty.util.internal.LinkedTransferQueue; import org.jboss.netty.util.internal.QueueFactory;
/** /**
* Emulates blocking read operation. This handler stores all received messages * Emulates blocking read operation. This handler stores all received messages
@ -84,7 +84,7 @@ public class BlockingReadHandler<E> extends SimpleChannelUpstreamHandler {
* implementation. * implementation.
*/ */
public BlockingReadHandler() { public BlockingReadHandler() {
this(new LinkedTransferQueue<ChannelEvent>()); this(QueueFactory.createQueue(ChannelEvent.class));
} }
/** /**

View File

@ -33,7 +33,7 @@ import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.channel.SimpleChannelHandler; import org.jboss.netty.channel.SimpleChannelHandler;
import org.jboss.netty.channel.socket.nio.NioSocketChannelConfig; import org.jboss.netty.channel.socket.nio.NioSocketChannelConfig;
import org.jboss.netty.util.HashedWheelTimer; import org.jboss.netty.util.HashedWheelTimer;
import org.jboss.netty.util.internal.LinkedTransferQueue; import org.jboss.netty.util.internal.QueueFactory;
/** /**
* Emulates buffered write operation. This handler stores all write requests * Emulates buffered write operation. This handler stores all write requests
@ -193,7 +193,7 @@ public class BufferedWriteHandler extends SimpleChannelHandler {
* into a single write request on {@link #flush()} * into a single write request on {@link #flush()}
*/ */
public BufferedWriteHandler(boolean consolidateOnFlush) { public BufferedWriteHandler(boolean consolidateOnFlush) {
this(new LinkedTransferQueue<MessageEvent>(), consolidateOnFlush); this(QueueFactory.createQueue(MessageEvent.class), consolidateOnFlush);
} }
/** /**

View File

@ -50,8 +50,8 @@ import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.handler.codec.frame.FrameDecoder; import org.jboss.netty.handler.codec.frame.FrameDecoder;
import org.jboss.netty.logging.InternalLogger; import org.jboss.netty.logging.InternalLogger;
import org.jboss.netty.logging.InternalLoggerFactory; import org.jboss.netty.logging.InternalLoggerFactory;
import org.jboss.netty.util.internal.LinkedTransferQueue;
import org.jboss.netty.util.internal.NonReentrantLock; import org.jboss.netty.util.internal.NonReentrantLock;
import org.jboss.netty.util.internal.QueueFactory;
/** /**
* Adds <a href="http://en.wikipedia.org/wiki/Transport_Layer_Security">SSL * Adds <a href="http://en.wikipedia.org/wiki/Transport_Layer_Security">SSL
@ -196,7 +196,7 @@ public class SslHandler extends FrameDecoder
int ignoreClosedChannelException; int ignoreClosedChannelException;
final Object ignoreClosedChannelExceptionLock = new Object(); final Object ignoreClosedChannelExceptionLock = new Object();
private final Queue<PendingWrite> pendingUnencryptedWrites = new LinkedList<PendingWrite>(); private final Queue<PendingWrite> pendingUnencryptedWrites = new LinkedList<PendingWrite>();
private final Queue<MessageEvent> pendingEncryptedWrites = new LinkedTransferQueue<MessageEvent>(); private final Queue<MessageEvent> pendingEncryptedWrites = QueueFactory.createQueue(MessageEvent.class);
private final NonReentrantLock pendingEncryptedWritesLock = new NonReentrantLock(); private final NonReentrantLock pendingEncryptedWritesLock = new NonReentrantLock();
private volatile boolean issueHandshake = false; private volatile boolean issueHandshake = false;

View File

@ -35,7 +35,7 @@ import org.jboss.netty.channel.Channels;
import org.jboss.netty.channel.MessageEvent; import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.logging.InternalLogger; import org.jboss.netty.logging.InternalLogger;
import org.jboss.netty.logging.InternalLoggerFactory; import org.jboss.netty.logging.InternalLoggerFactory;
import org.jboss.netty.util.internal.LinkedTransferQueue; import org.jboss.netty.util.internal.QueueFactory;
/** /**
* A {@link ChannelHandler} that adds support for writing a large data stream * A {@link ChannelHandler} that adds support for writing a large data stream
@ -79,8 +79,7 @@ public class ChunkedWriteHandler implements ChannelUpstreamHandler, ChannelDowns
private static final InternalLogger logger = private static final InternalLogger logger =
InternalLoggerFactory.getInstance(ChunkedWriteHandler.class); InternalLoggerFactory.getInstance(ChunkedWriteHandler.class);
private final Queue<MessageEvent> queue = private final Queue<MessageEvent> queue = QueueFactory.createQueue(MessageEvent.class);
new LinkedTransferQueue<MessageEvent>();
private ChannelHandlerContext ctx; private ChannelHandlerContext ctx;
private MessageEvent currentEvent; private MessageEvent currentEvent;

View File

@ -0,0 +1,48 @@
/*
* Copyright 2011 Red Hat, Inc.
*
* Red Hat licenses this file to you under the Apache License, version 2.0
* (the "License"); you may not use this file except in compliance with the
* License. You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*/
package org.jboss.netty.util;
/**
* Utility which checks if {@value #UNSAFE} class can be found in the classpath
*
*
*
* @author <a href="http://www.jboss.org/netty/">The Netty Project</a>
* @author <a href="http://www.murkycloud.com/">Norman Maurer</a>
*
*/
public class UnsafeDetectUtil {
private static final String UNSAFE = "sun.misc.Unsafe";
public static boolean isUnsafeFound(ClassLoader loader) {
try {
Class.forName(UNSAFE, true, loader);
return true;
} catch (ClassNotFoundException e) {
return false;
}
}
public static boolean isUnsafeFound() {
return isUnsafeFound(UnsafeDetectUtil.class.getClassLoader());
}
private UnsafeDetectUtil() {
// only static method supported
}
}

File diff suppressed because it is too large Load Diff

View File

@ -1,38 +1,32 @@
/*
* Copyright 2009 Red Hat, Inc.
*
* Red Hat licenses this file to you under the Apache License, version 2.0
* (the "License"); you may not use this file except in compliance with the
* License. You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*/
/* /*
* Written by Doug Lea with assistance from members of JCP JSR-166 * Written by Doug Lea with assistance from members of JCP JSR-166
* Expert Group and released to the public domain, as explained at * Expert Group and released to the public domain, as explained at
* http://creativecommons.org/licenses/publicdomain * http://creativecommons.org/publicdomain/zero/1.0/
*/ */
package org.jboss.netty.util.internal; package org.jboss.netty.util.internal;
import java.util.AbstractQueue; import java.util.AbstractQueue;
import java.util.Collection; import java.util.Collection;
import java.util.ConcurrentModificationException;
import java.util.Iterator; import java.util.Iterator;
import java.util.NoSuchElementException; import java.util.NoSuchElementException;
import java.util.Queue;
import java.util.concurrent.BlockingQueue; import java.util.concurrent.BlockingQueue;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicIntegerFieldUpdater;
import java.util.concurrent.atomic.AtomicReferenceFieldUpdater;
import java.util.concurrent.locks.LockSupport; import java.util.concurrent.locks.LockSupport;
/** /**
* This class is a copied from <a href="http://gee.cs.oswego.edu/cgi-bin/viewcvs.cgi/jsr166/src/jsr166y/LinkedTransferQueue.java"> URL revision 1.91 </a>
* <br>
* The only difference is that it replace {@link BlockingQueue} and any reference to the TransferQueue interface was removed
* <br>
*
* <strong>
* Please use {@link QueueFactory} to create a Queue as it will use the "optimal" implementation depending on the JVM
* </strong>
* <br>
* <br>
*
* An unbounded {@link BlockingQueue} based on linked nodes. * An unbounded {@link BlockingQueue} based on linked nodes.
* This queue orders elements FIFO (first-in-first-out) with respect * This queue orders elements FIFO (first-in-first-out) with respect
* to any given producer. The <em>head</em> of the queue is that * to any given producer. The <em>head</em> of the queue is that
@ -40,10 +34,17 @@ import java.util.concurrent.locks.LockSupport;
* producer. The <em>tail</em> of the queue is that element that has * producer. The <em>tail</em> of the queue is that element that has
* been on the queue the shortest time for some producer. * been on the queue the shortest time for some producer.
* *
* <p>Beware that, unlike in most collections, the {@code size} * <p>Beware that, unlike in most collections, the {@code size} method
* method is <em>NOT</em> a constant-time operation. Because of the * is <em>NOT</em> a constant-time operation. Because of the
* asynchronous nature of these queues, determining the current number * asynchronous nature of these queues, determining the current number
* of elements requires a traversal of the elements. * of elements requires a traversal of the elements, and so may report
* inaccurate results if this collection is modified during traversal.
* Additionally, the bulk operations {@code addAll},
* {@code removeAll}, {@code retainAll}, {@code containsAll},
* {@code equals}, and {@code toArray} are <em>not</em> guaranteed
* to be performed atomically. For example, an iterator operating
* concurrently with an {@code addAll} operation might view only some
* of the added elements.
* *
* <p>This class and its iterator implement all of the * <p>This class and its iterator implement all of the
* <em>optional</em> methods of the {@link Collection} and {@link * <em>optional</em> methods of the {@link Collection} and {@link
@ -60,10 +61,8 @@ import java.util.concurrent.locks.LockSupport;
* <a href="{@docRoot}/../technotes/guides/collections/index.html"> * <a href="{@docRoot}/../technotes/guides/collections/index.html">
* Java Collections Framework</a>. * Java Collections Framework</a>.
* *
* @author <a href="http://www.jboss.org/netty/">The Netty Project</a> * @since 1.7
* @author Doug Lea * @author Doug Lea
* @author <a href="http://gleamynode.net/">Trustin Lee</a>
*
* @param <E> the type of elements held in this collection * @param <E> the type of elements held in this collection
*/ */
public class LinkedTransferQueue<E> extends AbstractQueue<E> public class LinkedTransferQueue<E> extends AbstractQueue<E>
@ -314,8 +313,8 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
* of less-contended queues. During spins threads check their * of less-contended queues. During spins threads check their
* interrupt status and generate a thread-local random number * interrupt status and generate a thread-local random number
* to decide to occasionally perform a Thread.yield. While * to decide to occasionally perform a Thread.yield. While
* yield has underdefined specs, we assume that might it help, * yield has underdefined specs, we assume that it might help,
* and will not hurt in limiting impact of spinning on busy * and will not hurt, in limiting impact of spinning on busy
* systems. We also use smaller (1/2) spins for nodes that are * systems. We also use smaller (1/2) spins for nodes that are
* not known to be front but whose predecessors have not * not known to be front but whose predecessors have not
* blocked -- these "chained" spins avoid artifacts of * blocked -- these "chained" spins avoid artifacts of
@ -436,35 +435,13 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
volatile Thread waiter; // null until waiting volatile Thread waiter; // null until waiting
// CAS methods for fields // CAS methods for fields
boolean casNext(Node cmp, Node val) { final boolean casNext(Node cmp, Node val) {
if (AtomicFieldUpdaterUtil.isAvailable()) { return UNSAFE.compareAndSwapObject(this, nextOffset, cmp, val);
return nextUpdater.compareAndSet(this, cmp, val);
} else {
synchronized (this) {
if (next == cmp) {
next = val;
return true;
} else {
return false;
}
}
}
} }
boolean casItem(Object cmp, Object val) { final boolean casItem(Object cmp, Object val) {
// assert cmp == null || cmp.getClass() != Node.class; // assert cmp == null || cmp.getClass() != Node.class;
if (AtomicFieldUpdaterUtil.isAvailable()) { return UNSAFE.compareAndSwapObject(this, itemOffset, cmp, val);
return itemUpdater.compareAndSet(this, cmp, val);
} else {
synchronized (this) {
if (item == cmp) {
item = val;
return true;
} else {
return false;
}
}
}
} }
/** /**
@ -472,7 +449,7 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
* only be seen after publication via casNext. * only be seen after publication via casNext.
*/ */
Node(Object item, boolean isData) { Node(Object item, boolean isData) {
this.item = item; UNSAFE.putObject(this, itemOffset, item); // relaxed write
this.isData = isData; this.isData = isData;
} }
@ -480,37 +457,37 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
* Links node to itself to avoid garbage retention. Called * Links node to itself to avoid garbage retention. Called
* only after CASing head field, so uses relaxed write. * only after CASing head field, so uses relaxed write.
*/ */
void forgetNext() { final void forgetNext() {
this.next = this; UNSAFE.putObject(this, nextOffset, this);
} }
/** /**
* Sets item to self and waiter to null, to avoid garbage * Sets item to self and waiter to null, to avoid garbage
* retention after matching or cancelling. Uses relaxed writes * retention after matching or cancelling. Uses relaxed writes
* bacause order is already constrained in the only calling * because order is already constrained in the only calling
* contexts: item is forgotten only after volatile/atomic * contexts: item is forgotten only after volatile/atomic
* mechanics that extract items. Similarly, clearing waiter * mechanics that extract items. Similarly, clearing waiter
* follows either CAS or return from park (if ever parked; * follows either CAS or return from park (if ever parked;
* else we don't care). * else we don't care).
*/ */
void forgetContents() { final void forgetContents() {
this.item = this; UNSAFE.putObject(this, itemOffset, this);
this.waiter = null; UNSAFE.putObject(this, waiterOffset, null);
} }
/** /**
* Returns true if this node has been matched, including the * Returns true if this node has been matched, including the
* case of artificial matches due to cancellation. * case of artificial matches due to cancellation.
*/ */
boolean isMatched() { final boolean isMatched() {
Object x = item; Object x = item;
return x == this || x == null == isData; return (x == this) || ((x == null) == isData);
} }
/** /**
* Returns true if this is an unmatched request node. * Returns true if this is an unmatched request node.
*/ */
boolean isUnmatchedRequest() { final boolean isUnmatchedRequest() {
return !isData && item == null; return !isData && item == null;
} }
@ -519,16 +496,16 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
* appended to this node because this node is unmatched and * appended to this node because this node is unmatched and
* has opposite data mode. * has opposite data mode.
*/ */
boolean cannotPrecede(boolean haveData) { final boolean cannotPrecede(boolean haveData) {
boolean d = isData; boolean d = isData;
Object x; Object x;
return d != haveData && (x = item) != this && x != null == d; return d != haveData && (x = item) != this && (x != null) == d;
} }
/** /**
* Tries to artificially match a data node -- used by remove. * Tries to artificially match a data node -- used by remove.
*/ */
boolean tryMatchData() { final boolean tryMatchData() {
// assert isData; // assert isData;
Object x = item; Object x = item;
if (x != null && x != this && casItem(x, null)) { if (x != null && x != this && casItem(x, null)) {
@ -538,66 +515,49 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
return false; return false;
} }
private static final AtomicReferenceFieldUpdater<Node, Node> nextUpdater = private static final long serialVersionUID = -3375979862319811754L;
AtomicFieldUpdaterUtil.newRefUpdater(Node.class, Node.class, "next");
private static final AtomicReferenceFieldUpdater<Node, Object> itemUpdater =
AtomicFieldUpdaterUtil.newRefUpdater(Node.class, Object.class, "item");
// Unsafe mechanics
private static final sun.misc.Unsafe UNSAFE;
private static final long itemOffset;
private static final long nextOffset;
private static final long waiterOffset;
static {
try {
UNSAFE = getUnsafe();
Class<?> k = Node.class;
itemOffset = UNSAFE.objectFieldOffset
(k.getDeclaredField("item"));
nextOffset = UNSAFE.objectFieldOffset
(k.getDeclaredField("next"));
waiterOffset = UNSAFE.objectFieldOffset
(k.getDeclaredField("waiter"));
} catch (Exception e) {
throw new Error(e);
}
}
} }
/** head of the queue; null until first enqueue */ /** head of the queue; null until first enqueue */
transient volatile Node head; transient volatile Node head;
/** tail of the queue; null until first append */ /** tail of the queue; null until first append */
transient volatile Node tail; private transient volatile Node tail;
/** The number of apparent failures to unsplice removed nodes */ /** The number of apparent failures to unsplice removed nodes */
transient volatile int sweepVotes; private transient volatile int sweepVotes;
// CAS methods for fields // CAS methods for fields
private boolean casTail(Node cmp, Node val) { private boolean casTail(Node cmp, Node val) {
if (AtomicFieldUpdaterUtil.isAvailable()) { return UNSAFE.compareAndSwapObject(this, tailOffset, cmp, val);
return tailUpdater.compareAndSet(this, cmp, val);
} else {
synchronized (this) {
if (tail == cmp) {
tail = val;
return true;
} else {
return false;
}
}
}
} }
private boolean casHead(Node cmp, Node val) { private boolean casHead(Node cmp, Node val) {
if (AtomicFieldUpdaterUtil.isAvailable()) { return UNSAFE.compareAndSwapObject(this, headOffset, cmp, val);
return headUpdater.compareAndSet(this, cmp, val);
} else {
synchronized (this) {
if (head == cmp) {
head = val;
return true;
} else {
return false;
}
}
}
} }
private boolean casSweepVotes(int cmp, int val) { private boolean casSweepVotes(int cmp, int val) {
if (AtomicFieldUpdaterUtil.isAvailable()) { return UNSAFE.compareAndSwapInt(this, sweepVotesOffset, cmp, val);
return sweepVotesUpdater.compareAndSet(this, cmp, val);
} else {
synchronized (this) {
if (sweepVotes == cmp) {
sweepVotes = val;
return true;
} else {
return false;
}
}
}
} }
/* /*
@ -625,51 +585,46 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
* @throws NullPointerException if haveData mode but e is null * @throws NullPointerException if haveData mode but e is null
*/ */
private E xfer(E e, boolean haveData, int how, long nanos) { private E xfer(E e, boolean haveData, int how, long nanos) {
if (haveData && e == null) { if (haveData && (e == null))
throw new NullPointerException(); throw new NullPointerException();
}
Node s = null; // the node to append, if needed Node s = null; // the node to append, if needed
retry: for (;;) { // restart on append race retry:
for (;;) { // restart on append race
for (Node h = head, p = h; p != null;) { // find & match first node for (Node h = head, p = h; p != null;) { // find & match first node
boolean isData = p.isData; boolean isData = p.isData;
Object item = p.item; Object item = p.item;
if (item != p && item != null == isData) { // unmatched if (item != p && (item != null) == isData) { // unmatched
if (isData == haveData) { // can't match if (isData == haveData) // can't match
break; break;
}
if (p.casItem(item, e)) { // match if (p.casItem(item, e)) { // match
for (Node q = p; q != h;) { for (Node q = p; q != h;) {
Node n = q.next; // update by 2 unless singleton Node n = q.next; // update by 2 unless singleton
if (head == h && casHead(h, n == null? q : n)) { if (head == h && casHead(h, n == null ? q : n)) {
h.forgetNext(); h.forgetNext();
break; break;
} // advance and retry } // advance and retry
if ((h = head) == null || if ((h = head) == null ||
(q = h.next) == null || !q.isMatched()) { (q = h.next) == null || !q.isMatched())
break; // unless slack < 2 break; // unless slack < 2
}
} }
LockSupport.unpark(p.waiter); LockSupport.unpark(p.waiter);
return LinkedTransferQueue.cast(item); return LinkedTransferQueue.<E>cast(item);
} }
} }
Node n = p.next; Node n = p.next;
p = p != n ? n : (h = head); // Use head if p offlist p = (p != n) ? n : (h = head); // Use head if p offlist
} }
if (how != NOW) { // No matches available if (how != NOW) { // No matches available
if (s == null) { if (s == null)
s = new Node(e, haveData); s = new Node(e, haveData);
}
Node pred = tryAppend(s, haveData); Node pred = tryAppend(s, haveData);
if (pred == null) { if (pred == null)
continue retry; // lost race vs opposite mode continue retry; // lost race vs opposite mode
} if (how != ASYNC)
if (how != ASYNC) {
return awaitMatch(s, pred, e, (how == TIMED), nanos); return awaitMatch(s, pred, e, (how == TIMED), nanos);
}
} }
return e; // not waiting return e; // not waiting
} }
@ -688,25 +643,22 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
for (Node t = tail, p = t;;) { // move p to last node and append for (Node t = tail, p = t;;) { // move p to last node and append
Node n, u; // temps for reads of next & tail Node n, u; // temps for reads of next & tail
if (p == null && (p = head) == null) { if (p == null && (p = head) == null) {
if (casHead(null, s)) { if (casHead(null, s))
return s; // initialize return s; // initialize
}
} }
else if (p.cannotPrecede(haveData)) { else if (p.cannotPrecede(haveData))
return null; // lost race vs opposite mode return null; // lost race vs opposite mode
} else if ((n = p.next) != null) { // not last; keep traversing else if ((n = p.next) != null) // not last; keep traversing
p = p != t && t != (u = tail) ? (t = u) : // stale tail p = p != t && t != (u = tail) ? (t = u) : // stale tail
p != n ? n : null; // restart if off list (p != n) ? n : null; // restart if off list
} else if (!p.casNext(null, s)) { else if (!p.casNext(null, s))
p = p.next; // re-read on CAS failure p = p.next; // re-read on CAS failure
} else { else {
if (p != t) { // update if slack now >= 2 if (p != t) { // update if slack now >= 2
while ((tail != t || !casTail(t, s)) && while ((tail != t || !casTail(t, s)) &&
(t = tail) != null && (t = tail) != null &&
(s = t.next) != null && // advance and retry (s = t.next) != null && // advance and retry
(s = s.next) != null && s != t) { (s = s.next) != null && s != t);
continue;
}
} }
return p; return p;
} }
@ -736,37 +688,34 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
if (item != e) { // matched if (item != e) { // matched
// assert item != s; // assert item != s;
s.forgetContents(); // avoid garbage s.forgetContents(); // avoid garbage
return LinkedTransferQueue.cast(item); return LinkedTransferQueue.<E>cast(item);
} }
if ((w.isInterrupted() || timed && nanos <= 0) && if ((w.isInterrupted() || (timed && nanos <= 0)) &&
s.casItem(e, s)) { // cancel s.casItem(e, s)) { // cancel
unsplice(pred, s); unsplice(pred, s);
return e; return e;
} }
if (spins < 0) { // establish spins at/near front if (spins < 0) { // establish spins at/near front
if ((spins = spinsFor(pred, s.isData)) > 0) { if ((spins = spinsFor(pred, s.isData)) > 0)
randomYields = ThreadLocalRandom.current(); randomYields = ThreadLocalRandom.current();
}
} }
else if (spins > 0) { // spin else if (spins > 0) { // spin
--spins; --spins;
if (randomYields.nextInt(CHAINED_SPINS) == 0) { if (randomYields.nextInt(CHAINED_SPINS) == 0)
Thread.yield(); // occasionally yield Thread.yield(); // occasionally yield
}
} }
else if (s.waiter == null) { else if (s.waiter == null) {
s.waiter = w; // request unpark then recheck s.waiter = w; // request unpark then recheck
} }
else if (timed) { else if (timed) {
long now = System.nanoTime(); long now = System.nanoTime();
if ((nanos -= now - lastTime) > 0) { if ((nanos -= now - lastTime) > 0)
LockSupport.parkNanos(nanos); LockSupport.parkNanos(this, nanos);
}
lastTime = now; lastTime = now;
} }
else { else {
LockSupport.park(); LockSupport.park(this);
} }
} }
} }
@ -777,15 +726,12 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
*/ */
private static int spinsFor(Node pred, boolean haveData) { private static int spinsFor(Node pred, boolean haveData) {
if (MP && pred != null) { if (MP && pred != null) {
if (pred.isData != haveData) { // phase change if (pred.isData != haveData) // phase change
return FRONT_SPINS + CHAINED_SPINS; return FRONT_SPINS + CHAINED_SPINS;
} if (pred.isMatched()) // probably at front
if (pred.isMatched()) { // probably at front
return FRONT_SPINS; return FRONT_SPINS;
} if (pred.waiter == null) // pred apparently spinning
if (pred.waiter == null) { // pred apparently spinning
return CHAINED_SPINS; return CHAINED_SPINS;
}
} }
return 0; return 0;
} }
@ -799,7 +745,7 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
*/ */
final Node succ(Node p) { final Node succ(Node p) {
Node next = p.next; Node next = p.next;
return p == next ? head : next; return (p == next) ? head : next;
} }
/** /**
@ -808,9 +754,8 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
*/ */
private Node firstOfMode(boolean isData) { private Node firstOfMode(boolean isData) {
for (Node p = head; p != null; p = succ(p)) { for (Node p = head; p != null; p = succ(p)) {
if (!p.isMatched()) { if (!p.isMatched())
return p.isData == isData ? p : null; return (p.isData == isData) ? p : null;
}
} }
return null; return null;
} }
@ -823,13 +768,11 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
for (Node p = head; p != null; p = succ(p)) { for (Node p = head; p != null; p = succ(p)) {
Object item = p.item; Object item = p.item;
if (p.isData) { if (p.isData) {
if (item != null && item != p) { if (item != null && item != p)
return LinkedTransferQueue.cast(item); return LinkedTransferQueue.<E>cast(item);
}
} }
else if (item == null) { else if (item == null)
return null; return null;
}
} }
return null; return null;
} }
@ -842,17 +785,15 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
int count = 0; int count = 0;
for (Node p = head; p != null; ) { for (Node p = head; p != null; ) {
if (!p.isMatched()) { if (!p.isMatched()) {
if (p.isData != data) { if (p.isData != data)
return 0; return 0;
} if (++count == Integer.MAX_VALUE) // saturated
if (++count == Integer.MAX_VALUE) { // saturated
break; break;
}
} }
Node n = p.next; Node n = p.next;
if (n != p) { if (n != p)
p = n; p = n;
} else { else {
count = 0; count = 0;
p = head; p = head;
} }
@ -870,54 +811,86 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
* Moves to next node after prev, or first node if prev null. * Moves to next node after prev, or first node if prev null.
*/ */
private void advance(Node prev) { private void advance(Node prev) {
lastPred = lastRet; /*
lastRet = prev; * To track and avoid buildup of deleted nodes in the face
for (Node p = prev == null ? head : succ(prev); * of calls to both Queue.remove and Itr.remove, we must
p != null; p = succ(p)) { * include variants of unsplice and sweep upon each
Object item = p.item; * advance: Upon Itr.remove, we may need to catch up links
if (p.isData) { * from lastPred, and upon other removes, we might need to
if (item != null && item != p) { * skip ahead from stale nodes and unsplice deleted ones
nextItem = LinkedTransferQueue.cast(item); * found while advancing.
nextNode = p; */
Node r, b; // reset lastPred upon possible deletion of lastRet
if ((r = lastRet) != null && !r.isMatched())
lastPred = r; // next lastPred is old lastRet
else if ((b = lastPred) == null || b.isMatched())
lastPred = null; // at start of list
else {
Node s, n; // help with removal of lastPred.next
while ((s = b.next) != null &&
s != b && s.isMatched() &&
(n = s.next) != null && n != s)
b.casNext(s, n);
}
this.lastRet = prev;
for (Node p = prev, s, n;;) {
s = (p == null) ? head : p.next;
if (s == null)
break;
else if (s == p) {
p = null;
continue;
}
Object item = s.item;
if (s.isData) {
if (item != null && item != s) {
nextItem = LinkedTransferQueue.<E>cast(item);
nextNode = s;
return; return;
} }
} }
else if (item == null) { else if (item == null)
break; break;
} // assert s.isMatched();
if (p == null)
p = s;
else if ((n = s.next) == null)
break;
else if (s == n)
p = null;
else
p.casNext(s, n);
} }
nextNode = null; nextNode = null;
nextItem = null;
} }
Itr() { Itr() {
advance(null); advance(null);
} }
@Override public final boolean hasNext() {
public boolean hasNext() {
return nextNode != null; return nextNode != null;
} }
@Override public final E next() {
public E next() {
Node p = nextNode; Node p = nextNode;
if (p == null) { if (p == null) throw new NoSuchElementException();
throw new NoSuchElementException();
}
E e = nextItem; E e = nextItem;
advance(p); advance(p);
return e; return e;
} }
@Override public final void remove() {
public void remove() { final Node lastRet = this.lastRet;
Node p = lastRet; if (lastRet == null)
if (p == null) {
throw new IllegalStateException(); throw new IllegalStateException();
} this.lastRet = null;
if (p.tryMatchData()) { if (lastRet.tryMatchData())
unsplice(lastPred, p); unsplice(lastPred, lastRet);
}
} }
} }
@ -943,30 +916,25 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
if (pred != null && pred != s && pred.next == s) { if (pred != null && pred != s && pred.next == s) {
Node n = s.next; Node n = s.next;
if (n == null || if (n == null ||
n != s && pred.casNext(s, n) && pred.isMatched()) { (n != s && pred.casNext(s, n) && pred.isMatched())) {
for (;;) { // check if at, or could be, head for (;;) { // check if at, or could be, head
Node h = head; Node h = head;
if (h == pred || h == s || h == null) { if (h == pred || h == s || h == null)
return; // at head or list empty return; // at head or list empty
} if (!h.isMatched())
if (!h.isMatched()) {
break; break;
}
Node hn = h.next; Node hn = h.next;
if (hn == null) { if (hn == null)
return; // now empty return; // now empty
} if (hn != h && casHead(h, hn))
if (hn != h && casHead(h, hn)) {
h.forgetNext(); // advance head h.forgetNext(); // advance head
}
} }
if (pred.next != pred && s.next != s) { // recheck if offlist if (pred.next != pred && s.next != s) { // recheck if offlist
for (;;) { // sweep now if enough votes for (;;) { // sweep now if enough votes
int v = sweepVotes; int v = sweepVotes;
if (v < SWEEP_THRESHOLD) { if (v < SWEEP_THRESHOLD) {
if (casSweepVotes(v, v + 1)) { if (casSweepVotes(v, v + 1))
break; break;
}
} }
else if (casSweepVotes(v, 0)) { else if (casSweepVotes(v, 0)) {
sweep(); sweep();
@ -984,17 +952,16 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
*/ */
private void sweep() { private void sweep() {
for (Node p = head, s, n; p != null && (s = p.next) != null; ) { for (Node p = head, s, n; p != null && (s = p.next) != null; ) {
if (!s.isMatched()) { if (!s.isMatched())
// Unmatched nodes are never self-linked // Unmatched nodes are never self-linked
p = s; p = s;
} else if ((n = s.next) == null) { // trailing node is pinned else if ((n = s.next) == null) // trailing node is pinned
break; break;
} else if (s == n) { // stale else if (s == n) // stale
// No need to also check for p == s, since that implies s == n // No need to also check for p == s, since that implies s == n
p = head; p = head;
} else { else
p.casNext(s, n); p.casNext(s, n);
}
} }
} }
@ -1012,9 +979,8 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
return true; return true;
} }
} }
else if (item == null) { else if (item == null)
break; break;
}
pred = p; pred = p;
if ((p = p.next) == pred) { // stale if ((p = p.next) == pred) { // stale
pred = null; pred = null;
@ -1052,7 +1018,6 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
* *
* @throws NullPointerException if the specified element is null * @throws NullPointerException if the specified element is null
*/ */
@Override
public void put(E e) { public void put(E e) {
xfer(e, true, ASYNC, 0); xfer(e, true, ASYNC, 0);
} }
@ -1063,10 +1028,10 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
* return {@code false}. * return {@code false}.
* *
* @return {@code true} (as specified by * @return {@code true} (as specified by
* {@link BlockingQueue#offer(Object,long,TimeUnit) BlockingQueue.offer}) * {@link java.util.concurrent.BlockingQueue#offer(Object,long,TimeUnit)
* BlockingQueue.offer})
* @throws NullPointerException if the specified element is null * @throws NullPointerException if the specified element is null
*/ */
@Override
public boolean offer(E e, long timeout, TimeUnit unit) { public boolean offer(E e, long timeout, TimeUnit unit) {
xfer(e, true, ASYNC, 0); xfer(e, true, ASYNC, 0);
return true; return true;
@ -1076,11 +1041,9 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
* Inserts the specified element at the tail of this queue. * Inserts the specified element at the tail of this queue.
* As the queue is unbounded, this method will never return {@code false}. * As the queue is unbounded, this method will never return {@code false}.
* *
* @return {@code true} (as specified by * @return {@code true} (as specified by {@link Queue#offer})
* {@link BlockingQueue#offer(Object) BlockingQueue.offer})
* @throws NullPointerException if the specified element is null * @throws NullPointerException if the specified element is null
*/ */
@Override
public boolean offer(E e) { public boolean offer(E e) {
xfer(e, true, ASYNC, 0); xfer(e, true, ASYNC, 0);
return true; return true;
@ -1094,7 +1057,6 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
* @return {@code true} (as specified by {@link Collection#add}) * @return {@code true} (as specified by {@link Collection#add})
* @throws NullPointerException if the specified element is null * @throws NullPointerException if the specified element is null
*/ */
@Override
public boolean add(E e) { public boolean add(E e) {
xfer(e, true, ASYNC, 0); xfer(e, true, ASYNC, 0);
return true; return true;
@ -1148,35 +1110,28 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
*/ */
public boolean tryTransfer(E e, long timeout, TimeUnit unit) public boolean tryTransfer(E e, long timeout, TimeUnit unit)
throws InterruptedException { throws InterruptedException {
if (xfer(e, true, TIMED, unit.toNanos(timeout)) == null) { if (xfer(e, true, TIMED, unit.toNanos(timeout)) == null)
return true; return true;
} if (!Thread.interrupted())
if (!Thread.interrupted()) {
return false; return false;
}
throw new InterruptedException(); throw new InterruptedException();
} }
@Override
public E take() throws InterruptedException { public E take() throws InterruptedException {
E e = xfer(null, false, SYNC, 0); E e = xfer(null, false, SYNC, 0);
if (e != null) { if (e != null)
return e; return e;
}
Thread.interrupted(); Thread.interrupted();
throw new InterruptedException(); throw new InterruptedException();
} }
@Override
public E poll(long timeout, TimeUnit unit) throws InterruptedException { public E poll(long timeout, TimeUnit unit) throws InterruptedException {
E e = xfer(null, false, TIMED, unit.toNanos(timeout)); E e = xfer(null, false, TIMED, unit.toNanos(timeout));
if (e != null || !Thread.interrupted()) { if (e != null || !Thread.interrupted())
return e; return e;
}
throw new InterruptedException(); throw new InterruptedException();
} }
@Override
public E poll() { public E poll() {
return xfer(null, false, NOW, 0); return xfer(null, false, NOW, 0);
} }
@ -1185,17 +1140,13 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
* @throws NullPointerException {@inheritDoc} * @throws NullPointerException {@inheritDoc}
* @throws IllegalArgumentException {@inheritDoc} * @throws IllegalArgumentException {@inheritDoc}
*/ */
@Override
public int drainTo(Collection<? super E> c) { public int drainTo(Collection<? super E> c) {
if (c == null) { if (c == null)
throw new NullPointerException(); throw new NullPointerException();
} if (c == this)
if (c == this) {
throw new IllegalArgumentException(); throw new IllegalArgumentException();
}
int n = 0; int n = 0;
E e; for (E e; (e = poll()) != null;) {
while ( (e = poll()) != null) {
c.add(e); c.add(e);
++n; ++n;
} }
@ -1206,17 +1157,13 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
* @throws NullPointerException {@inheritDoc} * @throws NullPointerException {@inheritDoc}
* @throws IllegalArgumentException {@inheritDoc} * @throws IllegalArgumentException {@inheritDoc}
*/ */
@Override
public int drainTo(Collection<? super E> c, int maxElements) { public int drainTo(Collection<? super E> c, int maxElements) {
if (c == null) { if (c == null)
throw new NullPointerException(); throw new NullPointerException();
} if (c == this)
if (c == this) {
throw new IllegalArgumentException(); throw new IllegalArgumentException();
}
int n = 0; int n = 0;
E e; for (E e; n < maxElements && (e = poll()) != null;) {
while (n < maxElements && (e = poll()) != null) {
c.add(e); c.add(e);
++n; ++n;
} }
@ -1224,24 +1171,22 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
} }
/** /**
* Returns an iterator over the elements in this queue in proper * Returns an iterator over the elements in this queue in proper sequence.
* sequence, from head to tail. * The elements will be returned in order from first (head) to last (tail).
* *
* <p>The returned iterator is a "weakly consistent" iterator that * <p>The returned iterator is a "weakly consistent" iterator that
* will never throw * will never throw {@link java.util.ConcurrentModificationException
* {@link ConcurrentModificationException ConcurrentModificationException}, * ConcurrentModificationException}, and guarantees to traverse
* and guarantees to traverse elements as they existed upon * elements as they existed upon construction of the iterator, and
* construction of the iterator, and may (but is not guaranteed * may (but is not guaranteed to) reflect any modifications
* to) reflect any modifications subsequent to construction. * subsequent to construction.
* *
* @return an iterator over the elements in this queue in proper sequence * @return an iterator over the elements in this queue in proper sequence
*/ */
@Override
public Iterator<E> iterator() { public Iterator<E> iterator() {
return new Itr(); return new Itr();
} }
@Override
public E peek() { public E peek() {
return firstDataItem(); return firstDataItem();
} }
@ -1251,12 +1196,10 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
* *
* @return {@code true} if this queue contains no elements * @return {@code true} if this queue contains no elements
*/ */
@Override
public boolean isEmpty() { public boolean isEmpty() {
for (Node p = head; p != null; p = succ(p)) { for (Node p = head; p != null; p = succ(p)) {
if (!p.isMatched()) { if (!p.isMatched())
return !p.isData; return !p.isData;
}
} }
return true; return true;
} }
@ -1277,7 +1220,6 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
* *
* @return the number of elements in this queue * @return the number of elements in this queue
*/ */
@Override
public int size() { public int size() {
return countOfMode(true); return countOfMode(true);
} }
@ -1297,19 +1239,40 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
* @param o element to be removed from this queue, if present * @param o element to be removed from this queue, if present
* @return {@code true} if this queue changed as a result of the call * @return {@code true} if this queue changed as a result of the call
*/ */
@Override
public boolean remove(Object o) { public boolean remove(Object o) {
return findAndRemove(o); return findAndRemove(o);
} }
/**
* Returns {@code true} if this queue contains the specified element.
* More formally, returns {@code true} if and only if this queue contains
* at least one element {@code e} such that {@code o.equals(e)}.
*
* @param o object to be checked for containment in this queue
* @return {@code true} if this queue contains the specified element
*/
public boolean contains(Object o) {
if (o == null) return false;
for (Node p = head; p != null; p = succ(p)) {
Object item = p.item;
if (p.isData) {
if (item != null && item != p && o.equals(item))
return true;
}
else if (item == null)
break;
}
return false;
}
/** /**
* Always returns {@code Integer.MAX_VALUE} because a * Always returns {@code Integer.MAX_VALUE} because a
* {@code LinkedTransferQueue} is not capacity constrained. * {@code LinkedTransferQueue} is not capacity constrained.
* *
* @return {@code Integer.MAX_VALUE} (as specified by * @return {@code Integer.MAX_VALUE} (as specified by
* {@link BlockingQueue#remainingCapacity()}) * {@link java.util.concurrent.BlockingQueue#remainingCapacity()
* BlockingQueue.remainingCapacity})
*/ */
@Override
public int remainingCapacity() { public int remainingCapacity() {
return Integer.MAX_VALUE; return Integer.MAX_VALUE;
} }
@ -1324,9 +1287,8 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
private void writeObject(java.io.ObjectOutputStream s) private void writeObject(java.io.ObjectOutputStream s)
throws java.io.IOException { throws java.io.IOException {
s.defaultWriteObject(); s.defaultWriteObject();
for (E e : this) { for (E e : this)
s.writeObject(e); s.writeObject(e);
}
// Use trailing null as sentinel // Use trailing null as sentinel
s.writeObject(null); s.writeObject(null);
} }
@ -1341,23 +1303,62 @@ public class LinkedTransferQueue<E> extends AbstractQueue<E>
throws java.io.IOException, ClassNotFoundException { throws java.io.IOException, ClassNotFoundException {
s.defaultReadObject(); s.defaultReadObject();
for (;;) { for (;;) {
@SuppressWarnings("unchecked") E item = (E) s.readObject(); @SuppressWarnings("unchecked")
if (item == null) { E item = (E) s.readObject();
if (item == null)
break; break;
} else { else
offer(item); offer(item);
}
}
// Unsafe mechanics
private static final sun.misc.Unsafe UNSAFE;
private static final long headOffset;
private static final long tailOffset;
private static final long sweepVotesOffset;
static {
try {
UNSAFE = getUnsafe();
Class<?> k = LinkedTransferQueue.class;
headOffset = UNSAFE.objectFieldOffset
(k.getDeclaredField("head"));
tailOffset = UNSAFE.objectFieldOffset
(k.getDeclaredField("tail"));
sweepVotesOffset = UNSAFE.objectFieldOffset
(k.getDeclaredField("sweepVotes"));
} catch (Exception e) {
throw new Error(e);
}
}
/**
* Returns a sun.misc.Unsafe. Suitable for use in a 3rd party package.
* Replace with a simple call to Unsafe.getUnsafe when integrating
* into a jdk.
*
* @return a sun.misc.Unsafe
*/
static sun.misc.Unsafe getUnsafe() {
try {
return sun.misc.Unsafe.getUnsafe();
} catch (SecurityException se) {
try {
return java.security.AccessController.doPrivileged
(new java.security
.PrivilegedExceptionAction<sun.misc.Unsafe>() {
public sun.misc.Unsafe run() throws Exception {
java.lang.reflect.Field f = sun.misc
.Unsafe.class.getDeclaredField("theUnsafe");
f.setAccessible(true);
return (sun.misc.Unsafe) f.get(null);
}});
} catch (java.security.PrivilegedActionException e) {
throw new RuntimeException("Could not initialize intrinsics",
e.getCause());
} }
} }
} }
@SuppressWarnings("rawtypes")
private static final AtomicReferenceFieldUpdater<LinkedTransferQueue, Node> headUpdater =
AtomicFieldUpdaterUtil.newRefUpdater(LinkedTransferQueue.class, Node.class, "head");
@SuppressWarnings("rawtypes")
private static final AtomicReferenceFieldUpdater<LinkedTransferQueue, Node> tailUpdater =
AtomicFieldUpdaterUtil.newRefUpdater(LinkedTransferQueue.class, Node.class, "tail");
@SuppressWarnings("rawtypes")
private static final AtomicIntegerFieldUpdater<LinkedTransferQueue> sweepVotesUpdater =
AtomicFieldUpdaterUtil.newIntUpdater(LinkedTransferQueue.class, "sweepVotes");
} }

View File

@ -0,0 +1,69 @@
/*
* Copyright 2011 Red Hat, Inc.
*
* Red Hat licenses this file to you under the Apache License, version 2.0
* (the "License"); you may not use this file except in compliance with the
* License. You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations
* under the License.
*/
package org.jboss.netty.util.internal;
import java.util.Collection;
import java.util.concurrent.BlockingQueue;
import org.jboss.netty.util.UnsafeDetectUtil;
/**
* This factory should be used to create the "optimal" {@link BlockingQueue} instance for the running JVM.
*
*
*
* @author <a href="http://www.jboss.org/netty/">The Netty Project</a>
* @author <a href="http://www.murkycloud.com/">Norman Maurer</a>
*
*/
public class QueueFactory {
private static final boolean useUnsafe = UnsafeDetectUtil.isUnsafeFound(QueueFactory.class.getClassLoader());
private QueueFactory() {
// only use static methods!
}
/**
* Create a new unbound {@link BlockingQueue}
*
* @param itemClass the {@link Class} type which will be used as {@link BlockingQueue} items
* @return queue the {@link BlockingQueue} implementation
*/
public static final <T> BlockingQueue<T> createQueue(Class<T> itemClass) {
if (useUnsafe) {
return new LinkedTransferQueue<T>();
} else {
return new LegacyLinkedTransferQueue<T>();
}
}
/**
* Create a new unbound {@link BlockingQueue}
*
* @param collection the collection which should get copied to the newly created {@link BlockingQueue}
* @param itemClass the {@link Class} type which will be used as {@link BlockingQueue} items
* @return queue the {@link BlockingQueue} implementation
*/
public static final <T> BlockingQueue<T> createQueue(Collection<? extends T> collection, Class<T> itemClass) {
if (useUnsafe) {
return new LinkedTransferQueue<T>(collection);
} else {
return new LegacyLinkedTransferQueue<T>(collection);
}
}
}