public class ResultPartitionManager extends Object implements ResultPartitionProvider
Constructor and Description |
---|
ResultPartitionManager() |
ResultPartitionManager(int partitionListenerTimeout,
ScheduledExecutor scheduledExecutor) |
Modifier and Type | Method and Description |
---|---|
ResultSubpartitionView |
createSubpartitionView(ResultPartitionID partitionId,
ResultSubpartitionIndexSet subpartitionIndexSet,
BufferAvailabilityListener availabilityListener)
Returns the requested intermediate result partition input view.
|
Optional<ResultSubpartitionView> |
createSubpartitionViewOrRegisterListener(ResultPartitionID partitionId,
ResultSubpartitionIndexSet subpartitionIndexSet,
BufferAvailabilityListener availabilityListener,
PartitionRequestListener partitionRequestListener)
If the upstream task's partition has been registered, returns the result subpartition input
view immediately, otherwise register the listener and return empty.
|
Map<ResultPartitionID,PartitionRequestListenerManager> |
getListenerManagers() |
Collection<ResultPartitionID> |
getUnreleasedPartitions() |
void |
registerResultPartition(ResultPartition partition) |
void |
releasePartition(ResultPartitionID partitionId,
Throwable cause) |
void |
releasePartitionRequestListener(PartitionRequestListener listener)
Release the given listener in this result partition provider.
|
void |
shutdown() |
@VisibleForTesting public ResultPartitionManager()
public ResultPartitionManager(int partitionListenerTimeout, ScheduledExecutor scheduledExecutor)
public void registerResultPartition(ResultPartition partition) throws IOException
IOException
public ResultSubpartitionView createSubpartitionView(ResultPartitionID partitionId, ResultSubpartitionIndexSet subpartitionIndexSet, BufferAvailabilityListener availabilityListener) throws IOException
ResultPartitionProvider
createSubpartitionView
in interface ResultPartitionProvider
IOException
public Optional<ResultSubpartitionView> createSubpartitionViewOrRegisterListener(ResultPartitionID partitionId, ResultSubpartitionIndexSet subpartitionIndexSet, BufferAvailabilityListener availabilityListener, PartitionRequestListener partitionRequestListener) throws IOException
ResultPartitionProvider
createSubpartitionViewOrRegisterListener
in interface ResultPartitionProvider
partitionId
- the result partition idsubpartitionIndexSet
- the index setavailabilityListener
- the buffer availability listenerpartitionRequestListener
- the partition request listenerIOException
- the thrown exceptionpublic void releasePartitionRequestListener(PartitionRequestListener listener)
ResultPartitionProvider
releasePartitionRequestListener
in interface ResultPartitionProvider
listener
- the given listenerpublic void releasePartition(ResultPartitionID partitionId, Throwable cause)
public void shutdown()
@VisibleForTesting public Map<ResultPartitionID,PartitionRequestListenerManager> getListenerManagers()
public Collection<ResultPartitionID> getUnreleasedPartitions()
Copyright © 2014–2024 The Apache Software Foundation. All rights reserved.