This commit is contained in:
Your Name
2026-04-17 12:30:31 +08:00
parent 954a6a2c65
commit dc34a4e294
4 changed files with 472 additions and 1 deletions

View File

@@ -161,6 +161,35 @@ def register_node_heartbeat(
conn.commit()
def cleanup_imported_runtime_nodes(*, region: str, role: str, keep_node_code: str) -> None:
normalized_region = str(region or "").strip() or "unknown"
normalized_role = str(role or "").strip() or "unknown"
preserved_node_code = str(keep_node_code or "").strip()
if not preserved_node_code:
return
with get_db() as conn:
with conn.cursor() as cur:
cur.execute(
"""
DELETE FROM detect_worker_nodes
WHERE region = %s
AND role = %s
AND node_code <> %s
AND (
node_code = %s
OR (metadata_json->>'service') = 'runtime-ingest'
)
""",
(
normalized_region,
normalized_role,
preserved_node_code,
f"{normalized_region}-{normalized_role}-imported",
),
)
conn.commit()
def register_local_control_heartbeat() -> None:
register_node_heartbeat(
node_code=settings.node_code,