radical.asyncflow.backends.execution.concurrent¶

def __await__(self):
    """Make backend awaitable."""
    return self._async_init().__await__()

async def submit_tasks(self, tasks: List[Dict[str, Any]]) -> List[asyncio.Task]:
    """Submit tasks for execution."""
    submitted_tasks = []

    for task in tasks:
        future = asyncio.create_task(self._handle_task(task))
        submitted_tasks.append(future)

        self.tasks[task['uid']] = task
        self.tasks[task['uid']]['future'] = future

    return submitted_tasks

async def cancel_task(self, uid: str) -> bool:
    """Cancel a task by its UID.

    Args:
        uid (str): The UID of the task to cancel.

    Returns:
        bool: True if the task was found and cancellation was attempted, False otherwise.
    """
    if uid in self.tasks:
        task = self.tasks[uid]
        future = task['future']
        if future and future.cancel():
            self._callback_func(task, 'CANCELED')
            return True
    return False

async def cancel_all_tasks(self) -> int:
    """Cancel all running tasks."""
    cancelled_count = 0
    for task in self.tasks.values():
        future = task['future']
        future.cancel()
        cancelled_count += 1
    self.tasks.clear()
    return cancelled_count

async def shutdown(self) -> None:
    """Shutdown the executor."""
    await self.cancel_all_tasks()
    self.executor.shutdown(wait=True)
    print('Shutdown complete')

async def __aenter__(self):
    """Async context manager entry."""
    if not self._initialized:
        await self._async_init()
    return self

async def __aexit__(self, exc_type, exc_val, exc_tb):
    """Async context manager exit."""
    await self.shutdown()

@classmethod
async def create(cls, executor: Union[ThreadPoolExecutor, ProcessPoolExecutor]):
    """Factory method for creating initialized backend."""
    backend = cls(executor)
    return await backend