You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
111 lines
3.5 KiB
Python
111 lines
3.5 KiB
Python
from datetime import timedelta
|
|
from temporalio import workflow
|
|
from temporalio.client import Client
|
|
import os
|
|
with workflow.unsafe.imports_passed_through():
|
|
from temporal.activity import calculate_plant_eaf_activity, execute_simulation_activity, update_contribution_bulk_mappings_activity, update_equipment_for_simulation_activity,call_callback_ahm, acquire_lock, release_lock,execute_update_node
|
|
|
|
|
|
TEMPORAL_URL = os.environ.get("TEMPORAL_URL", "http://192.168.1.86:7233")
|
|
|
|
@workflow.defn
|
|
class SimulationWorkflow:
|
|
def __init__(self):
|
|
self.status = "INITIALIZED"
|
|
self.isDone = False
|
|
|
|
@workflow.query
|
|
def get_status(self) -> str:
|
|
return self.status
|
|
|
|
@workflow.query
|
|
def is_done(self) -> bool:
|
|
return self.isDone
|
|
|
|
@workflow.run
|
|
async def run(self, sim_data: dict) -> str:
|
|
self.status = "QUEUE"
|
|
|
|
await workflow.execute_activity(
|
|
acquire_lock,
|
|
sim_data["HubCnnId"],
|
|
start_to_close_timeout=timedelta(days=10)
|
|
)
|
|
|
|
self.status = "RUNNING"
|
|
|
|
# 1. Update equipment for simulation
|
|
results = await workflow.execute_activity(
|
|
update_equipment_for_simulation_activity,
|
|
{**sim_data}, # ✅ one positional argument
|
|
start_to_close_timeout=timedelta(days=1)
|
|
)
|
|
|
|
await workflow.execute_activity(
|
|
execute_update_node,
|
|
{"projectName": sim_data['projectName'], "equipment_nodes": results['equipment_nodes']},
|
|
start_to_close_timeout=timedelta(days=1)
|
|
)
|
|
|
|
# 2. Execute simulation
|
|
await workflow.execute_activity(
|
|
execute_simulation_activity,
|
|
{"sim_data": sim_data, "eq_update": results['reliability']},
|
|
start_to_close_timeout=timedelta(days=1)
|
|
)
|
|
|
|
# 3. Calculate EAF
|
|
await workflow.execute_activity(
|
|
calculate_plant_eaf_activity,
|
|
sim_data,
|
|
start_to_close_timeout=timedelta(days=1)
|
|
)
|
|
|
|
await workflow.execute_activity(
|
|
update_contribution_bulk_mappings_activity,
|
|
sim_data["HubCnnId"],
|
|
start_to_close_timeout=timedelta(days=1)
|
|
)
|
|
|
|
|
|
if sim_data.get("AhmJobId"):
|
|
job_id = sim_data["AhmJobId"]
|
|
sim_id = sim_data["HubCnnId"]
|
|
if "|" in job_id:
|
|
wf_id, sim_type = job_id.split("|", 1)
|
|
else:
|
|
wf_id = job_id
|
|
sim_type = "unknown"
|
|
|
|
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"
|
|
# etc…
|
|
self.isDone = True
|
|
|
|
await workflow.execute_activity(
|
|
release_lock,
|
|
sim_data["HubCnnId"],
|
|
start_to_close_timeout=timedelta(minutes=10),
|
|
)
|
|
|
|
if sim_data.get("CallbackWorkflowId"):
|
|
handle = workflow.get_external_workflow_handle(
|
|
sim_data["CallbackWorkflowId"]
|
|
)
|
|
await handle.signal("notify_done")
|
|
|
|
|
|
return sim_data["HubCnnId"] # simulation_id
|