org.apache.flink.runtime.state

Interface KeyedStateBackend<K>

Type Parameters:

K - The key by which state is keyed.

All Superinterfaces:

org.apache.flink.util.Disposable, InternalKeyContext<K>, KeyedStateFactory, PriorityQueueSetFactory

All Known Implementing Classes:

AbstractKeyedStateBackend, HeapKeyedStateBackend

public interface KeyedStateBackend<K>
extends InternalKeyContext<K>, KeyedStateFactory, PriorityQueueSetFactory, org.apache.flink.util.Disposable

A keyed state backend provides methods for managing keyed state.

Method Summary

Modifier and Type	Method and Description
<N,S extends org.apache.flink.api.common.state.State,T> void	applyToAllKeys(N namespace, org.apache.flink.api.common.typeutils.TypeSerializer<N> namespaceSerializer, org.apache.flink.api.common.state.StateDescriptor<S,T> stateDescriptor, KeyedStateFunction<K,S> function) Applies the provided KeyedStateFunction to the state with the provided StateDescriptor of all the currently active keys.
void	dispose()
<N> java.util.stream.Stream<K>	getKeys(String state, N namespace)
<N,S extends org.apache.flink.api.common.state.State,T> S	getOrCreateKeyedState(org.apache.flink.api.common.typeutils.TypeSerializer<N> namespaceSerializer, org.apache.flink.api.common.state.StateDescriptor<S,T> stateDescriptor) Creates or retrieves a keyed state backed by this state backend.
<N,S extends org.apache.flink.api.common.state.State> S	getPartitionedState(N namespace, org.apache.flink.api.common.typeutils.TypeSerializer<N> namespaceSerializer, org.apache.flink.api.common.state.StateDescriptor<S,?> stateDescriptor) Creates or retrieves a partitioned state backed by this state backend.
void	setCurrentKey(K newKey) Sets the current key that is used for partitioned state.

Methods inherited from interface org.apache.flink.runtime.state.heap.InternalKeyContext

getCurrentKey, getCurrentKeyGroupIndex, getKeyGroupRange, getKeySerializer, getNumberOfKeyGroups

Methods inherited from interface org.apache.flink.runtime.state.KeyedStateFactory

createInternalState, createInternalState

Methods inherited from interface org.apache.flink.runtime.state.PriorityQueueSetFactory

create

Method Detail

setCurrentKey

void setCurrentKey(K newKey)

Sets the current key that is used for partitioned state.

Parameters:

newKey - The new current key.

applyToAllKeys

<N,S extends org.apache.flink.api.common.state.State,T> void applyToAllKeys(N namespace,
                                                                            org.apache.flink.api.common.typeutils.TypeSerializer<N> namespaceSerializer,
                                                                            org.apache.flink.api.common.state.StateDescriptor<S,T> stateDescriptor,
                                                                            KeyedStateFunction<K,S> function)
                                                                     throws Exception

Applies the provided KeyedStateFunction to the state with the provided StateDescriptor of all the currently active keys.

Type Parameters:

N - The type of the namespace.

S - The type of the state.

Parameters:

namespace - the namespace of the state.

namespaceSerializer - the serializer for the namespace.

stateDescriptor - the descriptor of the state to which the function is going to be applied.

function - the function to be applied to the keyed state.

Throws:

Exception

getKeys

<N> java.util.stream.Stream<K> getKeys(String state,
                                       N namespace)

Parameters:

state - State variable for which existing keys will be returned.

namespace - Namespace for which existing keys will be returned.

Returns:

A stream of all keys for the given state and namespace. Modifications to the state during iterating over it keys are not supported.

getOrCreateKeyedState

<N,S extends org.apache.flink.api.common.state.State,T> S getOrCreateKeyedState(org.apache.flink.api.common.typeutils.TypeSerializer<N> namespaceSerializer,
                                                                                org.apache.flink.api.common.state.StateDescriptor<S,T> stateDescriptor)
                                                                         throws Exception

Creates or retrieves a keyed state backed by this state backend.

Type Parameters:

N - The type of the namespace.

S - The type of the state.

Parameters:

namespaceSerializer - The serializer used for the namespace type of the state

stateDescriptor - The identifier for the state. This contains name and can create a default state value.

Returns:

A new key/value state backed by this backend.

Throws:

Exception - Exceptions may occur during initialization of the state and should be forwarded.

getPartitionedState

<N,S extends org.apache.flink.api.common.state.State> S getPartitionedState(N namespace,
                                                                            org.apache.flink.api.common.typeutils.TypeSerializer<N> namespaceSerializer,
                                                                            org.apache.flink.api.common.state.StateDescriptor<S,?> stateDescriptor)
                                                                     throws Exception

Creates or retrieves a partitioned state backed by this state backend. TODO: NOTE: This method does a lot of work caching / retrieving states just to update the namespace. This method should be removed for the sake of namespaces being lazily fetched from the keyed state backend, or being set on the state directly.

Type Parameters:

N - The type of the namespace.

S - The type of the state.

Parameters:

stateDescriptor - The identifier for the state. This contains name and can create a default state value.

Returns:

A new key/value state backed by this backend.

Throws:

Exception - Exceptions may occur during initialization of the state and should be forwarded.

dispose

void dispose()

Specified by:

dispose in interface org.apache.flink.util.Disposable