Enable stream backpressure by default, add MsgStream.ctx: Context

goodboy · Dec 12, 2022 · 304ee6c · 304ee6c
1 parent 5dc07f0
commit 304ee6c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tractor/_streaming.py b/tractor/_streaming.py
@@ -69,7 +69,7 @@ class ReceiveMsgStream(trio.abc.ReceiveChannel):
     '''
     def __init__(
         self,
-        ctx: 'Context',  # typing: ignore # noqa
+        ctx: Context,  # typing: ignore # noqa
         rx_chan: trio.MemoryReceiveChannel,
         _broadcaster: Optional[BroadcastReceiver] = None,
 
@@ -82,6 +82,9 @@ def __init__(
         self._eoc: bool = False
         self._closed: bool = False
 
+    def ctx(self) -> Context:
+        return self._ctx
+
     # delegate directly to underlying mem channel
     def receive_nowait(self):
         msg = self._rx_chan.receive_nowait()
@@ -381,7 +384,7 @@ class Context:
     # only set on the callee side
     _scope_nursery: Optional[trio.Nursery] = None
 
-    _backpressure: bool = False
+    _backpressure: bool = True
 
     async def send_yield(self, data: Any) -> None: