|
4 | 4 | from functools import partial |
5 | 5 | from inspect import isawaitable |
6 | 6 | from logging import Logger, getLogger |
7 | | -from typing import Awaitable, Callable |
| 7 | +from typing import Awaitable, Callable, Tuple |
8 | 8 |
|
9 | 9 | from anyio import ( |
10 | 10 | TASK_STATUS_IGNORED, |
@@ -37,12 +37,11 @@ class YRoom: |
37 | 37 | _on_message: Callable[[bytes], Awaitable[bool] | bool] | None |
38 | 38 | _update_send_stream: MemoryObjectSendStream |
39 | 39 | _update_receive_stream: MemoryObjectReceiveStream |
40 | | - _task_group: TaskGroup | None = None |
41 | | - _started: Event | None = None |
| 40 | + _task_group: TaskGroup | None |
| 41 | + _started: Event | None |
42 | 42 | _stopped: Event |
43 | | - __start_lock: Lock | None = None |
44 | | - _subscription: Subscription | None = None |
45 | | - |
| 43 | + __start_lock: Lock | None |
| 44 | + _subscription: Subscription | None |
46 | 45 | def __init__( |
47 | 46 | self, |
48 | 47 | ready: bool = True, |
@@ -82,6 +81,13 @@ def __init__( |
82 | 81 | self._on_message = None |
83 | 82 | self.exception_handler = exception_handler |
84 | 83 | self._stopped = Event() |
| 84 | + self._update_send_stream, self._update_receive_stream = create_memory_object_stream( |
| 85 | + max_buffer_size=65536 |
| 86 | + ) |
| 87 | + self._task_group = None |
| 88 | + self._started = None |
| 89 | + self.__start_lock = None |
| 90 | + self._subscription= None |
85 | 91 |
|
86 | 92 | @property |
87 | 93 | def _start_lock(self) -> Lock: |
|
0 commit comments