- add subscription via MessageDispatcher
This commit is contained in:
@@ -5,10 +5,13 @@ import abc
|
||||
|
||||
class WsServer:
|
||||
def __init__(self, loop=None):
|
||||
if loop is None:
|
||||
self.loop = asyncio.get_event_loop()
|
||||
self.loop = loop
|
||||
|
||||
def run_forever(self):
|
||||
if self.loop is None:
|
||||
asyncio.get_event_loop().run_forever()
|
||||
else:
|
||||
self.loop = loop
|
||||
self.loop.run_forever()
|
||||
|
||||
async def listen(self, host, port):
|
||||
return await websockets.serve(self.handler, host, port, loop=self.loop)
|
||||
|
||||
@@ -21,7 +21,6 @@ class WsServerMultiUser(WsServer):
|
||||
self.listener = listener
|
||||
self.USERS = UserSet()
|
||||
self.global_state = {}
|
||||
self.listener.on_connect(self.loop)
|
||||
|
||||
async def notify_state(self, data):
|
||||
if self.USERS: # asyncio.wait doesn't accept an empty list
|
||||
|
||||
Reference in New Issue
Block a user