org.apache.flink.runtime.taskexecutor

Class TaskExecutor

java.lang.Object

org.apache.flink.runtime.rpc.RpcEndpoint

org.apache.flink.runtime.taskexecutor.TaskExecutor

All Implemented Interfaces:

AutoCloseable, RpcGateway, TaskExecutorGateway, org.apache.flink.util.AutoCloseableAsync

public class TaskExecutor
extends RpcEndpoint
implements TaskExecutorGateway

TaskExecutor implementation. The task executor is responsible for the execution of multiple Task.

Nested Class Summary

Nested classes/interfaces inherited from class org.apache.flink.runtime.rpc.RpcEndpoint

RpcEndpoint.MainThreadExecutor

Field Summary

Fields

Modifier and Type	Field and Description
static String	TASK_MANAGER_NAME

Fields inherited from class org.apache.flink.runtime.rpc.RpcEndpoint

log, rpcServer

Constructor Summary

Constructors

Constructor and Description
TaskExecutor(RpcService rpcService, TaskManagerConfiguration taskManagerConfiguration, HighAvailabilityServices haServices, TaskManagerServices taskExecutorServices, HeartbeatServices heartbeatServices, TaskManagerMetricGroup taskManagerMetricGroup, String metricQueryServicePath, BlobCacheService blobCacheService, FatalErrorHandler fatalErrorHandler)

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>	confirmCheckpoint(ExecutionAttemptID executionAttemptID, long checkpointId, long checkpointTimestamp) Confirm a checkpoint for the given task.
void	disconnectJobManager(org.apache.flink.api.common.JobID jobId, Exception cause) Disconnects the given JobManager from the TaskManager.
void	disconnectResourceManager(Exception cause) Disconnects the ResourceManager from the TaskManager.
void	failPartition(ExecutionAttemptID executionAttemptID) Fail all intermediate result partitions of 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.
ResourceID	getResourceID()
void	heartbeatFromJobManager(ResourceID resourceID) Heartbeat request from the job manager.
void	heartbeatFromResourceManager(ResourceID resourceID) Heartbeat request from the resource manager.
void	onStart() User overridable callback.
CompletableFuture<Void>	onStop() Called to shut down the TaskManager.
CompletableFuture<TransientBlobKey>	requestFileUpload(FileType fileType, org.apache.flink.api.common.time.Time timeout) Requests the file upload of the specified type to the cluster's BlobServer.
CompletableFuture<org.apache.flink.types.SerializableOptional<String>>	requestMetricQueryServiceAddress(org.apache.flink.api.common.time.Time timeout) Returns the fully qualified address of Metric Query Service on the TaskManager.
CompletableFuture<Acknowledge>	requestSlot(SlotID slotId, org.apache.flink.api.common.JobID jobId, AllocationID allocationId, String targetAddress, ResourceManagerId resourceManagerId, org.apache.flink.api.common.time.Time timeout) Requests a slot from the TaskManager.
CompletableFuture<StackTraceSampleResponse>	requestStackTraceSample(ExecutionAttemptID executionAttemptId, int sampleId, int numSamples, org.apache.flink.api.common.time.Time delayBetweenSamples, int maxStackTraceDepth, org.apache.flink.api.common.time.Time timeout)
CompletableFuture<Acknowledge>	stopTask(ExecutionAttemptID executionAttemptID, org.apache.flink.api.common.time.Time timeout) Stop the given task.
CompletableFuture<Acknowledge>	submitTask(TaskDeploymentDescriptor tdd, JobMasterId jobMasterId, org.apache.flink.api.common.time.Time timeout) Submit a Task to the TaskExecutor.
CompletableFuture<Acknowledge>	triggerCheckpoint(ExecutionAttemptID executionAttemptID, long checkpointId, long checkpointTimestamp, CheckpointOptions checkpointOptions) Trigger the checkpoint for the given task.
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 org.apache.flink.runtime.rpc.RpcEndpoint

callAsync, closeAsync, getAddress, getEndpointId, getHostname, getMainThreadExecutor, getRpcService, getSelfGateway, getTerminationFuture, runAsync, scheduleRunAsync, scheduleRunAsync, start, stop, validateRunsInMainThread

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.flink.runtime.rpc.RpcGateway

getAddress, getHostname

Methods inherited from interface org.apache.flink.util.AutoCloseableAsync

close

Field Detail

TASK_MANAGER_NAME

