[BEAM-1672] Use Accumulable MetricsContainers in Direct runner.

apache · Apr 24, 2017 · ebbb72a · ebbb72a
1 parent 5204c53
commit ebbb72a
Show file tree

Hide file tree

Showing 9 changed files with 33 additions and 630 deletions.
diff --git a/runners/direct-java/src/main/java/org/apache/beam/runners/direct/DirectMetrics.java b/runners/direct-java/src/main/java/org/apache/beam/runners/direct/DirectMetrics.java
diff --git a/runners/direct-java/src/main/java/org/apache/beam/runners/direct/DirectRunner.java b/runners/direct-java/src/main/java/org/apache/beam/runners/direct/DirectRunner.java
@@ -41,6 +41,7 @@
 import org.apache.beam.sdk.Pipeline.PipelineExecutionException;
 import org.apache.beam.sdk.PipelineResult;
 import org.apache.beam.sdk.io.Read;
+import org.apache.beam.sdk.metrics.AccumulatedMetricResults;
 import org.apache.beam.sdk.metrics.MetricResults;
 import org.apache.beam.sdk.metrics.MetricsEnvironment;
 import org.apache.beam.sdk.options.PipelineOptions;
@@ -411,7 +412,9 @@ public String toString() {
 
  @Override
  public MetricResults metrics() {
- return evaluationContext.getMetrics();
+ return new AccumulatedMetricResults(
+ evaluationContext.getAttemptedMetrics(),
+ evaluationContext.getCommittedMetrics());
  }
 
  /**