From a0a6b54f3cb13b7b9152ff3e2b9a7405fa9559cf Mon Sep 17 00:00:00 2001
From: Moritz Hoffmann <antiguru@gmail.com>
Date: Thu, 13 Apr 2023 20:50:04 -0400
Subject: [PATCH] Expose TraceAgent::operator

Signed-off-by: Moritz Hoffmann <antiguru@gmail.com>
---
 src/operators/arrange/agent.rs | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/operators/arrange/agent.rs b/src/operators/arrange/agent.rs
index 63f950bc7..4a9e8f242 100644
--- a/src/operators/arrange/agent.rs
+++ b/src/operators/arrange/agent.rs
@@ -5,7 +5,7 @@ use std::cell::RefCell;
 use std::collections::VecDeque;
 
 use timely::dataflow::Scope;
-use timely::dataflow::operators::generic::source;
+use timely::dataflow::operators::generic::{OperatorInfo, source};
 use timely::progress::Timestamp;
 use timely::progress::{Antichain, frontier::AntichainRef};
 use timely::dataflow::operators::CapabilitySet;
@@ -38,7 +38,7 @@ where
     physical_compaction: Antichain<Tr::Time>,
     temp_antichain: Antichain<Tr::Time>,
 
-    operator: ::timely::dataflow::operators::generic::OperatorInfo,
+    operator: OperatorInfo,
     logging: Option<::logging::Logger>,
 }
 
@@ -89,7 +89,7 @@ where
     Tr::Time: Timestamp+Lattice,
 {
     /// Creates a new agent from a trace reader.
-    pub fn new(trace: Tr, operator: ::timely::dataflow::operators::generic::OperatorInfo, logging: Option<::logging::Logger>) -> (Self, TraceWriter<Tr>)
+    pub fn new(trace: Tr, operator: OperatorInfo, logging: Option<::logging::Logger>) -> (Self, TraceWriter<Tr>)
     where
         Tr: Trace,
         Tr::Batch: Batch,
@@ -155,6 +155,11 @@ where
         reference.0.activate();
         reference
     }
+
+    /// The [OperatorInfo] of the underlying Timely operator
+    pub fn operator(&self) -> &OperatorInfo {
+        &self.operator
+    }
 }
 
 impl<Tr> TraceAgent<Tr>