From 040877ae69ebad98ea9d8abe6257780872ec3085 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jon=20H=C3=A4ggblad?= <jon.haggblad@gmail.com>
Date: Mon, 21 Aug 2023 12:22:55 +0200
Subject: [PATCH] remove on drop

---
 .../websocket/connection_handler/authenticated.rs              | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/gateway/src/node/client_handling/websocket/connection_handler/authenticated.rs b/gateway/src/node/client_handling/websocket/connection_handler/authenticated.rs
index a51f7b3ee9..c451293454 100644
--- a/gateway/src/node/client_handling/websocket/connection_handler/authenticated.rs
+++ b/gateway/src/node/client_handling/websocket/connection_handler/authenticated.rs
@@ -102,6 +102,9 @@ pub(crate) struct AuthenticatedHandler<R, S, St> {
 // explicitly remove handle from the global store upon being dropped
 impl<R, S, St> Drop for AuthenticatedHandler<R, S, St> {
     fn drop(&mut self) {
+        // WIP(JON): Also remove from is_active_pending_replies
+        self.inner.is_active_pending_replies.remove(&self.client.address);
+
         self.inner
             .active_clients_store
             .disconnect(self.client.address)