Fix NPE in UnsafeInMemorySorter.free()

apache · Jan 7, 2016 · 2bc7827 · 2bc7827
1 parent 174e72c
commit 2bc7827
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/core/src/main/java/org/apache/spark/util/collection/unsafe/sort/UnsafeInMemorySorter.java b/core/src/main/java/org/apache/spark/util/collection/unsafe/sort/UnsafeInMemorySorter.java
@@ -111,8 +111,10 @@ public UnsafeInMemorySorter(
    * Free the memory used by pointer array.
    */
   public void free() {
-    consumer.freeArray(array);
-    array = null;
+    if (consumer != null) {
+      consumer.freeArray(array);
+      array = null;
+    }
   }
 
   public void reset() {