org.apache.flink.streaming.api.runners.python.beam.state

Class BeamStateRequestHandler

java.lang.Object

org.apache.flink.streaming.api.runners.python.beam.state.BeamStateRequestHandler

All Implemented Interfaces:

org.apache.beam.runners.fnexecution.state.StateRequestHandler

public class BeamStateRequestHandler
extends Object
implements org.apache.beam.runners.fnexecution.state.StateRequestHandler

The handler for Beam state requests sent from Python side, which does actual operations on Flink state.

Constructor Summary

Constructors

Constructor and Description
BeamStateRequestHandler(BeamStateStore keyedStateStore, BeamStateStore operatorStateStore, BeamStateHandler<ListState<byte[]>> bagStateHandler, BeamStateHandler<MapState<ByteArrayWrapper,byte[]>> mapStateHandler)

Method Summary

Modifier and Type	Method and Description
Iterable<org.apache.beam.model.fnexecution.v1.BeamFnApi.ProcessBundleRequest.CacheToken>	getCacheTokens()
CompletionStage<org.apache.beam.model.fnexecution.v1.BeamFnApi.StateResponse.Builder>	handle(org.apache.beam.model.fnexecution.v1.BeamFnApi.StateRequest request)
static BeamStateRequestHandler	of(KeyedStateBackend<?> keyedStateBackend, OperatorStateBackend operatorStateBackend, TypeSerializer<?> keySerializer, TypeSerializer<?> namespaceSerializer, ReadableConfig config) Create a BeamStateRequestHandler.

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.beam.runners.fnexecution.state.StateRequestHandler

unsupported

Constructor Detail

BeamStateRequestHandler

public BeamStateRequestHandler(BeamStateStore keyedStateStore,
                               BeamStateStore operatorStateStore,
                               BeamStateHandler<ListState<byte[]>> bagStateHandler,
                               BeamStateHandler<MapState<ByteArrayWrapper,byte[]>> mapStateHandler)

Method Detail

handle

public CompletionStage<org.apache.beam.model.fnexecution.v1.BeamFnApi.StateResponse.Builder> handle(org.apache.beam.model.fnexecution.v1.BeamFnApi.StateRequest request)
                                                                                             throws Exception

Specified by:

handle in interface org.apache.beam.runners.fnexecution.state.StateRequestHandler

Throws:

Exception

getCacheTokens

public Iterable<org.apache.beam.model.fnexecution.v1.BeamFnApi.ProcessBundleRequest.CacheToken> getCacheTokens()

Specified by:

getCacheTokens in interface org.apache.beam.runners.fnexecution.state.StateRequestHandler

of

public static BeamStateRequestHandler of(@Nullable
                                         KeyedStateBackend<?> keyedStateBackend,
                                         @Nullable
                                         OperatorStateBackend operatorStateBackend,
                                         @Nullable
                                         TypeSerializer<?> keySerializer,
                                         @Nullable
                                         TypeSerializer<?> namespaceSerializer,
                                         ReadableConfig config)

Create a BeamStateRequestHandler.

Parameters:

keyedStateBackend - if null, BeamStateRequestHandler would throw an error when receive keyed-state requests.

operatorStateBackend - if null, BeamStateRequestHandler would throw an error when receive operator-state requests.

keySerializer - key serializer for KeyedStateBackend, must not be null if keyedStatedBackend is not null.

namespaceSerializer - namespace serializer for KeyedStateBackend, could be null when there's no window logic involved.

config - state-related configurations

Returns:

A new BeamBagStateHandler