async-search: clean up old query keys.

1 files changed, 2 insertions, 0 deletions

diff --git a/kad/src/Network/Kademlia/Search.hs b/kad/src/Network/Kademlia/Search.hs
index 19d0df69..1f0cfa13 100644
--- a/kad/src/Network/Kademlia/Search.hs
+++ b/kad/src/Network/Kademlia/Search.hs
@@ -132,6 +132,7 @@ reset nearestNodes qsearch target st = do
         writeTVar (searchInformant    st) MM.empty
         writeTVar (searchVisited      st) Set.empty
         writeTVar (searchPendingCount st) 0
+        writeTVar (searchPending      st) Map.empty
         return st
 
 grokQuery :: forall addr nid tok ni r qk.
@@ -209,6 +210,7 @@ searchCancel SearchState{..} = do
     writeTVar searchQueued Nothing
     m <- readTVar searchPending
     foldr (>>) (return ()) m
+    writeTVar searchPending Map.empty
 
 search ::
     ( Ord r