syndicate-lang
/
mini-syndicate-py
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Better fault handling: a delayed send/close will only fire if self.ws is still present

This commit is contained in:
Tony Garnock-Jones 2019-05-16 15:00:00 +01:00
parent 2f119b81af
commit d514baac38
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
syndicate/mini/core.py
View File

@ -224,12 +224,18 @@ class WebsocketConnection(Connection):
self.ws = None
def _send(self, bs):
if self.ws:
self.loop.call_soon_threadsafe(lambda: self.loop.create_task(self.ws.send(bs)))
if self.loop:
def _do_send():
if self.ws:
self.loop.create_task(self.ws.send(bs))
self.loop.call_soon_threadsafe(_do_send)
def _disconnect(self):
if self.ws:
self.loop.call_soon_threadsafe(lambda: self.loop.create_task(self.ws.close()))
if self.loop:
def _do_disconnect():
if self.ws:
self.loop.create_task(self.ws.close())
self.loop.call_soon_threadsafe(_do_disconnect)
async def main(self, loop, on_connected=None):
if self.ws is not None: