mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-10-02 00:27:16 +00:00
21 lines
611 B
Python
21 lines
611 B
Python
"""传输任务运行时注册表（单实例部署）"""
|
|
|
|
|
|
class TransferTaskRegistry:
|
|
"""维护任务取消标志等运行时状态（不持久化，重启后任务按状态恢复）"""
|
|
|
|
def __init__(self) -> None:
|
|
self._cancel_flags: dict[int, bool] = {}
|
|
|
|
def mark_cancel(self, task_id: int) -> None:
|
|
self._cancel_flags[task_id] = True
|
|
|
|
def is_canceled(self, task_id: int) -> bool:
|
|
return self._cancel_flags.get(task_id, False)
|
|
|
|
def clear(self, task_id: int) -> None:
|
|
self._cancel_flags.pop(task_id, None)
|
|
|
|
|
|
transfer_task_registry = TransferTaskRegistry()
|