Rename to broadcast mod, don't expect mem chan specifically

live_on_air_from_tokio
Tyler Goodlet 2021-08-09 07:35:42 -04:00
parent 1af7dbb732
commit 64358f6525
1 changed files with 5 additions and 5 deletions

View File

@ -15,9 +15,9 @@ import tractor
from trio.lowlevel import current_task from trio.lowlevel import current_task
from trio.abc import ReceiveChannel from trio.abc import ReceiveChannel
from trio._core._run import Task from trio._core._run import Task
from trio._channel import ( # from trio._channel import (
MemoryReceiveChannel, # MemoryReceiveChannel,
) # )
class Lagged(trio.TooSlowError): class Lagged(trio.TooSlowError):
@ -43,7 +43,7 @@ class BroadcastReceiver(ReceiveChannel):
self._rx = rx_chan self._rx = rx_chan
self._queue = queue self._queue = queue
self._subs: dict[Task, int] = {} # {id(current_task()): -1} self._subs: dict[Task, int] = {} # {id(current_task()): -1}
self._clones: dict[Task, MemoryReceiveChannel] = {} self._clones: dict[Task, ReceiveChannel] = {}
self._value_received: Optional[trio.Event] = None self._value_received: Optional[trio.Event] = None
async def receive(self): async def receive(self):
@ -138,7 +138,7 @@ class BroadcastReceiver(ReceiveChannel):
def broadcast_receiver( def broadcast_receiver(
recv_chan: MemoryReceiveChannel, recv_chan: ReceiveChannel,
max_buffer_size: int, max_buffer_size: int,
) -> BroadcastReceiver: ) -> BroadcastReceiver: