57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
|
# =============================================
|
|||
|
|
# =============== 简单任务管理器 ===============
|
|||
|
|
# =============================================
|
|||
|
|
|
|||
|
|
|
|||
|
|
from PySide2.QtCore import QMutex
|
|||
|
|
from ..utils.thread_pool import threadRun # 异步执行函数
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SimpleMission:
|
|||
|
|
def __init__(self, msnTask):
|
|||
|
|
self._msnTask = msnTask # 任务函数
|
|||
|
|
self._msnList = [] # 任务队列
|
|||
|
|
self._msnMutex = QMutex() # 任务队列的锁
|
|||
|
|
self._task = None # 异步任务对象
|
|||
|
|
self._taskMutex = QMutex() # 任务对象的锁
|
|||
|
|
|
|||
|
|
def addMissionList(self, msnList): # 添加一条任务队列，返回任务ID
|
|||
|
|
if len(msnList) < 1:
|
|||
|
|
return "[Error] len(msnList) < 1 !"
|
|||
|
|
# 添加到任务队列
|
|||
|
|
self._msnMutex.lock() # 上锁
|
|||
|
|
self._msnList += msnList
|
|||
|
|
self._msnMutex.unlock() # 解锁
|
|||
|
|
# 启动任务
|
|||
|
|
self._startMsns()
|
|||
|
|
return "[Success]"
|
|||
|
|
|
|||
|
|
def _startMsns(self): # 启动异步任务，执行所有任务列表
|
|||
|
|
# 若当前异步任务对象为空，则创建工作线程
|
|||
|
|
self._taskMutex.lock() # 上锁
|
|||
|
|
if self._task == None:
|
|||
|
|
self._task = threadRun(self._taskRun)
|
|||
|
|
self._taskMutex.unlock() # 解锁
|
|||
|
|
|
|||
|
|
# ========================= 【子线程 方法】 =========================
|
|||
|
|
|
|||
|
|
def _taskRun(self): # 异步执行任务字典的流程
|
|||
|
|
# 循环，直到任务队列的列表为空
|
|||
|
|
while True:
|
|||
|
|
self._msnMutex.lock() # 上锁
|
|||
|
|
dl = len(self._msnList) # 任务字典长度
|
|||
|
|
if dl == 0: # 任务队列已空
|
|||
|
|
self._msnMutex.unlock() # 解锁
|
|||
|
|
break
|
|||
|
|
msn = self._msnList.pop(0) # 取一个任务
|
|||
|
|
self._msnMutex.unlock() # 解锁
|
|||
|
|
if callable(self._msnTask):
|
|||
|
|
self._msnTask(msn)
|
|||
|
|
# 完成
|
|||
|
|
self._taskFinish()
|
|||
|
|
|
|||
|
|
def _taskFinish(self): # 任务结束
|
|||
|
|
self._taskMutex.lock() # 上锁
|
|||
|
|
self._task = None
|
|||
|
|
self._taskMutex.unlock() # 解锁
|