From c2e5052877f08b625279b85e1bdd3c623f469514 Mon Sep 17 00:00:00 2001
From: Emily Toop <etoop@mozilla.com>
Date: Fri, 9 Mar 2018 12:17:58 +0000
Subject: [PATCH] Allow registration and unregistration of transaction
 observers from Conn

---
 src/conn.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

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<TxObserver>) {
+        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)]