From ed6e292d2567fb6dcf1182c64a5cd7d239d6adb9 Mon Sep 17 00:00:00 2001
From: James Wilson <james@jsdw.me>
Date: Fri, 24 Sep 2021 10:48:54 +0100
Subject: [PATCH] Clean up shard channel when shard disconnects (#411)

---
 backend/telemetry_core/src/aggregator/inner_loop.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/backend/telemetry_core/src/aggregator/inner_loop.rs b/backend/telemetry_core/src/aggregator/inner_loop.rs
index c24df52..0f8a1a1 100644
--- a/backend/telemetry_core/src/aggregator/inner_loop.rs
+++ b/backend/telemetry_core/src/aggregator/inner_loop.rs
@@ -439,6 +439,8 @@ impl InnerLoop {
                 }
             }
             FromShardWebsocket::Disconnected => {
+                self.shard_channels.remove(&shard_conn_id);
+
                 // Find all nodes associated with this shard connection ID:
                 let node_ids_to_remove: Vec<NodeId> = self
                     .node_ids