org.infinispan.stream.impl

Class ClusterStreamManagerImpl<K>

java.lang.Object

org.infinispan.stream.impl.ClusterStreamManagerImpl<K>

Type Parameters:

K - the cache key type

All Implemented Interfaces:

ClusterStreamManager<K>

Direct Known Subclasses:

PartitionAwareClusterStreamManager

public class ClusterStreamManagerImpl<K>
extends Object
implements ClusterStreamManager<K>

Cluster stream manager that sends all requests using the RpcManager to do the underlying communications.

Nested Class Summary

Nested classes/interfaces inherited from interface org.infinispan.stream.impl.ClusterStreamManager

ClusterStreamManager.ResultsCallback<R>

Field Summary

Fields

Modifier and Type	Field and Description
protected Map<String,org.infinispan.stream.impl.ClusterStreamManagerImpl.RequestTracker>	currentlyRunning
protected CommandsFactory	factory
protected Address	localAddress
protected static Log	log
protected AtomicInteger	requestId
protected RpcManager	rpc

Constructor Summary

Constructors

Constructor and Description
ClusterStreamManagerImpl()

Method Summary

Modifier and Type	Method and Description
boolean	awaitCompletion(Object id, long time, TimeUnit unit) Awaits completion of the given request.
void	forgetOperation(Object id) Frees any resources related to this operation and signals to any ongoing remote operations to no longer continue processing
void	inject(RpcManager rpc, CommandsFactory factory)
boolean	isComplete(Object id) Tests whether this operation is still pending or not.
protected static void	markTrackerWithException(org.infinispan.stream.impl.ClusterStreamManagerImpl.RequestTracker<?> tracker, Address dest, Throwable e, Object uuid)
<R1> boolean	receiveResponse(Object id, Address origin, boolean complete, Set<Integer> missingSegments, R1 response) Receives a response for a given request
<R> Object	remoteStreamOperation(boolean parallelDistribution, boolean parallelStream, ConsistentHash ch, Set<Integer> segments, Set<K> keysToInclude, Map<Integer,Set<K>> keysToExclude, boolean includeLoader, KeyTrackingTerminalOperation<K,R,?> operation, ClusterStreamManager.ResultsCallback<Collection<R>> callback) Key tracking remote operation that doesn't have rehash enabled.
<R> Object	remoteStreamOperation(boolean parallelDistribution, boolean parallelStream, ConsistentHash ch, Set<Integer> segments, Set<K> keysToInclude, Map<Integer,Set<K>> keysToExclude, boolean includeLoader, TerminalOperation<R> operation, ClusterStreamManager.ResultsCallback<R> callback, Predicate<? super R> earlyTerminatePredicate) Performs the remote stream operation without rehash awareness.
<R2> Object	remoteStreamOperationRehashAware(boolean parallelDistribution, boolean parallelStream, ConsistentHash ch, Set<Integer> segments, Set<K> keysToInclude, Map<Integer,Set<K>> keysToExclude, boolean includeLoader, KeyTrackingTerminalOperation<K,?,R2> operation, ClusterStreamManager.ResultsCallback<Map<K,R2>> callback) Key tracking remote operation that has rehash enabled
<R> Object	remoteStreamOperationRehashAware(boolean parallelDistribution, boolean parallelStream, ConsistentHash ch, Set<Integer> segments, Set<K> keysToInclude, Map<Integer,Set<K>> keysToExclude, boolean includeLoader, TerminalOperation<R> operation, ClusterStreamManager.ResultsCallback<R> callback, Predicate<? super R> earlyTerminatePredicate) Performs the remote stream operation with rehash awareness.
void	start()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

currentlyRunning

protected final Map<String,org.infinispan.stream.impl.ClusterStreamManagerImpl.RequestTracker> currentlyRunning

requestId

protected final AtomicInteger requestId

rpc

protected RpcManager rpc

factory

protected CommandsFactory factory

localAddress

protected Address localAddress

log

protected static final Log log

Constructor Detail

ClusterStreamManagerImpl

public ClusterStreamManagerImpl()

Method Detail

inject

public void inject(RpcManager rpc,
                   CommandsFactory factory)

start

public void start()

remoteStreamOperation

public <R> Object remoteStreamOperation(boolean parallelDistribution,
                                        boolean parallelStream,
                                        ConsistentHash ch,
                                        Set<Integer> segments,
                                        Set<K> keysToInclude,
                                        Map<Integer,Set<K>> keysToExclude,
                                        boolean includeLoader,
                                        TerminalOperation<R> operation,
                                        ClusterStreamManager.ResultsCallback<R> callback,
                                        Predicate<? super R> earlyTerminatePredicate)

Description copied from interface: ClusterStreamManager

Performs the remote stream operation without rehash awareness.

Specified by:

remoteStreamOperation in interface ClusterStreamManager<K>

Type Parameters:

R - the type of response

Parameters:

parallelDistribution - whether or not parallel distribution is enabled

parallelStream - whether or not the stream is paralllel

ch - the consistent hash to use when determining segment ownership

segments - the segments that this request should utilize

keysToInclude - which keys to include in the request

keysToExclude - which keys to exclude in the request

includeLoader - whether or not to use a loader

operation - the actual operation to perform

callback - the callback to collect individual node results

earlyTerminatePredicate - a predicate to determine if this operation should stop based on intermediate results

Returns:

the operation id to be used for further calls

remoteStreamOperationRehashAware

