Tolerate double .remove()s of stream on portal teardowns

goodboy · Jul 14, 2022 · 1c16de5 · 1c16de5
1 parent dd6ee73
commit 1c16de5
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tractor/_streaming.py b/tractor/_streaming.py
@@ -601,7 +601,14 @@ async def open_stream(
 
             finally:
                 if self._portal:
-                    self._portal._streams.remove(rchan)
+                    try:
+                        self._portal._streams.remove(stream)
+                    except KeyError:
+                        log.warning(
+                            f'Stream was already destroyed?\n'
+                            f'actor: {self.chan.uid}\n'
+                            f'ctx id: {self.cid}'
+                        )
 
     async def result(self) -> Any:
         '''