Add OperatorContext#aggregateSystemMemoryContext method

trinodb · Mar 12, 2019 · e4d97d1 · e4d97d1
1 parent 98bad41
commit e4d97d1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/presto-main/src/main/java/io/prestosql/operator/OperatorContext.java b/presto-main/src/main/java/io/prestosql/operator/OperatorContext.java
@@ -269,6 +269,12 @@ public AggregatedMemoryContext aggregateUserMemoryContext()
         return new InternalAggregatedMemoryContext(operatorMemoryContext.aggregateUserMemoryContext(), memoryFuture, this::updatePeakMemoryReservations, false);
     }
 
+    // caller shouldn't close this context as it's managed by the OperatorContext
+    public AggregatedMemoryContext aggregateSystemMemoryContext()
+    {
+        return new InternalAggregatedMemoryContext(operatorMemoryContext.aggregateSystemMemoryContext(), memoryFuture, this::updatePeakMemoryReservations, false);
+    }
+
     // caller shouldn't close this context as it's managed by the OperatorContext
     public AggregatedMemoryContext aggregateRevocableMemoryContext()
     {

diff --git a/presto-memory-context/src/main/java/io/prestosql/memory/context/MemoryTrackingContext.java b/presto-memory-context/src/main/java/io/prestosql/memory/context/MemoryTrackingContext.java
@@ -117,6 +117,11 @@ public AggregatedMemoryContext aggregateRevocableMemoryContext()
         return revocableAggregateMemoryContext;
     }
 
+    public AggregatedMemoryContext aggregateSystemMemoryContext()
+    {
+        return systemAggregateMemoryContext;
+    }
+
     public AggregatedMemoryContext newAggregateSystemMemoryContext()
     {
         return systemAggregateMemoryContext.newAggregatedMemoryContext();