yield chunk

ppigazzini · Apr 25, 2024 · 0f018e0 · 0f018e0
1 parent 7792324
commit 0f018e0
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/server/fishtest/rundb.py b/server/fishtest/rundb.py
@@ -89,11 +89,12 @@ def __init__(self, generator):
     def read(self, size=-1):
         while size < 0 or len(self.buffer) < size:
             try:
-                self.buffer += next(self.generator)
+                chunk = next(self.generator)
+                self.buffer += chunk
+                result, self.buffer = self.buffer[:size], self.buffer[size:]
+                yield result
             except StopIteration:
                 break
-        result, self.buffer = self.buffer[:size], self.buffer[size:]
-        return result
 
     def close(self):
         pass  # No cleanup needed, but method is required