org.jctools.queues.MpscLinkedQueue Maven / Gradle / Ivy
/*
* Licensed 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.jctools.queues;
import org.jctools.util.UnsafeAccess;
import java.util.Queue;
import static org.jctools.util.UnsafeAccess.UNSAFE;
/**
* This is a Java port of the MPSC algorithm as presented
* on
* 1024 Cores by D. Vyukov. The original has been adapted to Java and it's quirks with regards to memory
* model and layout:
*
* - Use inheritance to ensure no false sharing occurs between producer/consumer node reference fields.
*
- Use XCHG functionality to the best of the JDK ability (see differences in JDK7/8 impls).
*
- Conform to {@link java.util.Queue} contract on poll. The original semantics are available via relaxedPoll.
*
* The queue is initialized with a stub node which is set to both the producer and consumer node references.
* From this point follow the notes on offer/poll.
*/
public class MpscLinkedQueue extends BaseLinkedQueue
{
public MpscLinkedQueue()
{
LinkedQueueNode node = newNode();
spConsumerNode(node);
xchgProducerNode(node);
}
/**
* {@inheritDoc}
*
* IMPLEMENTATION NOTES:
* Offer is allowed from multiple threads.
* Offer allocates a new node and:
*
* - Swaps it atomically with current producer node (only one producer 'wins')
*
- Sets the new node as the node following from the swapped producer node
*
* This works because each producer is guaranteed to 'plant' a new node and link the old node. No 2
* producers can get the same producer node as part of XCHG guarantee.
*
* @see MessagePassingQueue#offer(Object)
* @see java.util.Queue#offer(java.lang.Object)
*/
@Override
public boolean offer(final E e)
{
if (null == e)
{
throw new NullPointerException();
}
final LinkedQueueNode nextNode = newNode(e);
final LinkedQueueNode prevProducerNode = xchgProducerNode(nextNode);
// Should a producer thread get interrupted here the chain WILL be broken until that thread is resumed
// and completes the store in prev.next. This is a "bubble".
prevProducerNode.soNext(nextNode);
return true;
}
/**
* {@inheritDoc}
*
* This method is only safe to call from the (single) consumer thread, and is subject to best effort when racing
* with producers. This method is potentially blocking when "bubble"s in the queue are visible.
*/
@Override
public boolean remove(Object o)
{
if (null == o)
{
return false; // Null elements are not permitted, so null will never be removed.
}
final LinkedQueueNode originalConsumerNode = lpConsumerNode();
LinkedQueueNode prevConsumerNode = originalConsumerNode;
LinkedQueueNode currConsumerNode = getNextConsumerNode(originalConsumerNode);
while (currConsumerNode != null)
{
if (o.equals(currConsumerNode.lpValue()))
{
LinkedQueueNode nextNode = getNextConsumerNode(currConsumerNode);
// e.g.: consumerNode -> node0 -> node1(o==v) -> node2 ... => consumerNode -> node0 -> node2
if (nextNode != null)
{
// We are removing an interior node.
prevConsumerNode.soNext(nextNode);
}
// This case reflects: prevConsumerNode != originalConsumerNode && nextNode == null
// At rest, this would be the producerNode, but we must contend with racing. Changes to subclassed
// queues need to consider remove() when implementing offer().
else
{
// producerNode is currConsumerNode, try to atomically update the reference to move it to the
// previous node.
prevConsumerNode.soNext(null);
if (!casProducerNode(currConsumerNode, prevConsumerNode))
{
// If the producer(s) have offered more items we need to remove the currConsumerNode link.
nextNode = spinWaitForNextNode(currConsumerNode);
prevConsumerNode.soNext(nextNode);
}
}
// Avoid GC nepotism because we are discarding the current node.
currConsumerNode.soNext(null);
currConsumerNode.spValue(null);
return true;
}
prevConsumerNode = currConsumerNode;
currConsumerNode = getNextConsumerNode(currConsumerNode);
}
return false;
}
@Override
public int fill(Supplier s)
{
return MessagePassingQueueUtil.fillUnbounded(this, s);
}
@Override
public int fill(Supplier s, int limit)
{
if (null == s)
throw new IllegalArgumentException("supplier is null");
if (limit < 0)
throw new IllegalArgumentException("limit is negative:" + limit);
if (limit == 0)
return 0;
LinkedQueueNode tail = newNode(s.get());
final LinkedQueueNode head = tail;
for (int i = 1; i < limit; i++)
{
final LinkedQueueNode temp = newNode(s.get());
// spNext: xchgProducerNode ensures correct construction
tail.spNext(temp);
tail = temp;
}
final LinkedQueueNode oldPNode = xchgProducerNode(tail);
oldPNode.soNext(head);
return limit;
}
@Override
public void fill(Supplier s, WaitStrategy wait, ExitCondition exit)
{
MessagePassingQueueUtil.fill(this, s, wait, exit);
}
// $gen:ignore
private LinkedQueueNode xchgProducerNode(LinkedQueueNode newVal)
{
if (UnsafeAccess.SUPPORTS_GET_AND_SET_REF)
{
return (LinkedQueueNode) UNSAFE.getAndSetObject(this, P_NODE_OFFSET, newVal);
}
else
{
LinkedQueueNode oldVal;
do
{
oldVal = lvProducerNode();
}
while (!UNSAFE.compareAndSwapObject(this, P_NODE_OFFSET, oldVal, newVal));
return oldVal;
}
}
private LinkedQueueNode getNextConsumerNode(LinkedQueueNode currConsumerNode)
{
LinkedQueueNode nextNode = currConsumerNode.lvNext();
if (nextNode == null && currConsumerNode != lvProducerNode())
{
nextNode = spinWaitForNextNode(currConsumerNode);
}
return nextNode;
}
}