Chia-Network/chia-blockchain
1
0
Fork 0
mirror of https://github.com/Chia-Network/chia-blockchain.git synced 2024-09-11 10:56:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

harvester just manage (#16755)

Browse Source
This commit is contained in:
Kyle Altendorf 2023-11-21 18:13:32 -05:00 committed by GitHub
parent 06e680d421
commit 799b4928c7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
chia/harvester/harvester.py
View File

@ -131,26 +131,18 @@ class Harvester:
@contextlib.asynccontextmanager
async def manage(self) -> AsyncIterator[None]:
await self._start()
self._refresh_lock = asyncio.Lock()
self.event_loop = asyncio.get_running_loop()
try:
yield
finally:
self._close()
await self._await_closed()
self._shut_down = True
self.executor.shutdown(wait=True)
self.plot_manager.stop_refreshing()
self.plot_manager.reset()
self.plot_sync_sender.stop()
async def _start(self) -> None:
self._refresh_lock = asyncio.Lock()
self.event_loop = asyncio.get_running_loop()
def _close(self) -> None:
self._shut_down = True
self.executor.shutdown(wait=True)
self.plot_manager.stop_refreshing()
self.plot_manager.reset()
self.plot_sync_sender.stop()
async def _await_closed(self) -> None:
await self.plot_sync_sender.await_closed()
await self.plot_sync_sender.await_closed()
def get_connections(self, request_node_type: Optional[NodeType]) -> List[Dict[str, Any]]:
return default_get_connections(server=self.server, request_node_type=request_node_type)