Remove confusing taskMetrics.shuffleWriteMetrics() optional call

nemccarthy · May 28, 2015 · 96811b4 · 96811b4
1 parent 8522b6a
commit 96811b4
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/core/src/main/java/org/apache/spark/shuffle/sort/BypassMergeSortShuffleWriter.java b/core/src/main/java/org/apache/spark/shuffle/sort/BypassMergeSortShuffleWriter.java
@@ -22,7 +22,6 @@
 import java.io.FileOutputStream;
 import java.io.IOException;
 
-import scala.Option;
 import scala.Product2;
 import scala.Tuple2;
 import scala.collection.Iterator;
@@ -157,10 +156,7 @@ public long[] writePartitionedFile(
       threwException = false;
     } finally {
       Closeables.close(out, threwException);
-      Option<ShuffleWriteMetrics> maybeWriteMetrics = context.taskMetrics().shuffleWriteMetrics();
-      if (maybeWriteMetrics.isDefined()) {
-        maybeWriteMetrics.get().incShuffleWriteTime(System.nanoTime() - writeStartTime);
-      }
+      writeMetrics.incShuffleWriteTime(System.nanoTime() - writeStartTime);
     }
     return lengths;
   }