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, org.apache.flink.api.common.time.Time timeout) Cancel the given task.
CompletableFuture<Acknowledge>	freeSlot(AllocationID allocationId, Throwable cause, org.apache.flink.api.common.time.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, org.apache.flink.api.common.JobID jobId, long checkpointId, long timestamp) Notify the given task about a completed checkpoint.
void	releasePartitions(org.apache.flink.api.common.JobID jobId, Set<ResultPartitionID> partitionIds) Batch release intermediate result partitions.
CompletableFuture<TaskBackPressureResponse>	requestTaskBackPressure(ExecutionAttemptID executionAttemptID, int requestId, org.apache.flink.api.common.time.Time timeout) Request the back pressure ratio for the given task.
CompletableFuture<Acknowledge>	submitTask(TaskDeploymentDescriptor tdd, org.apache.flink.api.common.time.Time timeout) Submit a task to the task manager.
void	triggerCheckpoint(ExecutionAttemptID executionAttemptID, org.apache.flink.api.common.JobID jobId, long checkpointId, long timestamp, CheckpointOptions checkpointOptions, boolean advanceToEndOfEventTime) Trigger for the given task a checkpoint.
CompletableFuture<Acknowledge>	updatePartitions(ExecutionAttemptID executionAttemptID, Iterable<PartitionInfo> partitionInfos, org.apache.flink.api.common.time.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.

requestTaskBackPressure

public CompletableFuture<TaskBackPressureResponse> requestTaskBackPressure(ExecutionAttemptID executionAttemptID,
                                                                           int requestId,
                                                                           org.apache.flink.api.common.time.Time timeout)

Description copied from interface: TaskManagerGateway

Request the back pressure ratio for the given task.

Specified by:

requestTaskBackPressure in interface TaskManagerGateway

Parameters:

executionAttemptID - identifying the task to request.

requestId - id of the request.

timeout - rpc request timeout.

Returns:

A future of the task back pressure result.

submitTask

public CompletableFuture<Acknowledge> submitTask(TaskDeploymentDescriptor tdd,
                                                 org.apache.flink.api.common.time.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

cancelTask

public CompletableFuture<Acknowledge> cancelTask(ExecutionAttemptID executionAttemptID,
                                                 org.apache.flink.api.common.time.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,
                                                       org.apache.flink.api.common.time.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

releasePartitions

public void releasePartitions(org.apache.flink.api.common.JobID jobId,
                              Set<ResultPartitionID> partitionIds)

Description copied from interface: TaskManagerGateway

Batch release intermediate result partitions.

Specified by:

releasePartitions in interface TaskManagerGateway

Parameters:

jobId - id of the job that the partitions belong to

partitionIds - partition ids to release

notifyCheckpointComplete

public void notifyCheckpointComplete(ExecutionAttemptID executionAttemptID,
                                     org.apache.flink.api.common.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,
                              org.apache.flink.api.common.JobID jobId,
                              long checkpointId,
                              long timestamp,
                              CheckpointOptions checkpointOptions,
                              boolean advanceToEndOfEventTime)

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

advanceToEndOfEventTime - Flag indicating if the source should inject a MAX_WATERMARK in the pipeline to fire any registered event-time timers

freeSlot

public CompletableFuture<Acknowledge> freeSlot(AllocationID allocationId,
                                               Throwable cause,
                                               org.apache.flink.api.common.time.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