From d2aed35b1c581296f2266fea0e79e3b4d2cc4f4c Mon Sep 17 00:00:00 2001 From: Frank McSherry Date: Fri, 26 Sep 2025 17:02:34 -0400 Subject: [PATCH] Provide access for wrapped builder --- timely/src/dataflow/operators/generic/handles.rs | 3 +++ 1 file changed, 3 insertions(+) diff --git a/timely/src/dataflow/operators/generic/handles.rs b/timely/src/dataflow/operators/generic/handles.rs index c6fa7a5b1..e6d64df72 100644 --- a/timely/src/dataflow/operators/generic/handles.rs +++ b/timely/src/dataflow/operators/generic/handles.rs @@ -176,6 +176,9 @@ pub struct Session<'a: 'b, 'b, T: Timestamp, CB: ContainerBuilder, CT: Capabilit impl<'a: 'b, 'b, T: Timestamp, CB: ContainerBuilder, CT: CapabilityTrait> Session<'a, 'b, T, CB, CT> { + /// Provides access to the underlying container builder. + pub fn builder(&mut self) -> &mut CB { &mut self.buffer.builder } + /// Provides one record at the time specified by the `Session`. #[inline] pub fn give(&mut self, data: D) where CB: PushInto { self.buffer.builder.push_into(data);