feat: add store prefix to DAQJobHandleStats

src/daq/jobs/handle_stats.py

@@ -62,11 +62,17 @@ def unpack_record(record: DAQJobStatsRecord):
                 ]
             )
 
+        if message.daq_job_info and message.daq_job_info.supervisor_config:
+            prefix = message.daq_job_info.supervisor_config.supervisor_id
+        else:
+            prefix = None
+
         self._put_message_out(
             DAQJobMessageStore(
                 store_config=self.config.store_config,
                 keys=keys,
                 data=data_to_send,
+                prefix=prefix,
             )
         )
 

src/supervisor.py

@@ -17,7 +17,7 @@
     start_daq_jobs,
 )
 from daq.jobs.handle_stats import DAQJobMessageStats, DAQJobStatsDict
-from daq.models import DAQJobConfig, DAQJobMessage, DAQJobStats
+from daq.models import DAQJobConfig, DAQJobInfo, DAQJobMessage, DAQJobStats
 from daq.store.base import DAQJobStore
 from models import SupervisorConfig
 
@@ -137,7 +137,12 @@ def get_supervisor_messages(self):
         messages = []
 
         # Send stats message
-        messages.append(DAQJobMessageStats(stats=self.daq_job_stats))
+        messages.append(
+            DAQJobMessageStats(
+                stats=self.daq_job_stats,
+                daq_job_info=self._get_supervisor_daq_job_info(),
+            )
+        )
         return messages
 
     def get_daq_job_stats(
@@ -208,3 +213,12 @@ def _load_supervisor_config(self):
 
         with open(SUPERVISOR_CONFIG_FILE_PATH, "rb") as f:
             return msgspec.toml.decode(f.read(), type=SupervisorConfig)
+
+    def _get_supervisor_daq_job_info(self):
+        return DAQJobInfo(
+            daq_job_type="Supervisor",
+            daq_job_class_name="Supervisor",
+            supervisor_config=self.config,
+            unique_id=self.config.supervisor_id,
+            instance_id=0,
+        )