Class _ScriptService

task.py:2893–2990

class _ScriptService:
    _redirection_regexp = re.compile(r"\s+>\s+")

    def __init__(self):
        self.heap: List[Tuple[int, int, _Script]] = []
        self.lock = threading.Lock()
        self._my_pid = os.getpid()
        self._tasks: List[_Script] = []

    def register(self, name: str, cmd: str, interval: int):
        """注册任务"""
        task = _Script(name, cmd, interval)
        self._tasks.append(task)
        return task

    def register_scripts(self, *scripts: _Script):
        """注册任务"""
        for script in scripts:
            if script in self._tasks:
                continue
            self._tasks.append(script)

    def init_tasks(self):
        """添加任务到队列"""
        with self.lock:
            for idx, task in enumerate(self._tasks):
                next_execution_time = task.first() + idx*3
                heapq.heappush(self.heap, (next_execution_time, task.get_ident(), task))
        self._tasks = None

    def repetition_check(self, task: _Script) -> bool:
        real_cmd = task.cmd.strip()
        if real_cmd.startswith("nohup"):
            real_cmd = real_cmd[len("nohup"):].strip()
        if self._redirection_regexp.search(real_cmd):
            # 切分重定向
            real_cmd = self._redirection_regexp.split(real_cmd)[0].strip()

        real_cmd_list = real_cmd.split(" ")
        for p in psutil.pids():
            try:
                if p == self._my_pid:
                    continue
                proc = psutil.Process(p)
                cmd = proc.cmdline()
                if cmd[:len(real_cmd_list)] != real_cmd_list:
                    continue

                run_times = time.time() - proc.create_time()
                if run_times < task.interval:
                    write_log("任务[{}]未达到运行间隔，跳过".format(task.name))
                    return False
                if run_times < 10 * 60:
                    write_log("任务【{}】执行拥塞，跳过本次执行".format(task.name), _level='error', color='red')
                    return True
                else:
                    write_log("任务【{}】执行超时，已杀死".format(task.name), _level='error', color='red')
                    proc.kill()