From ab4a4bdb76c3bb3da14f52698f1b12545be84dd2 Mon Sep 17 00:00:00 2001
From: wakonig_k <klaus.wakonig@psi.ch>
Date: Tue, 5 Mar 2024 22:51:09 +0100
Subject: [PATCH] fix(nidaq_writer): fixed redis connector after refactoring

---
 bec_plugins/services/NIDAQ_writer/NIDAQ_writer.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/bec_plugins/services/NIDAQ_writer/NIDAQ_writer.py b/bec_plugins/services/NIDAQ_writer/NIDAQ_writer.py
index b2297c0..b5819f4 100644
--- a/bec_plugins/services/NIDAQ_writer/NIDAQ_writer.py
+++ b/bec_plugins/services/NIDAQ_writer/NIDAQ_writer.py
@@ -58,7 +58,7 @@ class NIDAQWriterService(BECService):
         """
         Callback for scan status messages.
         """
-        msg = messages.ScanStatusMessage.loads(message.value)
+        msg = message.value
         if not msg:
             return
         parent.handle_scan_status(msg)
@@ -99,7 +99,7 @@ class NIDAQWriterService(BECService):
 
         start_time = time.time()
         if self.use_redis_stream:
-            msg = self.producer.xread("ni_data")
+            msg = self.connector.xread("ni_data")
 
             if msg:
                 try:
@@ -118,7 +118,7 @@ class NIDAQWriterService(BECService):
                     logger.error(f"Failed to parse message: {content}")
             time.sleep(0.01)
         else:
-            msgs = self.producer.r.lpop("ni_data:val", 20)
+            msgs = self.connector._redis_conn.lpop("ni_data:val", 20)
             time.sleep(0.001)
             if msgs:
                 try: