Merge pull request #59 from openEDI/jm/broker_status

Add status REST command to broker

broker/server.py

@@ -16,6 +16,7 @@
 from fastapi.exceptions import HTTPException
 from fastapi.responses import FileResponse, JSONResponse
 from oedisi.types.common import HeathCheck, ServerReply
+from oedisi.tools.broker_utils import get_time_data
 
 app = FastAPI()
 
@@ -143,6 +144,7 @@ def run_simulation():
     initstring = f"-f {len(component_map)} --name=mainbroker --loglevel=trace --local_interface={broker_ip} --localport=23404"
     logging.info(f"Broker initaialization string: {initstring}")
     broker = h.helicsCreateBroker("zmq", "", initstring)
+    app.state.broker = broker
     logging.info(broker)
     isconnected = h.helicsBrokerIsConnected(broker)
     logging.info(f"Broker connected: {isconnected}")
@@ -177,6 +179,22 @@ async def run_feeder(background_tasks: BackgroundTasks):
         raise HTTPException(status_code=404, detail=str(err))
 
 
+@app.get("/status/")
+async def status():
+    try:
+        name_2_timedata = {}
+        connected = h.helicsBrokerIsConnected(app.state.broker)
+        if connected:
+            for time_data in get_time_data(app.state.broker):
+                if (time_data.name not in name_2_timedata) or (
+                    name_2_timedata[time_data.name] != time_data
+                ):
+                    name_2_timedata[time_data.name] = time_data
+        return {"connected": connected, "timedata": name_2_timedata, "error": False}
+    except AttributeError as e:
+        return {"reply": str(e), "error": True}
+
+
 if __name__ == "__main__":
     port = int(sys.argv[2])
     uvicorn.run(app, host="0.0.0.0", port=port)