phzhik/chain-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

abort chain logic

Browse Source
This commit is contained in:
Robert 2024-03-22 03:49:54 +07:00
parent adadadb92e
commit 17f5b9d797
No known key found for this signature in database
GPG Key ID: F631C7FD957D5F22
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
chain_service/services/progress_chain_runner.py
View File

@ -7,6 +7,7 @@ from chain_service.database.models.progress_chain import (
)
from chain_service.repositories.progress_chain import ProgressChainRepository
from chain_service.repositories.running_chain import RunningChainRepository
from loguru import logger
from datetime import datetime
@ -18,9 +19,11 @@ class ProgressChainRunnerService:
self,
progress_chain_repository: ProgressChainRepository,
progress_action_service_factory: ProgressActionServiceFactory,
running_chain_repository: RunningChainRepository,
):
self.progress_chain_repository = progress_chain_repository
self.progress_action_service_factory = progress_action_service_factory
self.running_chain_repository = running_chain_repository
async def process(self, progress_chain: ProgressChain):
@ -36,6 +39,8 @@ class ProgressChainRunnerService:
if not await self.process_action(progress_chain, progress_action):
break
await self.running_chain_repository.delete(progress_chain.chain_id)
async def process_action(
self, progress_chain: ProgressChain, progress_action: BaseProgressAction
):
@ -48,9 +53,15 @@ class ProgressChainRunnerService:
)
await progress_action_service.process()
assert await self.running_chain_repository.exists(progress_chain.chain_id)
progress_action.status = ProgressActionStatusEnum.DONE
return True
except AssertionError:
logger.info(f"Chain was aborted {progress_chain.chain_id}")
progress_action.status = ProgressActionStatusEnum.ABORTED
return False
except Exception as error:
logger.exception(f"Error during action process for {progress_chain.id}")
progress_action.status = ProgressActionStatusEnum.FAILED