feat: add some cleanup methods to the MessageBus

Signed-off-by: John DeAngelis <[email protected]>

backend/ops_api/ops/__init__.py

@@ -95,6 +95,7 @@ def shutdown_session(exception=None):
     def teardown_request(exception=None):
         if hasattr(request, "message_bus"):
             request.message_bus.handle()
+            request.message_bus.cleanup()
 
     @event.listens_for(db_session, "before_commit")
     def receive_before_commit(session: Session):

backend/ops_api/ops/services/can_messages.py

@@ -8,5 +8,5 @@ def can_history_trigger(
     event: OpsEvent,
     session: Session,
 ):
-    logger.debug(f"Handling event {event}")
+    logger.debug(f"Handling event {event.event_type} with details: {event.event_details}")
     assert session is not None

backend/ops_api/ops/services/message_bus.py

@@ -19,6 +19,7 @@ class MessageBus:
     """
 
     published_events: List[OpsEvent] = []
+    known_callbacks = []
 
     def handle(self):
         """
@@ -28,6 +29,7 @@ def handle(self):
             ops_signal = signal(event.event_type.name)
             ops_signal.send(event, session=current_app.db_session)
             logger.debug(f"Handling event {event}")
+        self.published_events.clear()
 
     def subscribe(self, event_type: OpsEventType, callback: callable):
         """
@@ -51,3 +53,13 @@ def publish(self, event_type: OpsEventType, event: OpsEvent):
         """
         logger.debug(f"Publishing event {event_type} with details {event}")
         self.published_events.append(event)
+
+    def cleanup(self):
+        """
+        Clean up all subscriptions and published events.
+        """
+        for callback in self.known_callbacks:
+            ops_signal = signal(callback)
+            ops_signal.disconnect(callback)
+        self.published_events.clear()
+        self.known_callbacks.clear()