MemoryReservation exposes MemoryConsumer (#8000)

... as a getter method.
apache · Oct 31, 2023 · 747cb50 · 747cb50
1 parent 27e64ae
commit 747cb50
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/datafusion/execution/src/memory_pool/mod.rs b/datafusion/execution/src/memory_pool/mod.rs
@@ -157,6 +157,11 @@ impl MemoryReservation {
         self.size
     }
 
+    /// Returns [MemoryConsumer] for this [MemoryReservation]
+    pub fn consumer(&self) -> &MemoryConsumer {
+        &self.registration.consumer
+    }
+
     /// Frees all bytes from this reservation back to the underlying
     /// pool, returning the number of bytes freed.
     pub fn free(&mut self) -> usize {