mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-10-29 11:27:43 +08:00
83 lines
2.8 KiB
Python
83 lines
2.8 KiB
Python
import threading
|
||
from concurrent.futures import Future
|
||
from contextvars import copy_context
|
||
from typing import Any, Callable, TypeVar, cast
|
||
|
||
from app.foundation.singleton import Singleton
|
||
from app.runtime.execution import OwnedThreadPoolExecutor
|
||
from app.runtime.settings import get_runtime_setting
|
||
|
||
_Result = TypeVar("_Result")
|
||
_THREAD_POOL_STOP_TIMEOUT_SECONDS = 10.0
|
||
|
||
# 阶段 64 曾在本模块引入该私有名；保留精确别名，避免测试或外部诊断代码失效。
|
||
_OwnedThreadPoolExecutor = OwnedThreadPoolExecutor
|
||
|
||
|
||
# strict mypy 跳过 foundation 实现导入，因此无法在本文件解析既有 Singleton 元类类型。
|
||
class ThreadHelper(metaclass=Singleton): # type: ignore[metaclass]
|
||
"""
|
||
共享后台线程池 owner，负责关联上下文传播和生命周期收敛。
|
||
"""
|
||
|
||
def __init__(self) -> None:
|
||
"""按系统配置创建共享后台线程池。"""
|
||
self._lifecycle_lock = threading.RLock()
|
||
self.pool = self._new_pool()
|
||
|
||
@staticmethod
|
||
def _new_pool() -> OwnedThreadPoolExecutor:
|
||
"""按当前运行配置构造一个新的共享线程池 owner。"""
|
||
return OwnedThreadPoolExecutor(
|
||
max_workers=get_runtime_setting('CONF').threadpool
|
||
)
|
||
|
||
def submit(
|
||
self,
|
||
func: Callable[..., _Result],
|
||
*args: Any,
|
||
**kwargs: Any,
|
||
) -> Future[_Result]:
|
||
"""
|
||
提交任务
|
||
:param func: 函数
|
||
:param args: 参数
|
||
:param kwargs: 参数
|
||
:return: future
|
||
"""
|
||
context = copy_context()
|
||
with self._lifecycle_lock:
|
||
# strict mypy 跳过 execution 实现导入，需要在兼容门面恢复 Future 泛型。
|
||
return cast(
|
||
Future[_Result],
|
||
self.pool.submit(context.run, func, *args, **kwargs),
|
||
)
|
||
|
||
def reopen(self) -> bool:
|
||
"""
|
||
为新的应用生命周期恢复线程池提交准入。
|
||
|
||
只有上一 owner 已完全收敛时才会构造新线程池，避免跨 lifespan
|
||
同时保留两组 worker。
|
||
"""
|
||
with self._lifecycle_lock:
|
||
if self.pool.accepting:
|
||
return True
|
||
if not self.pool.shutdown_bounded(timeout=0.0):
|
||
return False
|
||
self.pool = self._new_pool()
|
||
return True
|
||
|
||
def shutdown(
|
||
self,
|
||
timeout: float = _THREAD_POOL_STOP_TIMEOUT_SECONDS,
|
||
) -> bool:
|
||
"""
|
||
有限等待共享线程池关闭。
|
||
|
||
:param timeout: 等待已接受任务达到终态的最长秒数
|
||
:return: 全部任务和 worker 均已终止时返回 True，否则返回 False
|
||
"""
|
||
with self._lifecycle_lock:
|
||
return bool(self.pool.shutdown_bounded(timeout=timeout))
|