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 java.lang.Object
implements org.apache.beam.runners.core.DoFnRunner<InputT,OutputT>

DoFnRunner decorator which registers MetricsContainerImpl.

Constructor Summary

Constructors

Constructor	Description
DoFnRunnerWithMetrics(java.lang.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(java.lang.String timerId, java.lang.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 Detail

DoFnRunnerWithMetrics

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

Method Detail

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(java.lang.String timerId,
                           java.lang.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>