Attempt to aid memory issues

Author: vkottler

runtimepy/net/arbiter/housekeeping/__init__.py

@@ -21,6 +21,8 @@
 class ConnectionMetricsPoller(_ArbiterTask):
     """A class that periodically polls connection metrics."""
 
+    processors: list[Awaitable[None]]
+
     def __init__(
         self,
         name: str,
@@ -31,6 +33,7 @@ def __init__(
 
         super().__init__(name, **kwargs)
         self.manager = manager
+        self.processors = []
 
     def _init_state(self) -> None:
         """Add channels to this instance's channel environment."""
@@ -51,22 +54,22 @@ async def dispatch(self) -> bool:
             self.manager.poll_metrics()
 
         # Handle any incoming commands.
-        processors: list[Awaitable[None]] = []
         for mapping in (
             self.app.connections.values(),
             self.app.tasks.values(),
             self.app.structs.values(),
         ):
             for item in mapping:
                 if isinstance(item, AsyncCommandProcessingMixin):
-                    processors.append(item.process_command_queue())
+                    self.processors.append(item.process_command_queue())
 
         # Service connection tasks. The connection manager should probably do
         # this on its own at some point.
-        processors += list(self.app.conn_manager.connection_tasks)
+        self.processors += list(self.app.conn_manager.connection_tasks)
 
-        if processors:
-            await asyncio.gather(*processors)
+        if self.processors:
+            await asyncio.gather(*self.processors)
+            self.processors.clear()
 
         return True
 

runtimepy/net/server/app/env/tab/message.py

@@ -69,9 +69,7 @@ def handle_shown_state(
             for name in env.names:
                 self._setup_callback(name, state)
         else:
-            # Remove callbacks for primitives.
-            for name, val in state.callbacks.items():
-                state.primitives[name].remove_callback(val)
+            state.clear_telemetry()
 
         # Save current UI state.
         state.latest_ui_values.update(latest)

runtimepy/net/server/websocket/state.py

@@ -53,19 +53,36 @@ def frame(self, time: float) -> JsonMessage:
 
         return result
 
+    def clear_telemetry(self) -> None:
+        """Clear all telemetry interactions."""
+
+        # Remove callbacks for primitives.
+        for name, val in self.callbacks.items():
+            self.primitives[name].remove_callback(val)
+        self.callbacks.clear()
+        self.primitives.clear()
+
+        # Clear points.
+        self.points.clear()
+
     def clear_loggers(self) -> None:
         """Clear all logging handlers."""
 
+        # Remove handlers.
         for logger in self._loggers:
             logger.removeHandler(self.tab_logger)
+        self._loggers.clear()
+
         self.tab_logger.drain_str()
-        self._loggers = []
+
+        self.clear_telemetry()
 
     def add_logger(self, logger: logging.Logger) -> None:
         """Add a logger."""
 
-        logger.addHandler(self.tab_logger)
-        self._loggers.append(logger)
+        if logger not in self._loggers:
+            logger.addHandler(self.tab_logger)
+            self._loggers.append(logger)
 
     @staticmethod
     def create() -> "TabState":