org.apache.beam.runners.samza.metrics

Class DoFnRunnerWithMetrics<InT,OutT>

java.lang.Object

org.apache.beam.runners.samza.metrics.DoFnRunnerWithMetrics<InT,OutT>

All Implemented Interfaces:

org.apache.beam.runners.core.DoFnRunner<InT,OutT>

public class DoFnRunnerWithMetrics<InT,OutT>
extends java.lang.Object
implements org.apache.beam.runners.core.DoFnRunner<InT,OutT>

DoFnRunner wrapper with metrics. The class uses SamzaMetricsContainer to keep BEAM metrics results and update Samza metrics.

Method Summary

Modifier and Type	Method and Description
void	finishBundle()
org.apache.beam.sdk.transforms.DoFn<InT,OutT>	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)
<KeyT> void	onWindowExpiration(org.apache.beam.sdk.transforms.windowing.BoundedWindow window, org.joda.time.Instant timestamp, KeyT key)
void	processElement(org.apache.beam.sdk.util.WindowedValue<InT> elem)
void	startBundle()
static <InT,OutT> org.apache.beam.runners.core.DoFnRunner<InT,OutT>	wrap(org.apache.beam.runners.core.DoFnRunner<InT,OutT> doFnRunner, SamzaMetricsContainer metricsContainer, java.lang.String stepName)

Methods inherited from class java.lang.Object

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

Method Detail

wrap

public static <InT,OutT> org.apache.beam.runners.core.DoFnRunner<InT,OutT> wrap(org.apache.beam.runners.core.DoFnRunner<InT,OutT> doFnRunner,
                                                                                SamzaMetricsContainer metricsContainer,
                                                                                java.lang.String stepName)

startBundle

public void startBundle()

Specified by:

startBundle in interface org.apache.beam.runners.core.DoFnRunner<InT,OutT>

processElement

public void processElement(org.apache.beam.sdk.util.WindowedValue<InT> elem)

Specified by:

processElement in interface org.apache.beam.runners.core.DoFnRunner<InT,OutT>

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)

Specified by:

onTimer in interface org.apache.beam.runners.core.DoFnRunner<InT,OutT>

finishBundle

public void finishBundle()

Specified by:

finishBundle in interface org.apache.beam.runners.core.DoFnRunner<InT,OutT>

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<InT,OutT>

getFn

public org.apache.beam.sdk.transforms.DoFn<InT,OutT> getFn()

Specified by:

getFn in interface org.apache.beam.runners.core.DoFnRunner<InT,OutT>