blob: e766e71476e7a632dc72973840b9952290c395e6 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
from __future__ import annotations
from collections.abc import Callable
from contextlib import AsyncExitStack
from functools import partial
from typing import Any
import attrs
from anyio import CapacityLimiter, to_process
from .._structures import Job
from ..abc import JobExecutor
@attrs.define(eq=False, kw_only=True)
class ProcessPoolJobExecutor(JobExecutor):
"""
Executes functions in a process pool.
:param max_workers: the maximum number of worker processes to keep
"""
max_workers: int = 40
_limiter: CapacityLimiter = attrs.field(init=False)
async def start(self, exit_stack: AsyncExitStack) -> None:
self._limiter = CapacityLimiter(self.max_workers)
async def run_job(self, func: Callable[..., Any], job: Job) -> Any:
wrapped = partial(func, *job.args, **job.kwargs)
return await to_process.run_sync(
wrapped, cancellable=True, limiter=self._limiter
)
|