apache_beam.runners.direct.direct_userstate module

Support for user state in the BundleBasedDirectRunner.

class apache_beam.runners.direct.direct_userstate.DirectRuntimeState(state_spec, state_tag, current_value_accessor)[source]

Bases: apache_beam.transforms.userstate.RuntimeState

static for_spec(state_spec, state_tag, current_value_accessor)[source]
class apache_beam.runners.direct.direct_userstate.ReadModifyWriteRuntimeState(state_spec, state_tag, current_value_accessor)[source]

Bases: apache_beam.runners.direct.direct_userstate.DirectRuntimeState, apache_beam.transforms.userstate.ReadModifyWriteRuntimeState

read()[source]
write(value)[source]
clear()[source]
is_cleared()[source]
is_modified()[source]
class apache_beam.runners.direct.direct_userstate.BagRuntimeState(state_spec, state_tag, current_value_accessor)[source]

Bases: apache_beam.runners.direct.direct_userstate.DirectRuntimeState, apache_beam.transforms.userstate.BagRuntimeState

read()[source]
add(value)[source]
clear()[source]
class apache_beam.runners.direct.direct_userstate.SetRuntimeState(state_spec, state_tag, current_value_accessor)[source]

Bases: apache_beam.runners.direct.direct_userstate.DirectRuntimeState, apache_beam.transforms.userstate.SetRuntimeState

read()[source]
add(value)[source]
clear()[source]
is_modified()[source]
class apache_beam.runners.direct.direct_userstate.CombiningValueRuntimeState(state_spec, state_tag, current_value_accessor)[source]

Bases: apache_beam.runners.direct.direct_userstate.DirectRuntimeState, apache_beam.transforms.userstate.CombiningValueRuntimeState

Combining value state interface object passed to user code.

read()[source]
add(value)[source]
clear()[source]
finalize()[source]
class apache_beam.runners.direct.direct_userstate.DirectUserStateContext(step_context, dofn, key_coder)[source]

Bases: apache_beam.transforms.userstate.UserStateContext

userstate.UserStateContext for the BundleBasedDirectRunner.

The DirectUserStateContext buffers up updates that are to be committed by the TransformEvaluator after running a DoFn.

get_timer(timer_spec: apache_beam.transforms.userstate.TimerSpec, key, window, timestamp, pane) → apache_beam.transforms.userstate.RuntimeTimer[source]
get_state(state_spec, key, window)[source]
commit()[source]
reset()[source]