Fixed memory leak with the notifications server

ComprosoftCEO · Dec 23, 2021 · 4007a01 · 4007a01
1 parent 800080e
commit 4007a01
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/notifications/src/notifications/websocket_actor.rs b/notifications/src/notifications/websocket_actor.rs
@@ -72,6 +72,15 @@ where
 ///
 impl Actor for WebsocketActor {
   type Context = ws::WebsocketContext<Self>;
+
+  fn stopping(&mut self, ctx: &mut Self::Context) -> Running {
+    // Remove all references to this actor
+    self.subscription_manager.do_send(UnsubscribeAll {
+      me: ctx.address().recipient(),
+    });
+
+    Running::Stop
+  }
 }
 
 ///