public static final String TASK_MANAGER_NAME

See Also:

Constant Field Values

Constructor Detail

TaskExecutor

public TaskExecutor(RpcService rpcService,
                    TaskManagerConfiguration taskManagerConfiguration,
                    HighAvailabilityServices haServices,
                    TaskManagerServices taskExecutorServices,
                    HeartbeatServices heartbeatServices,
                    TaskManagerMetricGroup taskManagerMetricGroup,
                    @Nullable
                    String metricQueryServicePath,
                    BlobCacheService blobCacheService,
                    FatalErrorHandler fatalErrorHandler)

Method Detail

onStart

public void onStart()
             throws Exception

Description copied from class: RpcEndpoint

User overridable callback.

This method is called when the RpcEndpoint is being started. The method is guaranteed to be executed in the main thread context and can be used to start the rpc endpoint in the context of the rpc endpoint's main thread.

IMPORTANT: This method should never be called directly by the user.

Overrides:

onStart in class RpcEndpoint

Throws:

Exception - indicating that the rpc endpoint could not be started. If an exception occurs, then the rpc endpoint will automatically terminate.

onStop

public CompletableFuture<Void> onStop()

Called to shut down the TaskManager. The method closes all TaskManager services.

Overrides:

onStop in class RpcEndpoint

Returns:

Future which is completed once all post stop actions are completed. If an error occurs this future is completed exceptionally

requestStackTraceSample

public CompletableFuture<StackTraceSampleResponse> requestStackTraceSample(ExecutionAttemptID executionAttemptId,
                                                                           int sampleId,
                                                                           int numSamples,
                                                                           org.apache.flink.api.common.time.Time delayBetweenSamples,
                                                                           int maxStackTraceDepth,
                                                                           org.apache.flink.api.common.time.Time timeout)

Specified by:

requestStackTraceSample in interface TaskExecutorGateway

submitTask

public CompletableFuture<Acknowledge> submitTask(TaskDeploymentDescriptor tdd,
                                                 JobMasterId jobMasterId,
                                                 org.apache.flink.api.common.time.Time timeout)

Description copied from interface: TaskExecutorGateway

Submit a Task to the TaskExecutor.

Specified by:

submitTask in interface TaskExecutorGateway

Parameters:

tdd - describing the task to submit

jobMasterId - identifying the submitting JobMaster

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: TaskExecutorGateway

Cancel the given task.

Specified by:

cancelTask in interface TaskExecutorGateway

Parameters:

executionAttemptID - identifying the task

timeout - for the cancel operation

Returns:

Future acknowledge if the task is successfully canceled

stopTask

public CompletableFuture<Acknowledge> stopTask(ExecutionAttemptID executionAttemptID,
                                               org.apache.flink.api.common.time.Time timeout)

Description copied from interface: TaskExecutorGateway

Stop the given task.

Specified by:

stopTask in interface TaskExecutorGateway

Parameters:

executionAttemptID - identifying the task

timeout - for the stop operation

Returns:

Future acknowledge if the task is successfully stopped

updatePartitions

public CompletableFuture<Acknowledge> updatePartitions(ExecutionAttemptID executionAttemptID,
                                                       Iterable<PartitionInfo> partitionInfos,
                                                       org.apache.flink.api.common.time.Time timeout)

Description copied from interface: TaskExecutorGateway

Update the task where the given partitions can be found.

Specified by:

updatePartitions in interface TaskExecutorGateway

Parameters:

executionAttemptID - identifying the task

partitionInfos - telling where the partition can be retrieved from

timeout - for the update partitions operation

Returns:

Future acknowledge if the partitions have been successfully updated

failPartition

public void failPartition(ExecutionAttemptID executionAttemptID)

Description copied from interface: TaskExecutorGateway

Fail all intermediate result partitions of the given task.

Specified by:

failPartition in interface TaskExecutorGateway

Parameters:

executionAttemptID - identifying the task

heartbeatFromJobManager

public void heartbeatFromJobManager(ResourceID resourceID)

Description copied from interface: TaskExecutorGateway

Heartbeat request from the job manager.

Specified by:

heartbeatFromJobManager in interface TaskExecutorGateway

Parameters:

resourceID - unique id of the job manager

heartbeatFromResourceManager

public void heartbeatFromResourceManager(ResourceID resourceID)

Description copied from interface: TaskExecutorGateway

Heartbeat request from the resource manager.

Specified by:

