100 lines
3.1 KiB
Python
100 lines
3.1 KiB
Python
|
# 任务队列类
|
||
|
|
|
||
|
|
from uuid import uuid4
|
||
|
|
from typing import Callable, Any
|
||
|
|
|
||
|
|
from umi_log import logger
|
||
|
|
|
||
|
|
|
||
|
|
class MissionQueue:
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
msnList: list, # 任务内容列表，每项为一个任务元素
|
||
|
|
configs: dict = {}, # 任务控制参数
|
||
|
|
# 回调函数
|
||
|
|
# 整个队列 即将启动。回调参数(self)
|
||
|
|
onStart: Callable[["MissionQueue"], None] = None,
|
||
|
|
# 单个任务 准备进行。回调参数(self, 任务元素)
|
||
|
|
onReady: Callable[["MissionQueue", Any], None] = None,
|
||
|
|
# 单个任务 完成。回调参数(self, 任务元素, 任务结果)
|
||
|
|
onGet: Callable[["MissionQueue"], None] = None,
|
||
|
|
# 整个队列 已结束。回调参数(self, 结果标志)
|
||
|
|
# 结果标志："[Success]...", "[Warning]...", "[Error]..."
|
||
|
|
onEnd: Callable[["MissionQueue", str], None] = None,
|
||
|
|
):
|
||
|
|
self.msnList = msnList
|
||
|
|
self.configs = configs
|
||
|
|
self.id = str(uuid4()) # 任务ID
|
||
|
|
self.len = len(msnList) # 队列总长度
|
||
|
|
self.index = 0 # 当前任务下标
|
||
|
|
self._onStart = onStart if onStart else self._pass
|
||
|
|
self._onReady = onReady if onReady else self._pass
|
||
|
|
self._onGet = onGet if onGet else self._pass
|
||
|
|
self._onEnd = onEnd if onEnd else self._pass
|
||
|
|
# 工作状态： "waiting" "running"
|
||
|
|
self.state_work = "waiting"
|
||
|
|
# 控制状态： "" "pause" "stop"
|
||
|
|
self.state_ctrl = ""
|
||
|
|
|
||
|
|
# ==================== mission调用 ====================
|
||
|
|
|
||
|
|
# 获取剩余任务长度
|
||
|
|
def remainingLen(self):
|
||
|
|
return self.len - self.index
|
||
|
|
|
||
|
|
# 判空 【1】←在一轮任务循环中的调用顺序
|
||
|
|
def empty(self):
|
||
|
|
return self.index >= self.len
|
||
|
|
|
||
|
|
# 获取一个任务元素 【2】
|
||
|
|
def getMsn(self):
|
||
|
|
return self.msnList[self.index]
|
||
|
|
|
||
|
|
# 完成一个任务元素 【5】
|
||
|
|
def popMsn(self):
|
||
|
|
self.index += 1
|
||
|
|
|
||
|
|
# 恢复进行
|
||
|
|
def resume(self):
|
||
|
|
if self.state_ctrl == "pause":
|
||
|
|
self.state_ctrl = ""
|
||
|
|
|
||
|
|
# 暂停
|
||
|
|
def pause(self):
|
||
|
|
self.state_ctrl = "pause"
|
||
|
|
|
||
|
|
# 停止（废弃）
|
||
|
|
def stop(self):
|
||
|
|
self.state_ctrl = "stop"
|
||
|
|
|
||
|
|
# ==================== 回调函数 ====================
|
||
|
|
|
||
|
|
def onStart(self):
|
||
|
|
try:
|
||
|
|
self._onStart(self)
|
||
|
|
except Exception:
|
||
|
|
logger.error("任务 onStart 回调异常。", exc_info=True, stack_info=True)
|
||
|
|
|
||
|
|
def onReady(self): # 【3】
|
||
|
|
try:
|
||
|
|
self._onReady(self)
|
||
|
|
except Exception:
|
||
|
|
logger.error("任务 onReady 回调异常。", exc_info=True, stack_info=True)
|
||
|
|
|
||
|
|
def onGet(self, res): # 【4】
|
||
|
|
try:
|
||
|
|
self._onGet(self, res)
|
||
|
|
except Exception:
|
||
|
|
logger.error("任务 onGet 回调异常。", exc_info=True, stack_info=True)
|
||
|
|
|
||
|
|
def onEnd(self, flag):
|
||
|
|
# 结果标志："[Success]...", "[Warning]...", "[Error]..."
|
||
|
|
try:
|
||
|
|
self._onEnd(self, flag)
|
||
|
|
except Exception:
|
||
|
|
logger.error("任务 onEnd 回调异常。", exc_info=True, stack_info=True)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def _pass(*e): # 空函数
|
||
|
|
pass
|