[POAE7-961] fix null pointer issue when offheap enabled. (#21)

oap-project · Mar 25, 2021 · 4c83d5c · 4c83d5c
1 parent 6a5b99c
commit 4c83d5c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/java/org/apache/spark/util/collection/unsafe/sort/PMemWriter.java b/src/main/java/org/apache/spark/util/collection/unsafe/sort/PMemWriter.java
@@ -180,7 +180,8 @@ public void updateLongArray(LongArray sortedArray, int numRecords, int position)
             // update recordPointer in this array
             long originalRecordPointer = sortedArray.get(position);
             MemoryBlock page = taskMemoryManager.getOriginalPage(originalRecordPointer);
-            long offset = taskMemoryManager.getOffsetInPage(originalRecordPointer) - page.getBaseOffset();
+            long baseOffset = page == null? 0: page.getBaseOffset();
+            long offset = taskMemoryManager.getOffsetInPage(originalRecordPointer) - baseOffset;
             MemoryBlock pMemBlock = pageMap.get(page);
             long pMemOffset = pMemBlock.getBaseOffset() + offset;
             sortedArray.set(position, pMemOffset);