Package org.apache.beam.runners.fnexecution.state

Class InMemoryBagUserStateFactory<K,V,W extends org.apache.beam.sdk.transforms.windowing.BoundedWindow>

java.lang.Object

org.apache.beam.runners.fnexecution.state.InMemoryBagUserStateFactory<K,V,W>

All Implemented Interfaces:

StateRequestHandlers.BagUserStateHandlerFactory<K,V,W>

public class InMemoryBagUserStateFactory<K,V,W extends org.apache.beam.sdk.transforms.windowing.BoundedWindow>
extends java.lang.Object
implements StateRequestHandlers.BagUserStateHandlerFactory<K,V,W>

Holds user state in memory. Only one key is active at a time due to the GroupReduceFunction being called once per key. Needs to be reset via resetForNewKey() before processing a new key.

Constructor Summary

Constructors

Constructor	Description
InMemoryBagUserStateFactory()

Method Summary

Modifier and Type	Method	Description
StateRequestHandlers.BagUserStateHandler<K,V,W>	forUserState(java.lang.String pTransformId, java.lang.String userStateId, org.apache.beam.sdk.coders.Coder<K> keyCoder, org.apache.beam.sdk.coders.Coder<V> valueCoder, org.apache.beam.sdk.coders.Coder<W> windowCoder)
void	resetForNewKey()	Prepares previous emitted state handlers for processing a new key.

Methods inherited from class java.lang.Object

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

Constructor Detail

InMemoryBagUserStateFactory

public InMemoryBagUserStateFactory()

Method Detail

forUserState

public StateRequestHandlers.BagUserStateHandler<K,V,W> forUserState(java.lang.String pTransformId,
                                                                                java.lang.String userStateId,
                                                                                org.apache.beam.sdk.coders.Coder<K> keyCoder,
                                                                                org.apache.beam.sdk.coders.Coder<V> valueCoder,
                                                                                org.apache.beam.sdk.coders.Coder<W> windowCoder)

Specified by:

forUserState in interface StateRequestHandlers.BagUserStateHandlerFactory<K,V,W extends org.apache.beam.sdk.transforms.windowing.BoundedWindow>

resetForNewKey

public void resetForNewKey()

Prepares previous emitted state handlers for processing a new key.