From 5aa5b2075265acb9acac3ea1a1a4d871d5ff611d Mon Sep 17 00:00:00 2001
From: yudovin <artsiom.yudovin@profitero.com>
Date: Thu, 31 Oct 2019 20:21:20 +0300
Subject: [PATCH] Adding check for null

---
 .../util/collection/unsafe/sort/UnsafeInMemorySorter.java     | 4 ++++
 1 file changed, 4 insertions(+)

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
index 1a9453a8b3e80..e14964d68119b 100644
--- 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
@@ -205,6 +205,10 @@ public long getSortTimeNanos() {
   }
 
   public long getMemoryUsage() {
+    if (array == null) {
+      return 0L;
+    }
+
     return array.size() * 8;
   }