refactor: update timestamp handling in n1081b.py and handle_stats.py

src/daq/jobs/caen/n1081b.py

@@ -1,13 +1,13 @@
 import time
 from dataclasses import dataclass
-from datetime import datetime
 
 from N1081B import N1081B
 from websocket import WebSocket
 
 from daq.base import DAQJob
 from daq.models import DAQJobMessage
 from daq.store.models import DAQJobMessageStore, StorableDAQJobConfig
+from utils.time import get_now_unix_timestamp_ms
 
 N1081B_QUERY_INTERVAL_SECONDS = 1
 N1081B_WEBSOCKET_TIMEOUT_SECONDS = 5
@@ -88,7 +88,7 @@ def _poll_sections(self):
     def _send_store_message(self, data: dict, section):
         keys = ["timestamp", *[f"lemo_{x['lemo']}" for x in data["counters"]]]
         values = [
-            int(datetime.now().timestamp() * 1000),  # unix timestamp in milliseconds
+            get_now_unix_timestamp_ms(),  # unix timestamp in milliseconds
             *[x["value"] for x in data["counters"]],
         ]
         self.message_out.put(

src/daq/jobs/handle_stats.py

@@ -6,6 +6,7 @@
 from daq.base import DAQJob
 from daq.models import DAQJobMessage, DAQJobStats, DAQJobStatsRecord
 from daq.store.models import DAQJobMessageStore, StorableDAQJobConfig
+from utils.time import get_unix_timestamp_ms
 
 DAQJobStatsDict = Dict[type[DAQJob], DAQJobStats]
 
@@ -47,7 +48,7 @@ def handle_message(self, message: DAQJobMessageStats) -> bool:
         def datetime_to_str(dt: Optional[datetime]):
             if dt is None:
                 return "N/A"
-            return int(dt.timestamp() * 1000)
+            return get_unix_timestamp_ms(dt)
 
         def unpack_record(record: DAQJobStatsRecord):
             return [