2c7714268f
- 新增 task_report 任务级聚合指标 task_metrics - 补充创建任务幂等与失败路径/冲突测试 - 将后端测试基线提升到 20 passed - 新增 edge-agent 初始化代码、启动脚本与打包脚本 - 新增 http_health_check、check_port、check_process、grep_log 执行器 - 补充 edge-agent 基础测试并提升基线到 10 passed - 同步更新 backend README 与当前进度总结
95 lines
3.0 KiB
Python
95 lines
3.0 KiB
Python
from __future__ import annotations
|
|
|
|
import csv
|
|
import platform
|
|
import subprocess
|
|
from io import StringIO
|
|
from typing import Any
|
|
|
|
|
|
class ProcessCheckExecutor:
|
|
def execute(self, params: dict[str, Any]) -> tuple[bool, str, dict[str, Any], dict[str, Any]]:
|
|
process_name = params.get("process_name")
|
|
pid = params.get("pid")
|
|
|
|
if not process_name and pid is None:
|
|
return False, "process_name or pid is required", {}, {}
|
|
|
|
processes = self._list_processes()
|
|
matched = [item for item in processes if self._match_process(item, process_name=process_name, pid=pid)]
|
|
|
|
success = len(matched) > 0
|
|
message = "process found" if success else "process not found"
|
|
return (
|
|
success,
|
|
message,
|
|
{
|
|
"matched_count": len(matched),
|
|
"process_name": process_name,
|
|
"pid": pid,
|
|
},
|
|
{
|
|
"matches": matched,
|
|
},
|
|
)
|
|
|
|
def _list_processes(self) -> list[dict[str, Any]]:
|
|
system_name = platform.system().upper()
|
|
if system_name.startswith("WIN"):
|
|
return self._list_windows_processes()
|
|
return self._list_unix_processes()
|
|
|
|
def _list_windows_processes(self) -> list[dict[str, Any]]:
|
|
result = subprocess.run(
|
|
["tasklist", "/FO", "CSV", "/NH"],
|
|
capture_output=True,
|
|
text=True,
|
|
check=True,
|
|
)
|
|
reader = csv.reader(StringIO(result.stdout))
|
|
rows: list[dict[str, Any]] = []
|
|
for row in reader:
|
|
if len(row) < 2:
|
|
continue
|
|
rows.append(
|
|
{
|
|
"pid": int(row[1]),
|
|
"process_name": row[0],
|
|
"command": row[0],
|
|
}
|
|
)
|
|
return rows
|
|
|
|
def _list_unix_processes(self) -> list[dict[str, Any]]:
|
|
result = subprocess.run(
|
|
["ps", "-eo", "pid=,comm=,args="],
|
|
capture_output=True,
|
|
text=True,
|
|
check=True,
|
|
)
|
|
rows: list[dict[str, Any]] = []
|
|
for line in result.stdout.splitlines():
|
|
parts = line.strip().split(None, 2)
|
|
if len(parts) < 2:
|
|
continue
|
|
pid_text = parts[0]
|
|
process_name = parts[1]
|
|
command = parts[2] if len(parts) > 2 else process_name
|
|
rows.append(
|
|
{
|
|
"pid": int(pid_text),
|
|
"process_name": process_name,
|
|
"command": command,
|
|
}
|
|
)
|
|
return rows
|
|
|
|
def _match_process(self, item: dict[str, Any], process_name: str | None, pid: int | str | None) -> bool:
|
|
if pid is not None and item["pid"] != int(pid):
|
|
return False
|
|
if process_name:
|
|
name = str(process_name).lower()
|
|
if name not in item["process_name"].lower() and name not in item["command"].lower():
|
|
return False
|
|
return True
|