From 5ed572187bc76e07add827533197d05ee3816abd Mon Sep 17 00:00:00 2001 From: meejah Date: Fri, 12 Feb 2021 01:16:33 -0700 Subject: [PATCH] unregister completely --- src/wormhole_transit_relay/server_state.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/wormhole_transit_relay/server_state.py b/src/wormhole_transit_relay/server_state.py index 690b2c5..0e7c446 100644 --- a/src/wormhole_transit_relay/server_state.py +++ b/src/wormhole_transit_relay/server_state.py @@ -153,8 +153,14 @@ class PendingRequests(object): self._active = active_connections def unregister(self, token, side, tc): + """ + We no longer care about a particular client (e.g. it has + disconnected). + """ if token in self._requests: self._requests[token].discard((side, tc)) + if not self._requests[token]: + del self._requests[token] self._active.unregister(tc) def register_token(self, token, new_side, new_tc):