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)

Bases: apache_beam.transforms.userstate.RuntimeState

static for_spec(state_spec, state_tag, current_value_accessor)

class apache_beam.runners.direct.direct_userstate.ReadModifyWriteRuntimeState(state_spec, state_tag, current_value_accessor)

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

read()

write(value)

clear()

is_cleared()

is_modified()

class apache_beam.runners.direct.direct_userstate.BagRuntimeState(state_spec, state_tag, current_value_accessor)

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

read()

add(value)

clear()

class apache_beam.runners.direct.direct_userstate.SetRuntimeState(state_spec, state_tag, current_value_accessor)

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

read()

add(value)

clear()

is_modified()

class apache_beam.runners.direct.direct_userstate.CombiningValueRuntimeState(state_spec, state_tag, current_value_accessor)

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

Combining value state interface object passed to user code.

read()

add(value)

clear()

finalize()

class apache_beam.runners.direct.direct_userstate.DirectUserStateContext(step_context, dofn, key_coder)

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

get_state(state_spec, key, window)

commit()

reset()