refactor: replace direct workflow signaling with call_callback_ahm activity in workflow completion logic

main
Cizz22 1 month ago
parent 372ff30a93
commit f6f762f6fb

@ -69,26 +69,14 @@ class SimulationWorkflow:
if sim_data.get("AhmJobId"): if sim_data.get("AhmJobId"):
job_id = sim_data["AhmJobId"] await workflow.execute_activity(
sim_id = sim_data["HubCnnId"] call_callback_ahm,
if "|" in job_id: {
wf_id, sim_type = job_id.split("|", 1) "simulation_id": sim_data["HubCnnId"],
else: "job_id": sim_data["AhmJobId"]
wf_id = job_id },
sim_type = "unknown" start_to_close_timeout=timedelta(days=1)
)
try:
handle = workflow.get_external_workflow_handle(wf_id)
await handle.signal(
"rbd_simulation_completed",
{
"sim_id": sim_id,
"type": sim_type,
"status": "completed"
}
)
except Exception as e:
workflow.logger.error(f"Failed to signal OpsDock: {e}")
self.status = "COMPLETED" self.status = "COMPLETED"
# etc… # etc…

Loading…
Cancel
Save