Break loop after result retreival

goodboy · Apr 11, 2022 · 346cf6e · 346cf6e
1 parent 297d48b
commit 346cf6e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tractor/_streaming.py b/tractor/_streaming.py
@@ -272,9 +272,9 @@ async def aclose(self):
     @asynccontextmanager
     async def subscribe(
         self,
-
     ) -> AsyncIterator[BroadcastReceiver]:
-        '''Allocate and return a ``BroadcastReceiver`` which delegates
+        '''
+        Allocate and return a ``BroadcastReceiver`` which delegates
         to this message stream.
 
         This allows multiple local tasks to receive each their own copy
@@ -674,6 +674,7 @@ def consume(
                             result = consume(msg)
                             if result:
                                 self._result = result
+                                break
 
                 if not self._result:
                     while True: