org.apache.flink.runtime.jobmaster

Class RpcTaskManagerGateway

java.lang.Object

org.apache.flink.runtime.jobmaster.RpcTaskManagerGateway

All Implemented Interfaces:

TaskManagerGateway

public class RpcTaskManagerGateway
extends Object
implements TaskManagerGateway

Implementation of the TaskManagerGateway for Flink's RPC system.

Constructor Summary

Constructors

Constructor and Description
RpcTaskManagerGateway(TaskExecutorGateway taskExecutorGateway, JobMasterId jobMasterId)

Method Summary

Modifier and Type	Method and Description
CompletableFuture<Acknowledge>	cancelTask(ExecutionAttemptID executionAttemptID, Time timeout) Cancel the given task.
void	failPartition(ExecutionAttemptID executionAttemptID) Fail all intermediate result partitions of the given task.
CompletableFuture<Acknowledge>	freeSlot(AllocationID allocationId, Throwable cause, Time timeout) Frees the slot with the given allocation ID.
String	getAddress() Return the address of the task manager with which the gateway is associated.
void	notifyCheckpointComplete(ExecutionAttemptID executionAttemptID, JobID jobId, long checkpointId, long timestamp) Notify the given task about a completed checkpoint.
CompletableFuture<StackTraceSampleResponse>	requestStackTraceSample(ExecutionAttemptID executionAttemptID, int sampleId, int numSamples, Time delayBetweenSamples, int maxStackTraceDepth, Time timeout) Request a stack trace sample from the given task.
CompletableFuture<Acknowledge>	stopTask(ExecutionAttemptID executionAttemptID, Time timeout) Stop the given task.
CompletableFuture<Acknowledge>	submitTask(TaskDeploymentDescriptor tdd, Time timeout) Submit a task to the task manager.
void	triggerCheckpoint(ExecutionAttemptID executionAttemptID, JobID jobId, long checkpointId, long timestamp, CheckpointOptions checkpointOptions) Trigger for the given task a checkpoint.
CompletableFuture<Acknowledge>	updatePartitions(ExecutionAttemptID executionAttemptID, Iterable<PartitionInfo> partitionInfos, Time timeout) Update the task where the given partitions can be found.

Methods inherited from class java.lang.Object

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

Constructor Detail

RpcTaskManagerGateway

public RpcTaskManagerGateway(TaskExecutorGateway taskExecutorGateway,
                             JobMasterId jobMasterId)

Method Detail

getAddress

public String getAddress()

Description copied from interface: TaskManagerGateway

Return the address of the task manager with which the gateway is associated.

Specified by:

getAddress in interface TaskManagerGateway

Returns:

Address of the task manager with which this gateway is associated.

requestStackTraceSample

public CompletableFuture<StackTraceSampleResponse> requestStackTraceSample(ExecutionAttemptID executionAttemptID,
                                                                           int sampleId,
                                                                           int numSamples,
                                                                           Time delayBetweenSamples,
                                                                           int maxStackTraceDepth,
                                                                           Time timeout)

Description copied from interface: TaskManagerGateway

Request a stack trace sample from the given task.

Specified by:

requestStackTraceSample in interface TaskManagerGateway

Parameters:

executionAttemptID - identifying the task to sample

sampleId - of the sample

numSamples - to take from the given task

delayBetweenSamples - to wait for

maxStackTraceDepth - of the returned sample

timeout - of the request

Returns:

Future of stack trace sample response

submitTask

public CompletableFuture<Acknowledge> submitTask(TaskDeploymentDescriptor tdd,
                                                 Time timeout)

Description copied from interface: TaskManagerGateway

Submit a task to the task manager.

Specified by:

submitTask in interface TaskManagerGateway

Parameters:

tdd - describing the task to submit

timeout - of the submit operation

Returns:

Future acknowledge of the successful operation

stopTask

public CompletableFuture<Acknowledge> stopTask(ExecutionAttemptID executionAttemptID,
                                               Time timeout)

Description copied from interface: TaskManagerGateway

Stop the given task.

Specified by:

stopTask in interface TaskManagerGateway

Parameters:

executionAttemptID - identifying the task

timeout - of the submit operation

Returns:

Future acknowledge if the task is successfully stopped

cancelTask

public CompletableFuture<Acknowledge> cancelTask(ExecutionAttemptID executionAttemptID,
                                                 Time timeout)

Description copied from interface: TaskManagerGateway

Cancel the given task.

Specified by:

cancelTask in interface TaskManagerGateway

Parameters:

executionAttemptID - identifying the task

timeout - of the submit operation

Returns:

Future acknowledge if the task is successfully canceled

updatePartitions

public CompletableFuture<Acknowledge> updatePartitions(ExecutionAttemptID executionAttemptID,
                                                       Iterable<PartitionInfo> partitionInfos,
                                                       Time timeout)

Description copied from interface: TaskManagerGateway

Update the task where the given partitions can be found.

Specified by:

updatePartitions in interface TaskManagerGateway

Parameters:

executionAttemptID - identifying the task

partitionInfos - telling where the partition can be retrieved from

timeout - of the submit operation

Returns:

Future acknowledge if the partitions have been successfully updated

failPartition

public void failPartition(ExecutionAttemptID executionAttemptID)

Description copied from interface: TaskManagerGateway

Fail all intermediate result partitions of the given task.

Specified by:

failPartition in interface TaskManagerGateway

Parameters:

executionAttemptID - identifying the task

notifyCheckpointComplete

public void notifyCheckpointComplete(ExecutionAttemptID executionAttemptID,
                                     JobID jobId,
                                     long checkpointId,
                                     long timestamp)

Description copied from interface: TaskManagerGateway

Notify the given task about a completed checkpoint.

Specified by:

notifyCheckpointComplete in interface TaskManagerGateway

Parameters:

executionAttemptID - identifying the task

jobId - identifying the job to which the task belongs

checkpointId - of the completed checkpoint

timestamp - of the completed checkpoint

triggerCheckpoint

public void triggerCheckpoint(ExecutionAttemptID executionAttemptID,
                              JobID jobId,
                              long checkpointId,
                              long timestamp,
                              CheckpointOptions checkpointOptions)

Description copied from interface: TaskManagerGateway

Trigger for the given task a checkpoint.

Specified by:

triggerCheckpoint in interface TaskManagerGateway

Parameters:

executionAttemptID - identifying the task

jobId - identifying the job to which the task belongs

checkpointId - of the checkpoint to trigger

timestamp - of the checkpoint to trigger

checkpointOptions - of the checkpoint to trigger

freeSlot

public CompletableFuture<Acknowledge> freeSlot(AllocationID allocationId,
                                               Throwable cause,
                                               Time timeout)

Description copied from interface: TaskManagerGateway

Frees the slot with the given allocation ID.

Specified by:

freeSlot in interface TaskManagerGateway

Parameters:

allocationId - identifying the slot to free

cause - of the freeing operation

timeout - for the operation

Returns:

Future acknowledge which is returned once the slot has been freed