Package org.apache.beam.runners.spark.translation

Class DoFnRunnerWithMetrics<InputT,OutputT>

java.lang.Object

org.apache.beam.runners.spark.translation.DoFnRunnerWithMetrics<InputT,OutputT>

All Implemented Interfaces:

org.apache.beam.runners.core.DoFnRunner<InputT,OutputT>

public class DoFnRunnerWithMetrics<InputT,OutputT>
extends Object
implements org.apache.beam.runners.core.DoFnRunner<InputT,OutputT>

DoFnRunner decorator which registers MetricsContainerImpl.

Constructor Summary

Constructors

Constructor	Description
DoFnRunnerWithMetrics(String stepName, org.apache.beam.runners.core.DoFnRunner<InputT,OutputT> delegate, MetricsContainerStepMapAccumulator metricsAccum)

Method Summary

Modifier and Type	Method	Description
void	finishBundle()
org.apache.beam.sdk.transforms.DoFn<InputT,OutputT>	getFn()
<KeyT> void	onTimer(String timerId, String timerFamilyId, KeyT key, org.apache.beam.sdk.transforms.windowing.BoundedWindow window, org.joda.time.Instant timestamp, org.joda.time.Instant outputTimestamp, org.apache.beam.sdk.state.TimeDomain timeDomain, org.apache.beam.sdk.values.CausedByDrain causedByDrain)
<KeyT> void	onWindowExpiration(org.apache.beam.sdk.transforms.windowing.BoundedWindow window, org.joda.time.Instant timestamp, KeyT key)
void	processElement(org.apache.beam.sdk.values.WindowedValue<InputT> elem)
void	startBundle()

Methods inherited from class java.lang.Object

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

Constructor Details

DoFnRunnerWithMetrics

public DoFnRunnerWithMetrics(String stepName,
 org.apache.beam.runners.core.DoFnRunner<InputT,OutputT> delegate,
 MetricsContainerStepMapAccumulator metricsAccum)

Method Details

getFn

public org.apache.beam.sdk.transforms.DoFn<InputT,OutputT> getFn()

Specified by:

getFn in interface org.apache.beam.runners.core.DoFnRunner<InputT,OutputT>

startBundle

public void startBundle()

Specified by:

startBundle in interface org.apache.beam.runners.core.DoFnRunner<InputT,OutputT>

processElement

public void processElement(org.apache.beam.sdk.values.WindowedValue<InputT> elem)

Specified by:

processElement in interface org.apache.beam.runners.core.DoFnRunner<InputT,OutputT>

onTimer

public <KeyT> void onTimer(String timerId,
 String timerFamilyId,
 KeyT key,
 org.apache.beam.sdk.transforms.windowing.BoundedWindow window,
 org.joda.time.Instant timestamp,
 org.joda.time.Instant outputTimestamp,
 org.apache.beam.sdk.state.TimeDomain timeDomain,
 org.apache.beam.sdk.values.CausedByDrain causedByDrain)

Specified by:

onTimer in interface org.apache.beam.runners.core.DoFnRunner<InputT,OutputT>

finishBundle

public void finishBundle()

Specified by:

finishBundle in interface org.apache.beam.runners.core.DoFnRunner<InputT,OutputT>

onWindowExpiration

public <KeyT> void onWindowExpiration(org.apache.beam.sdk.transforms.windowing.BoundedWindow window,
 org.joda.time.Instant timestamp,
 KeyT key)

Specified by:

onWindowExpiration in interface org.apache.beam.runners.core.DoFnRunner<InputT,OutputT>