org.apache.flink.runtime.state

Class TaskLocalStateStoreImpl

java.lang.Object

org.apache.flink.runtime.state.TaskLocalStateStoreImpl

All Implemented Interfaces:

OwnedTaskLocalStateStore, TaskLocalStateStore

public class TaskLocalStateStoreImpl
extends Object
implements OwnedTaskLocalStateStore

Main implementation of a TaskLocalStateStore.

Constructor Summary

Constructors

Constructor and Description
TaskLocalStateStoreImpl(JobID jobID, AllocationID allocationID, JobVertexID jobVertexID, int subtaskIndex, LocalRecoveryConfig localRecoveryConfig, Executor discardExecutor)

Method Summary

Modifier and Type	Method and Description
void	abortCheckpoint(long abortedCheckpointId) Notifies that the checkpoint with the given id was confirmed as aborted.
void	confirmCheckpoint(long confirmedCheckpointId) Notifies that the checkpoint with the given id was confirmed as complete.
CompletableFuture<Void>	dispose() Disposes the state of all local snapshots managed by this object.
LocalRecoveryConfig	getLocalRecoveryConfig() Returns the LocalRecoveryConfig for this task local state store.
void	pruneMatchingCheckpoints(java.util.function.LongPredicate matcher) Remove all checkpoints from the store that match the given predicate.
TaskStateSnapshot	retrieveLocalState(long checkpointID) Returns the local state that is stored under the given checkpoint id or null if nothing was stored under the id.
void	storeLocalState(long checkpointId, TaskStateSnapshot localState) Stores the local state for the given checkpoint id.
String	toString()

Methods inherited from class java.lang.Object

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

Constructor Detail

TaskLocalStateStoreImpl

public TaskLocalStateStoreImpl(@Nonnull
                               JobID jobID,
                               @Nonnull
                               AllocationID allocationID,
                               @Nonnull
                               JobVertexID jobVertexID,
                               @Nonnegative
                               int subtaskIndex,
                               @Nonnull
                               LocalRecoveryConfig localRecoveryConfig,
                               @Nonnull
                               Executor discardExecutor)

Method Detail

storeLocalState

public void storeLocalState(@Nonnegative
                            long checkpointId,
                            @Nullable
                            TaskStateSnapshot localState)

Description copied from interface: TaskLocalStateStore

Stores the local state for the given checkpoint id.

Specified by:

storeLocalState in interface TaskLocalStateStore

Parameters:

checkpointId - id for the checkpoint that created the local state that will be stored.

localState - the local state to store.

retrieveLocalState

@Nullable
public TaskStateSnapshot retrieveLocalState(long checkpointID)

Description copied from interface: TaskLocalStateStore

Returns the local state that is stored under the given checkpoint id or null if nothing was stored under the id.

Specified by:

retrieveLocalState in interface TaskLocalStateStore

Parameters:

checkpointID - the checkpoint id by which we search for local state.

Returns:

the local state found for the given checkpoint id. Can be null

getLocalRecoveryConfig

@Nonnull
public LocalRecoveryConfig getLocalRecoveryConfig()

Description copied from interface: TaskLocalStateStore

Returns the LocalRecoveryConfig for this task local state store.

Specified by:

getLocalRecoveryConfig in interface TaskLocalStateStore

confirmCheckpoint

public void confirmCheckpoint(long confirmedCheckpointId)

Description copied from interface: TaskLocalStateStore

Notifies that the checkpoint with the given id was confirmed as complete. This prunes the checkpoint history and removes all local states with a checkpoint id that is smaller than the newly confirmed checkpoint id.

Specified by:

confirmCheckpoint in interface TaskLocalStateStore

abortCheckpoint

public void abortCheckpoint(long abortedCheckpointId)

Description copied from interface: TaskLocalStateStore

Notifies that the checkpoint with the given id was confirmed as aborted. This prunes the checkpoint history and removes states with a checkpoint id that is equal to the newly aborted checkpoint id.

Specified by:

abortCheckpoint in interface TaskLocalStateStore

pruneMatchingCheckpoints

public void pruneMatchingCheckpoints(@Nonnull
                                     java.util.function.LongPredicate matcher)

Description copied from interface: TaskLocalStateStore

Remove all checkpoints from the store that match the given predicate.

Specified by:

pruneMatchingCheckpoints in interface TaskLocalStateStore

Parameters:

matcher - the predicate that selects the checkpoints for pruning.

dispose

public CompletableFuture<Void> dispose()

Disposes the state of all local snapshots managed by this object.

Specified by:

dispose in interface OwnedTaskLocalStateStore

toString

public String toString()

Overrides:

toString in class Object