org.apache.flink.runtime.state

Class TaskStateManagerImpl

java.lang.Object

org.apache.flink.runtime.state.TaskStateManagerImpl

All Implemented Interfaces:

CheckpointListener, TaskStateManager

public class TaskStateManagerImpl
extends Object
implements TaskStateManager

This class is the default implementation of TaskStateManager and collaborates with the job manager through CheckpointResponder) as well as a task-manager-local state store. Like this, client code does not have to deal with the differences between remote or local state on recovery because this class handles both cases transparently.

Reported state is tagged by clients so that this class can properly forward to the right receiver for the checkpointed state.

Constructor Summary

Constructors

Constructor and Description
TaskStateManagerImpl(org.apache.flink.api.common.JobID jobId, ExecutionAttemptID executionAttemptID, TaskLocalStateStore localStateStore, JobManagerTaskRestore jobManagerTaskRestore, CheckpointResponder checkpointResponder)

Method Summary

Modifier and Type	Method and Description
LocalRecoveryConfig	createLocalRecoveryConfig() Returns the configuration for local recovery, i.e.
void	notifyCheckpointComplete(long checkpointId) Tracking when local state can be disposed.
PrioritizedOperatorSubtaskState	prioritizedOperatorState(OperatorID operatorID) Returns means to restore previously reported state of an operator running in the owning task.
void	reportTaskStateSnapshots(CheckpointMetaData checkpointMetaData, CheckpointMetrics checkpointMetrics, TaskStateSnapshot acknowledgedState, TaskStateSnapshot localState) Report the state snapshots for the operator instances running in the owning task.

Methods inherited from class java.lang.Object

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

Constructor Detail

TaskStateManagerImpl

public TaskStateManagerImpl(@Nonnull
                            org.apache.flink.api.common.JobID jobId,
                            @Nonnull
                            ExecutionAttemptID executionAttemptID,
                            @Nonnull
                            TaskLocalStateStore localStateStore,
                            @Nullable
                            JobManagerTaskRestore jobManagerTaskRestore,
                            @Nonnull
                            CheckpointResponder checkpointResponder)

Method Detail

reportTaskStateSnapshots

public void reportTaskStateSnapshots(@Nonnull
                                     CheckpointMetaData checkpointMetaData,
                                     @Nonnull
                                     CheckpointMetrics checkpointMetrics,
                                     @Nullable
                                     TaskStateSnapshot acknowledgedState,
                                     @Nullable
                                     TaskStateSnapshot localState)

Description copied from interface: TaskStateManager

Report the state snapshots for the operator instances running in the owning task.

Specified by:

reportTaskStateSnapshots in interface TaskStateManager

Parameters:

checkpointMetaData - meta data from the checkpoint request.

checkpointMetrics - task level metrics for the checkpoint.

acknowledgedState - the reported states to acknowledge to the job manager.

localState - the reported states for local recovery.

prioritizedOperatorState

@Nonnull
public PrioritizedOperatorSubtaskState prioritizedOperatorState(OperatorID operatorID)

Description copied from interface: TaskStateManager

Returns means to restore previously reported state of an operator running in the owning task.

Specified by:

prioritizedOperatorState in interface TaskStateManager

Parameters:

operatorID - the id of the operator for which we request state.

Returns:

Previous state for the operator. The previous state can be empty if the operator had no previous state.

createLocalRecoveryConfig

@Nonnull
public LocalRecoveryConfig createLocalRecoveryConfig()

Description copied from interface: TaskStateManager

Returns the configuration for local recovery, i.e. the base directories for all file-based local state of the owning subtask and the general mode for local recovery.

Specified by:

createLocalRecoveryConfig in interface TaskStateManager

notifyCheckpointComplete

public void notifyCheckpointComplete(long checkpointId)
                              throws Exception

Tracking when local state can be disposed.

Specified by:

notifyCheckpointComplete in interface CheckpointListener

Parameters:

checkpointId - The ID of the checkpoint that has been completed.

Throws:

Exception