refactor: register protocol stream tasks

This commit is contained in:
jxxghp
2026-08-23 15:13:05 +08:00
parent e7e232d625
commit 7afabeda02
6 changed files with 150 additions and 9 deletions
+8
View File
@@ -40,6 +40,14 @@ def get_background_task_registry(
return runtime.tasks
def get_background_task_registry_compat(request: Request) -> TaskRegistry:
"""返回协议兼容端点使用的任务登记器,允许未启动 lifespan 的旧调用回退。"""
runtime = getattr(request.app.state, "host_runtime", None)
if isinstance(runtime, HostRuntime):
return runtime.tasks
return get_task_registry()
def resolve_background_task_registry(value: object) -> TaskRegistry:
"""兼容直接调用 endpoint 的旧入口,并优先使用注入的任务登记器。"""
if isinstance(value, TaskRegistry):