feat: add `disconnect()` to node pool

2023-03-10 15:57:33 +02:00 · 2023-03-10 15:57:33 +02:00 · 4564e89b4e
parent c5ca63b014
commit 4564e89b4e
1 changed files with 6 additions and 1 deletions
--- a/pomice/pool.py
+++ b/pomice/pool.py
@ -722,3 +722,8 @@ class NodePool:
        await node.connect()
        cls._nodes[node._identifier] = node
        return node
+
+    async def disconnect(self) -> None:
+        """Disconnects all nodes from the node pool."""
+        for node in self._nodes.copy().values():
+            await node.disconnect()