diff --git a/src/conn.rs b/src/conn.rs index eb56c7aa..ce653ffd 100644 --- a/src/conn.rs +++ b/src/conn.rs @@ -58,6 +58,7 @@ use mentat_db::{ transact_terms, PartitionMap, TxObservationService, + TxObserver, TxReport, }; @@ -813,6 +814,14 @@ impl Conn { }, } } + + pub fn register_observer(&mut self, key: String, observer: Arc) { + self.tx_observer_service.lock().unwrap().register(key, observer); + } + + pub fn unregister_observer(&mut self, key: &String) { + self.tx_observer_service.lock().unwrap().deregister(key); + } } #[cfg(test)]