heartbeatFromResourceManager in interface TaskExecutorGateway

Parameters:

resourceID - unique id of the resource manager

triggerCheckpoint

public CompletableFuture<Acknowledge> triggerCheckpoint(ExecutionAttemptID executionAttemptID,
                                                        long checkpointId,
                                                        long checkpointTimestamp,
                                                        CheckpointOptions checkpointOptions)

Description copied from interface: TaskExecutorGateway

Trigger the checkpoint for the given task. The checkpoint is identified by the checkpoint ID and the checkpoint timestamp.

Specified by:

triggerCheckpoint in interface TaskExecutorGateway

Parameters:

executionAttemptID - identifying the task

checkpointId - unique id for the checkpoint

checkpointTimestamp - is the timestamp when the checkpoint has been initiated

checkpointOptions - for performing the checkpoint

Returns:

Future acknowledge if the checkpoint has been successfully triggered

confirmCheckpoint

public CompletableFuture<Acknowledge> confirmCheckpoint(ExecutionAttemptID executionAttemptID,
                                                        long checkpointId,
                                                        long checkpointTimestamp)

Description copied from interface: TaskExecutorGateway

Confirm a checkpoint for the given task. The checkpoint is identified by the checkpoint ID and the checkpoint timestamp.

Specified by:

confirmCheckpoint in interface TaskExecutorGateway

Parameters:

executionAttemptID - identifying the task

checkpointId - unique id for the checkpoint

checkpointTimestamp - is the timestamp when the checkpoint has been initiated

Returns:

Future acknowledge if the checkpoint has been successfully confirmed

requestSlot

public CompletableFuture<Acknowledge> requestSlot(SlotID slotId,
                                                  org.apache.flink.api.common.JobID jobId,
                                                  AllocationID allocationId,
                                                  String targetAddress,
                                                  ResourceManagerId resourceManagerId,
                                                  org.apache.flink.api.common.time.Time timeout)

Description copied from interface: TaskExecutorGateway

Requests a slot from the TaskManager.

Specified by:

requestSlot in interface TaskExecutorGateway

Parameters:

slotId - slot id for the request

jobId - for which to request a slot

allocationId - id for the request

targetAddress - to which to offer the requested slots

resourceManagerId - current leader id of the ResourceManager

timeout - for the operation

Returns:

answer to the slot request

freeSlot

public CompletableFuture<Acknowledge> freeSlot(AllocationID allocationId,
                                               Throwable cause,
                                               org.apache.flink.api.common.time.Time timeout)

Description copied from interface: TaskExecutorGateway

Frees the slot with the given allocation ID.

Specified by:

freeSlot in interface TaskExecutorGateway

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

requestFileUpload

public CompletableFuture<TransientBlobKey> requestFileUpload(FileType fileType,
                                                             org.apache.flink.api.common.time.Time timeout)

Description copied from interface: TaskExecutorGateway

Requests the file upload of the specified type to the cluster's BlobServer.

Specified by:

requestFileUpload in interface TaskExecutorGateway

Parameters:

fileType - to upload

timeout - for the asynchronous operation

Returns:

Future which is completed with the TransientBlobKey of the uploaded file.

requestMetricQueryServiceAddress

public CompletableFuture<org.apache.flink.types.SerializableOptional<String>> requestMetricQueryServiceAddress(org.apache.flink.api.common.time.Time timeout)

Description copied from interface: TaskExecutorGateway

Returns the fully qualified address of Metric Query Service on the TaskManager.

Specified by:

requestMetricQueryServiceAddress in interface TaskExecutorGateway

Returns:

Future String with Fully qualified (RPC) address of Metric Query Service on the TaskManager.

disconnectJobManager

public void disconnectJobManager(org.apache.flink.api.common.JobID jobId,
                                 Exception cause)

Description copied from interface: TaskExecutorGateway

Disconnects the given JobManager from the TaskManager.

Specified by:

disconnectJobManager in interface TaskExecutorGateway

Parameters:

jobId - JobID for which the JobManager was the leader

cause - for the disconnection from the JobManager

disconnectResourceManager

public void disconnectResourceManager(Exception cause)

Description copied from interface: TaskExecutorGateway

Disconnects the ResourceManager from the TaskManager.

Specified by:

disconnectResourceManager in interface TaskExecutorGateway

Parameters:

cause - for the disconnection from the ResourceManager

getResourceID

public ResourceID getResourceID()