Add missing closeandwriteoutput

holdenk · holdenk · commit 039d620c31a2 · 2015-06-20T19:34:36.000-07:00
diff --git a/core/src/main/java/org/apache/spark/shuffle/unsafe/UnsafeShuffleWriter.java b/core/src/main/java/org/apache/spark/shuffle/unsafe/UnsafeShuffleWriter.java
@@ -147,6 +147,7 @@ public void write(scala.collection.Iterator<Product2<K, V>> records) throws IOEx
       while (records.hasNext()) {
         insertRecordIntoSorter(records.next());
       }
+      closeAndWriteOutput();
       success = true;
     } finally {
       if (sorter != null) {

Original file line number	Diff line number	Diff line change
`@@ -147,6 +147,7 @@ public void write(scala.collection.Iterator<Product2<K, V>> records) throws IOEx`
`147`	`147`	`while (records.hasNext()) {`
`148`	`148`	`insertRecordIntoSorter(records.next());`
`149`	`149`	`}`
	`150`	`+ closeAndWriteOutput();`
`150`	`151`	`success = true;`
`151`	`152`	`} finally {`
`152`	`153`	`if (sorter != null) {`