org.apache.flink.streaming.api.operators

Class StreamOperatorStateHandler

java.lang.Object

org.apache.flink.streaming.api.operators.StreamOperatorStateHandler

@Internal
public class StreamOperatorStateHandler
extends Object

Class encapsulating various state backend handling logic for StreamOperator implementations.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static interface	StreamOperatorStateHandler.CheckpointedStreamOperator Custom state handling hooks to be invoked by StreamOperatorStateHandler.

Field Summary

Fields

Modifier and Type	Field and Description
protected static org.slf4j.Logger	LOG

Constructor Summary

Constructors

Constructor and Description
StreamOperatorStateHandler(StreamOperatorStateContext context, ExecutionConfig executionConfig, CloseableRegistry closeableRegistry)

Method Summary

Modifier and Type	Method and Description
void	dispose()
AsyncKeyedStateBackend	getAsyncKeyedStateBackend()
Object	getCurrentKey()
<K> KeyedStateBackend<K>	getKeyedStateBackend()
Optional<KeyedStateStore>	getKeyedStateStore()
Optional<KeyedStateStoreV2>	getKeyedStateStoreV2()
OperatorStateBackend	getOperatorStateBackend()
<N,S extends State,T> S	getOrCreateKeyedState(TypeSerializer<N> namespaceSerializer, StateDescriptor<S,T> stateDescriptor)
protected <S extends State,N> S	getPartitionedState(N namespace, TypeSerializer<N> namespaceSerializer, StateDescriptor<S,?> stateDescriptor) Creates a partitioned state handle, using the state backend configured for this task.
void	initializeOperatorState(StreamOperatorStateHandler.CheckpointedStreamOperator streamOperator)
void	notifyCheckpointAborted(long checkpointId)
void	notifyCheckpointComplete(long checkpointId)
static SnapshotStrategyRunner<KeyedStateHandle,? extends FullSnapshotResources<?>>	prepareCanonicalSavepoint(CheckpointableKeyedStateBackend<?> keyedStateBackend, CloseableRegistry closeableRegistry)
void	setCurrentKey(Object key)
OperatorSnapshotFutures	snapshotState(StreamOperatorStateHandler.CheckpointedStreamOperator streamOperator, Optional<InternalTimeServiceManager<?>> timeServiceManager, String operatorName, long checkpointId, long timestamp, CheckpointOptions checkpointOptions, CheckpointStreamFactory factory, boolean isUsingCustomRawKeyedState)

Methods inherited from class java.lang.Object

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

Field Detail

LOG

protected static final org.slf4j.Logger LOG

Constructor Detail

StreamOperatorStateHandler

public StreamOperatorStateHandler(StreamOperatorStateContext context,
                                  ExecutionConfig executionConfig,
                                  CloseableRegistry closeableRegistry)

Method Detail

initializeOperatorState

public void initializeOperatorState(StreamOperatorStateHandler.CheckpointedStreamOperator streamOperator)
                             throws Exception

Throws:

Exception

dispose

public void dispose()
             throws Exception

Throws:

Exception

snapshotState

public OperatorSnapshotFutures snapshotState(StreamOperatorStateHandler.CheckpointedStreamOperator streamOperator,
                                             Optional<InternalTimeServiceManager<?>> timeServiceManager,
                                             String operatorName,
                                             long checkpointId,
                                             long timestamp,
                                             CheckpointOptions checkpointOptions,
                                             CheckpointStreamFactory factory,
                                             boolean isUsingCustomRawKeyedState)
                                      throws CheckpointException

Throws:

CheckpointException

prepareCanonicalSavepoint

@Nonnull
public static SnapshotStrategyRunner<KeyedStateHandle,? extends FullSnapshotResources<?>> prepareCanonicalSavepoint(CheckpointableKeyedStateBackend<?> keyedStateBackend,
                                                                                                                             CloseableRegistry closeableRegistry)
                                                                                                                      throws Exception

Throws:

Exception

notifyCheckpointComplete

public void notifyCheckpointComplete(long checkpointId)
                              throws Exception

Throws:

Exception

notifyCheckpointAborted

public void notifyCheckpointAborted(long checkpointId)
                             throws Exception

Throws:

Exception

getKeyedStateBackend

public <K> KeyedStateBackend<K> getKeyedStateBackend()

getAsyncKeyedStateBackend

@Nullable
public AsyncKeyedStateBackend getAsyncKeyedStateBackend()

getOperatorStateBackend

public OperatorStateBackend getOperatorStateBackend()

getOrCreateKeyedState

public <N,S extends State,T> S getOrCreateKeyedState(TypeSerializer<N> namespaceSerializer,
                                                     StateDescriptor<S,T> stateDescriptor)
                                              throws Exception

Throws:

Exception

getPartitionedState

protected <S extends State,N> S getPartitionedState(N namespace,
                                                    TypeSerializer<N> namespaceSerializer,
                                                    StateDescriptor<S,?> stateDescriptor)
                                             throws Exception

Creates a partitioned state handle, using the state backend configured for this task.

Throws:

IllegalStateException - Thrown, if the key/value state was already initialized.

Exception - Thrown, if the state backend cannot create the key/value state.

setCurrentKey

public void setCurrentKey(Object key)

getCurrentKey

public Object getCurrentKey()

getKeyedStateStore

public Optional<KeyedStateStore> getKeyedStateStore()

getKeyedStateStoreV2

public Optional<KeyedStateStoreV2> getKeyedStateStoreV2()