diff --git a/pomice/pool.py b/pomice/pool.py index 8f72d31..cabb5e2 100644 --- a/pomice/pool.py +++ b/pomice/pool.py @@ -721,4 +721,9 @@ class NodePool: await node.connect() cls._nodes[node._identifier] = node - return node \ No newline at end of file + return node + + async def disconnect(self) -> None: + """Disconnects all nodes from the node pool.""" + for node in self._nodes.copy().values(): + await node.disconnect() \ No newline at end of file