com.google.cloud.dataflow.sdk.util

Class BaseExecutionContext

java.lang.Object

com.google.cloud.dataflow.sdk.util.BaseExecutionContext

All Implemented Interfaces:

ExecutionContext

Direct Known Subclasses:

BatchModeExecutionContext

public abstract class BaseExecutionContext
extends Object
implements ExecutionContext

Base class for implementations of ExecutionContext.

A concrete subclass should implement createStepContext(java.lang.String, java.lang.String) to create the appropriate ExecutionContext.StepContext implementation. Any StepContext created will be cached for the lifetime of this ExecutionContext.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	BaseExecutionContext.StepContext Base class for implementations of ExecutionContext.StepContext.

Constructor Summary

Constructors

Constructor and Description
BaseExecutionContext()

Method Summary

Modifier and Type	Method and Description
protected abstract ExecutionContext.StepContext	createStepContext(String stepName, String transformName) Implementations should override this to create the specific type of BaseExecutionContext.StepContext they need.
Collection<ExecutionContext.StepContext>	getAllStepContexts() Returns a collection view of all of the BaseExecutionContext.StepContexts.
ExecutionContext.StepContext	getStepContext(String stepName, String transformName) Returns the BaseExecutionContext.StepContext associated with the given step.
void	noteOutput(WindowedValue<?> output) Hook for subclasses to implement that will be called whenever DoFn.Context.output(OutputT) is called.
void	noteSideOutput(TupleTag<?> tag, WindowedValue<?> output) Hook for subclasses to implement that will be called whenever DoFn.Context.sideOutput(com.google.cloud.dataflow.sdk.values.TupleTag<T>, T) is called.

Methods inherited from class java.lang.Object

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

Constructor Detail

BaseExecutionContext

public BaseExecutionContext()

Method Detail

createStepContext

protected abstract ExecutionContext.StepContext createStepContext(String stepName,
                                                                  String transformName)

Implementations should override this to create the specific type of BaseExecutionContext.StepContext they need.

getStepContext

public ExecutionContext.StepContext getStepContext(String stepName,
                                                   String transformName)

Returns the BaseExecutionContext.StepContext associated with the given step.

Specified by:

getStepContext in interface ExecutionContext

getAllStepContexts

public Collection<ExecutionContext.StepContext> getAllStepContexts()

Returns a collection view of all of the BaseExecutionContext.StepContexts.

Specified by:

getAllStepContexts in interface ExecutionContext

noteOutput

public void noteOutput(WindowedValue<?> output)

Hook for subclasses to implement that will be called whenever DoFn.Context.output(OutputT) is called.

Specified by:

noteOutput in interface ExecutionContext

noteSideOutput

public void noteSideOutput(TupleTag<?> tag,
                           WindowedValue<?> output)

Hook for subclasses to implement that will be called whenever DoFn.Context.sideOutput(com.google.cloud.dataflow.sdk.values.TupleTag<T>, T) is called.

Specified by:

noteSideOutput in interface ExecutionContext