jc211
/
tractor
forked from goodboy/tractor
1
0
Fork 0
Code Pull Requests Activity

Wake all sleeping consumers on bcaster closure

context_caching
Tyler Goodlet 2021-12-15 16:20:57 -05:00
parent 213447008b
commit 11e64426f6
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tractor/trionics/_broadcast.py
View File

@ -331,10 +331,16 @@ class BroadcastReceiver(ReceiveChannel):
if self._closed: if self._closed:
return return
# if there are sleeping consumers wake
# them on closure.
rr = self._state.recv_ready
if rr:
_, event = rr
event.set()
# XXX: leaving it like this consumers can still get values # XXX: leaving it like this consumers can still get values
# up to the last received that still reside in the queue. # up to the last received that still reside in the queue.
self._state.subs.pop(self.key) self._state.subs.pop(self.key)
self._closed = True self._closed = True