org.infinispan.stream.impl

Class PartitionAwareClusterStreamManager<K>

java.lang.Object

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

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

All Implemented Interfaces:

ClusterStreamManager<K>

public class PartitionAwareClusterStreamManager<K>
extends ClusterStreamManagerImpl<K>

Cluster stream manager that also pays attention to partition status and properly closes iterators and throws exceptions when the availability mode changes.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
protected class	PartitionAwareClusterStreamManager.PartitionListener

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

ClusterStreamManager.ResultsCallback<R>

Field Summary

Fields

Modifier and Type	Field and Description
protected Cache<?,?>	cache
protected PartitionAwareClusterStreamManager.PartitionListener	listener

Fields inherited from class org.infinispan.stream.impl.ClusterStreamManagerImpl

currentlyRunning, factory, localAddress, log, requestId, rpc

Constructor Summary

Constructors

Constructor and Description
PartitionAwareClusterStreamManager()

Method Summary

Modifier and Type	Method and Description
boolean	awaitCompletion(Object id, long time, TimeUnit unit) Awaits completion of the given request.
void	inject(Cache<?,?> cache)
<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 org.infinispan.stream.impl.ClusterStreamManagerImpl

forgetOperation, inject, isComplete, markTrackerWithException, receiveResponse

Methods inherited from class java.lang.Object

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

Field Detail

listener

protected final PartitionAwareClusterStreamManager.PartitionListener listener

cache

protected Cache<?,?> cache

Constructor Detail

PartitionAwareClusterStreamManager

public PartitionAwareClusterStreamManager()

Method Detail

inject

public void inject(Cache<?,?> cache)

start

public void start()

Overrides:

start in class ClusterStreamManagerImpl<K>

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>

Overrides:

awaitCompletion in class ClusterStreamManagerImpl<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

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>

Overrides:

remoteStreamOperation in class ClusterStreamManagerImpl<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>

Overrides:

remoteStreamOperation in class ClusterStreamManagerImpl<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 <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>

Overrides:

remoteStreamOperationRehashAware in class ClusterStreamManagerImpl<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 <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>

Overrides:

remoteStreamOperationRehashAware in class ClusterStreamManagerImpl<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