From ede5f9dd484ebc894b2308decc0b53cc5f397391 Mon Sep 17 00:00:00 2001
From: Furkan <info@furkanbilgin.net>
Date: Mon, 11 Nov 2024 13:21:30 +0300
Subject: [PATCH] fix: fix a bug that causes `consume(nowait=False)` to hang
 forever

---
 src/daq/base.py | 23 ++++++++++++++++-------
 1 file changed, 16 insertions(+), 7 deletions(-)

diff --git a/src/daq/base.py b/src/daq/base.py
index 94f53c5..974e084 100644
--- a/src/daq/base.py
+++ b/src/daq/base.py
@@ -57,15 +57,24 @@ def __init__(
         self.info = self._create_info()
 
     def consume(self, nowait=True):
-        # consume messages from the queue
+        """
+        Consumes messages from the message_in queue.
+        If nowait is True, it will consume the message immediately.
+        Otherwise, it will wait until a message is available.
+        """
+
+        def _process_message(message):
+            if not self.handle_message(message):
+                self.message_in.put(message)
+
+        # Return immediately after consuming the message
+        if not nowait:
+            _process_message(self.message_in.get())
+            return
+
         while True:
             try:
-                if nowait:
-                    message = self.message_in.get_nowait()
-                else:
-                    message = self.message_in.get()
-                if not self.handle_message(message):
-                    self.message_in.put(message)
+                _process_message(self.message_in.get_nowait())
             except Empty:
                 break