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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
from __future__ import annotations
import asyncio
from functools import partial
from types import TracebackType
from typing import Any, Awaitable, Callable, Optional
from ..config import Config
from ..typing import AppWrapper, ASGIReceiveCallable, ASGIReceiveEvent, ASGISendEvent, Scope
try:
from asyncio import TaskGroup as AsyncioTaskGroup
except ImportError:
from taskgroup import TaskGroup as AsyncioTaskGroup # type: ignore
async def _handle(
app: AppWrapper,
config: Config,
scope: Scope,
receive: ASGIReceiveCallable,
send: Callable[[Optional[ASGISendEvent]], Awaitable[None]],
sync_spawn: Callable,
call_soon: Callable,
) -> None:
try:
await app(scope, receive, send, sync_spawn, call_soon)
except asyncio.CancelledError:
raise
except Exception:
await config.log.exception("Error in ASGI Framework")
finally:
await send(None)
class TaskGroup:
def __init__(self, loop: asyncio.AbstractEventLoop) -> None:
self._loop = loop
self._task_group = AsyncioTaskGroup()
async def spawn_app(
self,
app: AppWrapper,
config: Config,
scope: Scope,
send: Callable[[Optional[ASGISendEvent]], Awaitable[None]],
) -> Callable[[ASGIReceiveEvent], Awaitable[None]]:
app_queue: asyncio.Queue[ASGIReceiveEvent] = asyncio.Queue(config.max_app_queue_size)
def _call_soon(func: Callable, *args: Any) -> Any:
future = asyncio.run_coroutine_threadsafe(func(*args), self._loop)
return future.result()
self.spawn(
_handle,
app,
config,
scope,
app_queue.get,
send,
partial(self._loop.run_in_executor, None),
_call_soon,
)
return app_queue.put
def spawn(self, func: Callable, *args: Any) -> None:
self._task_group.create_task(func(*args))
async def __aenter__(self) -> "TaskGroup":
await self._task_group.__aenter__()
return self
async def __aexit__(self, exc_type: type, exc_value: BaseException, tb: TracebackType) -> None:
await self._task_group.__aexit__(exc_type, exc_value, tb)
|