public <R> Object remoteStreamOperationRehashAware(boolean parallelDistribution,
                                                   boolean parallelStream,
                                                   ConsistentHash ch,
                                                   Set<Integer> segments,
                                                   Set<K> keysToInclude,
                                                   Map<Integer,Set<K>> keysToExclude,
                                                   boolean includeLoader,
                                                   TerminalOperation<R> operation,
                                                   ClusterStreamManager.ResultsCallback<R> callback,
                                                   Predicate<? super R> earlyTerminatePredicate)

Description copied from interface: ClusterStreamManager

Performs the remote stream operation with rehash awareness.

Specified by:

remoteStreamOperationRehashAware in interface ClusterStreamManager<K>

Type Parameters:

R - the type of response

Parameters:

parallelDistribution - whether or not parallel distribution is enabled

parallelStream - whether or not the stream is paralllel

ch - the consistent hash to use when determining segment ownership

segments - the segments that this request should utilize

keysToInclude - which keys to include in the request

keysToExclude - which keys to exclude in the request

includeLoader - whether or not to use a loader

operation - the actual operation to perform

callback - the callback to collect individual node results

earlyTerminatePredicate - a predicate to determine if this operation should stop based on intermediate results

Returns:

the operation id to be used for further calls

remoteStreamOperation

public <R> Object remoteStreamOperation(boolean parallelDistribution,
                                        boolean parallelStream,
                                        ConsistentHash ch,
                                        Set<Integer> segments,
                                        Set<K> keysToInclude,
                                        Map<Integer,Set<K>> keysToExclude,
                                        boolean includeLoader,
                                        KeyTrackingTerminalOperation<K,R,?> operation,
                                        ClusterStreamManager.ResultsCallback<Collection<R>> callback)

Description copied from interface: ClusterStreamManager

Key tracking remote operation that doesn't have rehash enabled.

Specified by:

remoteStreamOperation in interface ClusterStreamManager<K>

Type Parameters:

R - the type of response

Parameters:

parallelDistribution - whether or not parallel distribution is enabled

parallelStream - whether or not the stream is paralllel

ch - the consistent hash to use when determining segment ownership

segments - the segments that this request should utilize

keysToInclude - which keys to include in the request

keysToExclude - which keys to exclude in the request

includeLoader - whether or not to use a loader

operation - the actual operation to perform

callback - the callback to collect individual node results

Returns:

the operation id to be used for further calls

remoteStreamOperationRehashAware

public <R2> Object remoteStreamOperationRehashAware(boolean parallelDistribution,
                                                    boolean parallelStream,
                                                    ConsistentHash ch,
                                                    Set<Integer> segments,
                                                    Set<K> keysToInclude,
                                                    Map<Integer,Set<K>> keysToExclude,
                                                    boolean includeLoader,
                                                    KeyTrackingTerminalOperation<K,?,R2> operation,
                                                    ClusterStreamManager.ResultsCallback<Map<K,R2>> callback)

Description copied from interface: ClusterStreamManager

Key tracking remote operation that has rehash enabled

Specified by:

remoteStreamOperationRehashAware in interface ClusterStreamManager<K>

Type Parameters:

R2 - the type of response

Parameters:

parallelDistribution - whether or not parallel distribution is enabled

parallelStream - whether or not the stream is paralllel

ch - the consistent hash to use when determining segment ownership

segments - the segments that this request should utilize

keysToInclude - which keys to include in the request

keysToExclude - which keys to exclude in the request

includeLoader - whether or not to use a loader

operation - the actual operation to perform

callback - the callback to collect individual node results

Returns:

the operation id to be used for further calls

markTrackerWithException

protected static void markTrackerWithException(org.infinispan.stream.impl.ClusterStreamManagerImpl.RequestTracker<?> tracker,
                                               Address dest,
                                               Throwable e,
                                               Object uuid)

isComplete

public boolean isComplete(Object id)

Description copied from interface: ClusterStreamManager

Tests whether this operation is still pending or not.

Specified by:

isComplete in interface ClusterStreamManager<K>

Parameters:

id - the id of the operation that was returned from the invocation

Returns:

whether or not it is completed

awaitCompletion

public boolean awaitCompletion(Object id,
                               long time,
                               TimeUnit unit)
                        throws InterruptedException

Description copied from interface: ClusterStreamManager

Awaits completion of the given request. Returns true when the request completes otherwise returns false after the time elapses

Specified by:

awaitCompletion in interface ClusterStreamManager<K>

Parameters:

id - the ide of the operation that was returned from the invocation - must be non null

time - how long to wait before returning false - must be greater than 0

unit - controls how long the time wait is

Returns:

whether or not the request is complete

Throws:

InterruptedException

forgetOperation

public void forgetOperation(Object id)

Description copied from interface: ClusterStreamManager

Frees any resources related to this operation and signals to any ongoing remote operations to no longer continue processing

Specified by:

forgetOperation in interface ClusterStreamManager<K>

Parameters:

id - the ide of the operation that was returned from the invocation - can be null in which case this is a noop

receiveResponse

public <R1> boolean receiveResponse(Object id,
                                    Address origin,
                                    boolean complete,
                                    Set<Integer> missingSegments,
                                    R1 response)

Description copied from interface: ClusterStreamManager

Receives a response for a given request

Specified by:

receiveResponse in interface ClusterStreamManager<K>

Type Parameters:

R1 - The type of the response

Parameters:

id - The request id

origin - The origin of the response

complete - Whether or not this is a completed response

missingSegments - The segments that were suspected

response - The actual response value

Returns:

Whether or not the operation should continue operating, only valid if complete was false