feat: support Anyio totally

Author: WSH032

src/fastapi_proxy_lib/core/_tool.py

@@ -309,8 +309,8 @@ def return_err_msg_response(
     err_response_json = ErrRseponseJson(detail=detail)
 
     # TODO: 请注意，logging是同步函数，每次会阻塞1ms左右，这可能会导致性能问题
-    # 特别是对于写入文件的log，最好把它放到 asyncio.to_thread 里执行
-    # https://docs.python.org/zh-cn/3/library/asyncio-task.html#coroutine
+    # 特别是对于写入文件的log，最好把它放到 `anyio.to_thread.run_sync()` 里执行
+    # https://anyio.readthedocs.io/en/stable/threads.html#running-a-function-in-a-worker-thread
 
     if logger is not None:
         # 只要传入了logger，就一定记录日志

src/fastapi_proxy_lib/fastapi/router.py

@@ -3,7 +3,6 @@
 The low-level API for [fastapi_proxy_lib.fastapi.app][].
 """
 
-import asyncio
 import warnings
 from contextlib import asynccontextmanager
 from typing import (
@@ -18,6 +17,7 @@
     Union,
 )
 
+import anyio
 from fastapi import APIRouter
 from starlette.requests import Request
 from starlette.responses import Response
@@ -273,6 +273,8 @@ async def shutdown_clients(*_: Any, **__: Any) -> AsyncIterator[None]:
                 When __aexit__ is called, will close all registered proxy.
             """
             yield
-            await asyncio.gather(*[proxy.aclose() for proxy in self.registered_proxy])
+            async with anyio.create_task_group() as tg:
+                for proxy in self.registered_proxy:
+                    tg.start_soon(proxy.aclose)
 
         return shutdown_clients