70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
|
import asyncio
|
||
|
|
|
||
|
|
from .base_manager import BaseManager
|
||
|
|
|
||
|
|
|
||
|
|
class AsyncManager(BaseManager):
|
||
|
|
"""Manage a client list for an asyncio server."""
|
||
|
|
async def can_disconnect(self, sid, namespace):
|
||
|
|
return self.is_connected(sid, namespace)
|
||
|
|
|
||
|
|
async def emit(self, event, data, namespace, room=None, skip_sid=None,
|
||
|
|
callback=None, **kwargs):
|
||
|
|
"""Emit a message to a single client, a room, or all the clients
|
||
|
|
connected to the namespace.
|
||
|
|
|
||
|
|
Note: this method is a coroutine.
|
||
|
|
"""
|
||
|
|
if namespace not in self.rooms:
|
||
|
|
return
|
||
|
|
tasks = []
|
||
|
|
if not isinstance(skip_sid, list):
|
||
|
|
skip_sid = [skip_sid]
|
||
|
|
for sid, eio_sid in self.get_participants(namespace, room):
|
||
|
|
if sid not in skip_sid:
|
||
|
|
if callback is not None:
|
||
|
|
id = self._generate_ack_id(sid, callback)
|
||
|
|
else:
|
||
|
|
id = None
|
||
|
|
tasks.append(asyncio.create_task(
|
||
|
|
self.server._emit_internal(eio_sid, event, data,
|
||
|
|
namespace, id)))
|
||
|
|
if tasks == []: # pragma: no cover
|
||
|
|
return
|
||
|
|
await asyncio.wait(tasks)
|
||
|
|
|
||
|
|
async def disconnect(self, sid, namespace, **kwargs):
|
||
|
|
"""Disconnect a client.
|
||
|
|
|
||
|
|
Note: this method is a coroutine.
|
||
|
|
"""
|
||
|
|
return super().disconnect(sid, namespace, **kwargs)
|
||
|
|
|
||
|
|
async def close_room(self, room, namespace):
|
||
|
|
"""Remove all participants from a room.
|
||
|
|
|
||
|
|
Note: this method is a coroutine.
|
||
|
|
"""
|
||
|
|
return super().close_room(room, namespace)
|
||
|
|
|
||
|
|
async def trigger_callback(self, sid, id, data):
|
||
|
|
"""Invoke an application callback.
|
||
|
|
|
||
|
|
Note: this method is a coroutine.
|
||
|
|
"""
|
||
|
|
callback = None
|
||
|
|
try:
|
||
|
|
callback = self.callbacks[sid][id]
|
||
|
|
except KeyError:
|
||
|
|
# if we get an unknown callback we just ignore it
|
||
|
|
self._get_logger().warning('Unknown callback received, ignoring.')
|
||
|
|
else:
|
||
|
|
del self.callbacks[sid][id]
|
||
|
|
if callback is not None:
|
||
|
|
ret = callback(*data)
|
||
|
|
if asyncio.iscoroutine(ret):
|
||
|
|
try:
|
||
|
|
await ret
|
||
|
|
except asyncio.CancelledError: # pragma: no cover
|
||
|
|
pass
|