org.apache.flink.runtime.shuffle

Class JobShuffleContextImpl

java.lang.Object

org.apache.flink.runtime.shuffle.JobShuffleContextImpl

All Implemented Interfaces:

JobShuffleContext

public class JobShuffleContextImpl
extends Object
implements JobShuffleContext

The default implementation of JobShuffleContext.

Constructor Summary

Constructors

Constructor and Description
JobShuffleContextImpl(org.apache.flink.api.common.JobID jobId, JobMasterGateway jobMasterGateway)

Method Summary

Modifier and Type	Method and Description
org.apache.flink.api.common.JobID	getJobId()
CompletableFuture<Collection<PartitionWithMetrics>>	getPartitionWithMetrics(Duration timeout, Set<ResultPartitionID> expectedPartitions) Retrieves specified partitions and their metrics (identified by expectedPartitions), the metrics include sizes of sub-partitions in a result partition.
void	notifyPartitionRecoveryStarted() Notifies that the recovery process of result partitions has started.
CompletableFuture<?>	stopTrackingAndReleasePartitions(Collection<ResultPartitionID> partitionIds) Notifies the job to stop tracking and release the target result partitions, which means these partitions will be removed and will be reproduced if used afterwards.

Methods inherited from class java.lang.Object

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

Constructor Detail

JobShuffleContextImpl

public JobShuffleContextImpl(org.apache.flink.api.common.JobID jobId,
                             JobMasterGateway jobMasterGateway)

Method Detail

getJobId

public org.apache.flink.api.common.JobID getJobId()

Specified by:

getJobId in interface JobShuffleContext

Returns:

the corresponding JobID.

stopTrackingAndReleasePartitions

public CompletableFuture<?> stopTrackingAndReleasePartitions(Collection<ResultPartitionID> partitionIds)

Description copied from interface: JobShuffleContext

Notifies the job to stop tracking and release the target result partitions, which means these partitions will be removed and will be reproduced if used afterwards.

Specified by:

stopTrackingAndReleasePartitions in interface JobShuffleContext

getPartitionWithMetrics

public CompletableFuture<Collection<PartitionWithMetrics>> getPartitionWithMetrics(Duration timeout,
                                                                                   Set<ResultPartitionID> expectedPartitions)

Description copied from interface: JobShuffleContext

Retrieves specified partitions and their metrics (identified by expectedPartitions), the metrics include sizes of sub-partitions in a result partition.

Specified by:

getPartitionWithMetrics in interface JobShuffleContext

Parameters:

timeout - The timeout used for retrieve the specified partitions.

expectedPartitions - The set of identifiers for the result partitions whose metrics are to be fetched.

Returns:

A future will contain a collection of the partitions with their metrics that could be retrieved from the expected partitions within the specified timeout period.

notifyPartitionRecoveryStarted

public void notifyPartitionRecoveryStarted()

Description copied from interface: JobShuffleContext

Notifies that the recovery process of result partitions has started.

Specified by:

notifyPartitionRecoveryStarted in interface JobShuffleContext