836343b29e
- 新增 backend/app/modules/compute_gateway（client/sync）计算网关模块 - 新增 backend/app/workers/compute_poller 计算轮询 worker - 新增 compute/agent/process_manager 进程管理器 - 新增 scripts/ 脚本目录 - 更新 Docker 部署配置（app/compute/nginx） - 更新后端平台 API、数据库 SQL、core 配置 - 更新前端多个视图组件及 API 模块 - 重构 frontend/dist 构建产物（新 hash） - 更新多项文档 Co-Authored-By: Claude <noreply@anthropic.com>
28 lines
1.1 KiB
Python
28 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from app.db.platform_store import get_platform_store
|
|
from app.modules.compute_gateway.client import ComputeNodeClient
|
|
|
|
|
|
def _node_for_task(task: dict[str, Any]) -> dict[str, Any] | None:
|
|
return next((node for node in get_platform_store().compute_nodes() if node["id"] == task.get("compute_node_id")), None)
|
|
|
|
|
|
async def poll_compute_jobs_once() -> dict[str, Any]:
|
|
store = get_platform_store()
|
|
synced: list[dict[str, Any]] = []
|
|
failed: list[dict[str, str]] = []
|
|
for task in store.running_compute_tasks():
|
|
node = _node_for_task(task)
|
|
if not node:
|
|
failed.append({"task_id": task["id"], "error": "compute node not found"})
|
|
continue
|
|
try:
|
|
job = await ComputeNodeClient(node["api_base_url"]).get_job(task["compute_job_id"])
|
|
synced.append(store.apply_compute_job(task["id"], job))
|
|
except Exception as exc: # noqa: BLE001 - keep polling other jobs
|
|
failed.append({"task_id": task["id"], "error": str(exc)})
|
|
return {"synced": len(synced), "failed": failed, "items": synced}
|