feat: add `disconnect()` to node pool

This commit is contained in:
NiceAesth 2023-03-10 15:57:33 +02:00
parent c5ca63b014
commit 4564e89b4e
1 changed files with 6 additions and 1 deletions

View File

@ -721,4 +721,9 @@ class NodePool:
await node.connect() await node.connect()
cls._nodes[node._identifier] = node cls._nodes[node._identifier] = node
return 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()