feat: add supervisor ID to DAQJobHandleStats data handling

ENRG-tr · Nov 26, 2024 · 4c2b774 · 4c2b774
1 parent c104404
commit 4c2b774
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/enrgdaq/daq/jobs/handle_stats.py b/src/enrgdaq/daq/jobs/handle_stats.py
@@ -45,6 +45,7 @@ def handle_message(self, message: DAQJobMessageStats) -> bool:
             return False
 
         keys = [
+            "supervisor",
             "daq_job",
             "is_alive",
             "last_message_in_date",
@@ -66,10 +67,15 @@ def unpack_record(record: DAQJobStatsRecord):
                 record.count,
             ]
 
+        if message.daq_job_info and message.daq_job_info.supervisor_config:
+            supervisor_id = message.daq_job_info.supervisor_config.supervisor_id
+        else:
+            supervisor_id = "N/A"
         data_to_send = []
         for daq_job_type, msg in message.stats.items():
             data_to_send.append(
                 [
+                    supervisor_id,
                     daq_job_type.__name__,
                     str(msg.is_alive).lower(),
                     *unpack_record(msg.message_in_stats),