feat: add kill_job tool

2026-06-24 14:44:10 +08:00
parent 1cd3128522
commit 0d60f9b246
2 changed files with 59 additions and 0 deletions
@@ -0,0 +1,23 @@
+# coding=utf-8
+"""
+@Time   :2026/6/24
+@Author :tao.chen
+"""
+from common.logging import logger
+from spark_executor.core.job_store import JobStore
+from spark_executor.core.yarn_client import kill_application
+
+store = JobStore()
+
+
+def kill_job(job_id: str) -> dict[str, str]:
+    job = store.get(job_id)
+    if job is None:
+        raise KeyError(f"Unknown job_id: {job_id}")
+    kill_application(job.application_id)
+    logger.info(f"kill_job job_id={job_id} application_id={job.application_id}")
+    return {
+        "job_id": job_id,
+        "application_id": job.application_id,
+        "status": "KILLED",
+    }