47 lines
2.1 KiB
Python
47 lines
2.1 KiB
Python
|
"""页面期望状态与托管进程心跳，二者分别更新。"""
|
||
|
|
|
||
|
|
from datetime import timedelta
|
||
|
|
|
||
|
|
from sqlalchemy import inspect, select, update
|
||
|
|
|
||
|
|
from app.core.exceptions import BadRequestError, ConflictError
|
||
|
|
from app.models import TPlusRuntimeControl
|
||
|
|
from app.services.authorization_service import require_permission
|
||
|
|
from app.services.tplus import task_service
|
||
|
|
from app.services.tplus.automation import local_now
|
||
|
|
|
||
|
|
|
||
|
|
def overview(session, *, now=None):
|
||
|
|
result = {"desired_state": "STOPPED", "actual_state": "OFFLINE", "worker_phase": "UNKNOWN",
|
||
|
|
"heartbeat_at": None, "last_error": None, "host": None, "platform": None,
|
||
|
|
"version": 0, "updated_by": None, "editable": False}
|
||
|
|
if not inspect(session.connection()).has_table(TPlusRuntimeControl.__tablename__):
|
||
|
|
return result
|
||
|
|
row = session.scalar(select(TPlusRuntimeControl).where(TPlusRuntimeControl.id == 1)
|
||
|
|
.execution_options(populate_existing=True))
|
||
|
|
if row is None:
|
||
|
|
return result
|
||
|
|
result.update({key: getattr(row, key) for key in result if key != "editable"})
|
||
|
|
result["editable"] = True
|
||
|
|
if row.heartbeat_at is None or (now or local_now()) - row.heartbeat_at > timedelta(seconds=90):
|
||
|
|
result["actual_state"] = "OFFLINE"
|
||
|
|
result["worker_phase"] = "UNKNOWN"
|
||
|
|
return result
|
||
|
|
|
||
|
|
|
||
|
|
def save(session, payload, actor):
|
||
|
|
require_permission(session, actor.user, "tplus:collect")
|
||
|
|
if not inspect(session.connection()).has_table(TPlusRuntimeControl.__tablename__):
|
||
|
|
raise BadRequestError("读取服务管理尚未升级，请联系维护人员")
|
||
|
|
with task_service._audit(session, actor, "修改读取服务状态", payload.model_dump()):
|
||
|
|
changed = session.execute(update(TPlusRuntimeControl).where(
|
||
|
|
TPlusRuntimeControl.id == 1,
|
||
|
|
TPlusRuntimeControl.version == payload.expected_version,
|
||
|
|
).values(desired_state=payload.desired_state, version=payload.expected_version + 1,
|
||
|
|
updated_by=actor.user.id))
|
||
|
|
if changed.rowcount != 1:
|
||
|
|
session.rollback()
|
||
|
|
raise ConflictError()
|
||
|
|
session.commit()
|
||
|
|
return overview(session)
|