feat(core): requests 兼容异步垫片 + 同步/异步桥

- core/aiohttp_compat.py:以 requests 同形 API(post/get + Response.json/text/
  status_code/raise_for_status + HTTPError/RequestException/Timeout)实现异步请求,
  使既有 requests 风格代码可「加 await + async def」最小改动迁移;4xx/5xx 不抛、
  仅 raise_for_status() 抛(与 requests 语义一致,保住风控判断逻辑)
- core/async_bridge.py:run_sync(coro) 经常驻后台事件循环线程执行协程并阻塞等待,
  供迁移期同步调用方(遗留 Flask 路由/线程池)复用异步实现

验证:pytest 16/16;ruff All checks passed;pyright 0 errors。
This commit is contained in:
snowy-yang committed 2026-09-26 00:27:47 +08:00
1 parent 9f56f68b66
commit 4962542e78
4 files changed
+301

No files matched your search

+56
View File
@@ -0,0 +1,56 @@
# -*- coding: utf-8 -*-
"""同步 ↔ 异步桥(迁移期用)。
场景:部分调用方仍是同步代码(遗留 Flask 路由、线程池下载器),而服务层已异步。
`run_sync(coro)` 把协程提交到**专用后台事件循环线程**执行并阻塞等待结果,
使同步代码得以调用异步实现,无需把调用链一次性全部改异步。
- 后台 loop 常驻单例(daemon 线程),避免每个请求新建 loop 的开销与
aiohttp 连接池失效;
- 在已有事件循环中调用 `run_sync` 会死锁,故检测到运行中的 loop 时直接报错,
提示改用 `await`。
"""
import asyncio
import threading
from collections.abc import Coroutine
from typing import Any
_loop: asyncio.AbstractEventLoop | None = None
_loop_lock = threading.Lock()
def _get_loop() -> asyncio.AbstractEventLoop:
global _loop
if _loop is not None and not _loop.is_closed():
return _loop
with _loop_lock:
if _loop is None or _loop.is_closed():
loop = asyncio.new_event_loop()
t = threading.Thread(target=loop.run_forever, name='async-bridge', daemon=True)
t.start()
_loop = loop
return _loop
def run_sync[T](coro: Coroutine[Any, Any, T], timeout: float | None = None) -> T:
"""在同步代码中运行协程并等待结果。
Raises:
RuntimeError: 在已有事件循环内调用(应直接 await 而非桥接)。
"""
try:
asyncio.get_running_loop()
except RuntimeError:
pass
else:
raise RuntimeError('run_sync 不能在事件循环内调用,请直接 await 协程')
fut = asyncio.run_coroutine_threadsafe(coro, _get_loop())
return fut.result(timeout)
def shutdown() -> None:
"""停止后台 loop(应用退出时调用)。"""
global _loop
if _loop is not None and not _loop.is_closed():
_loop.call_soon_threadsafe(_loop.stop)
_loop = None