diff --git a/ws/server/ws_server.py b/ws/server/ws_server.py index 5db8600..bbdda08 100644 --- a/ws/server/ws_server.py +++ b/ws/server/ws_server.py @@ -14,7 +14,7 @@ class WsServer: self.loop.run_forever() async def listen(self, host, port): - return await websockets.serve(self.handler, host, port, loop=self.loop) + return await websockets.serve(self.handler, host, port) def disconnect(self): pass @@ -23,8 +23,8 @@ class WsServer: print("handler: got connection from {}".format(websocket.remote_address)) await self.register(websocket) try: - consumer_task = asyncio.ensure_future(self.handler_recv(websocket, path), loop=self.loop) - producer_task = asyncio.ensure_future(self.handler_send(websocket, path), loop=self.loop) + consumer_task = asyncio.ensure_future(self.handler_recv(websocket, path)) + producer_task = asyncio.ensure_future(self.handler_send(websocket, path)) done, pending = await asyncio.wait([consumer_task, producer_task], return_when=asyncio.FIRST_COMPLETED,) for task in pending: task.cancel()