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]¶

finalize()¶

prefetch()¶

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]¶

commit()[source]¶

finalize()¶

static for_spec(state_spec, state_tag, current_value_accessor)¶

prefetch()¶

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]¶

commit()¶

finalize()¶

static for_spec(state_spec, state_tag, current_value_accessor)¶

prefetch()¶

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]¶

commit()¶

finalize()¶

static for_spec(state_spec, state_tag, current_value_accessor)¶

prefetch()¶

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]¶

commit()¶

static for_spec(state_spec, state_tag, current_value_accessor)¶

prefetch()¶

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]¶