org.apache.beam.runners.fnexecution.translation

Class BatchSideInputHandlerFactory

java.lang.Object

org.apache.beam.runners.fnexecution.translation.BatchSideInputHandlerFactory

All Implemented Interfaces:

StateRequestHandlers.SideInputHandlerFactory

public class BatchSideInputHandlerFactory
extends java.lang.Object
implements StateRequestHandlers.SideInputHandlerFactory

StateRequestHandler that uses a BatchSideInputHandlerFactory.SideInputGetter to access side inputs.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static interface	BatchSideInputHandlerFactory.SideInputGetter Returns the value for the side input with the given PCollection id from the runner.

Method Summary

Modifier and Type	Method and Description
<V,W extends BoundedWindow> StateRequestHandlers.IterableSideInputHandler<V,W>	forIterableSideInput(java.lang.String transformId, java.lang.String sideInputId, Coder<V> elementCoder, Coder<W> windowCoder) Returns an StateRequestHandlers.IterableSideInputHandler for the given pTransformId, sideInputId.
<K,V,W extends BoundedWindow> StateRequestHandlers.MultimapSideInputHandler<K,V,W>	forMultimapSideInput(java.lang.String transformId, java.lang.String sideInputId, KvCoder<K,V> elementCoder, Coder<W> windowCoder) Returns a StateRequestHandlers.MultimapSideInputHandler for the given pTransformId, sideInputId.
static BatchSideInputHandlerFactory	forStage(org.apache.beam.runners.core.construction.graph.ExecutableStage stage, BatchSideInputHandlerFactory.SideInputGetter sideInputGetter) Creates a new state handler for the given stage.

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.StateRequestHandlers.SideInputHandlerFactory

unsupported

Method Detail

forStage

public static BatchSideInputHandlerFactory forStage(org.apache.beam.runners.core.construction.graph.ExecutableStage stage,
                                                    BatchSideInputHandlerFactory.SideInputGetter sideInputGetter)

Creates a new state handler for the given stage. Note that this requires a traversal of the stage itself, so this should only be called once per stage rather than once per bundle.

forIterableSideInput

public <V,W extends BoundedWindow> StateRequestHandlers.IterableSideInputHandler<V,W> forIterableSideInput(java.lang.String transformId,
                                                                                                           java.lang.String sideInputId,
                                                                                                           Coder<V> elementCoder,
                                                                                                           Coder<W> windowCoder)

Description copied from interface: StateRequestHandlers.SideInputHandlerFactory

Returns an StateRequestHandlers.IterableSideInputHandler for the given pTransformId, sideInputId. The supplied elementCoder and windowCoder should be used to encode/decode their respective values.

Specified by:

forIterableSideInput in interface StateRequestHandlers.SideInputHandlerFactory

forMultimapSideInput

public <K,V,W extends BoundedWindow> StateRequestHandlers.MultimapSideInputHandler<K,V,W> forMultimapSideInput(java.lang.String transformId,
                                                                                                               java.lang.String sideInputId,
                                                                                                               KvCoder<K,V> elementCoder,
                                                                                                               Coder<W> windowCoder)

Description copied from interface: StateRequestHandlers.SideInputHandlerFactory

Returns a StateRequestHandlers.MultimapSideInputHandler for the given pTransformId, sideInputId. The supplied elementCoder and windowCoder should be used to encode/decode their respective values.

Specified by:

forMultimapSideInput in interface StateRequestHandlers.SideInputHandlerFactory