Fix memory accounting in ParallelHashBuildOperator

Use same logic as for HashBuilderOperator
soscpd · Sep 9, 2016 · e669bd5 · e669bd5
1 parent 8a8344f
commit e669bd5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/presto-main/src/main/java/com/facebook/presto/operator/ParallelHashBuildOperator.java b/presto-main/src/main/java/com/facebook/presto/operator/ParallelHashBuildOperator.java
@@ -197,8 +197,10 @@ public void addInput(Page page)
         checkState(!isFinished(), "Operator is already finished");
 
         index.addPage(page);
-
-        operatorContext.setMemoryReservation(page.getPositionCount());
+        if (!operatorContext.trySetMemoryReservation(index.getEstimatedSize().toBytes())) {
+            index.compact();
+        }
+        operatorContext.setMemoryReservation(index.getEstimatedSize().toBytes());
         operatorContext.recordGeneratedOutput(page.getSizeInBytes(), page.getPositionCount());
     }