Code import

venv/lib/python2.7/site-packages/socketio/asyncio_redis_manager.py

@@ -0,0 +1,78 @@
+import pickle
+from urllib.parse import urlparse
+
+try:
+    import aioredis
+except ImportError:
+    aioredis = None
+
+from .asyncio_pubsub_manager import AsyncPubSubManager
+
+
+def _parse_redis_url(url):
+    p = urlparse(url)
+    if p.scheme != 'redis':
+        raise ValueError('Invalid redis url')
+    if ':' in p.netloc:
+        host, port = p.netloc.split(':')
+        port = int(port)
+    else:
+        host = p.netloc or 'localhost'
+        port = 6379
+    if p.path:
+        db = int(p.path[1:])
+    else:
+        db = 0
+    if not host:
+        raise ValueError('Invalid redis hostname')
+    return host, port, db
+
+
+class AsyncRedisManager(AsyncPubSubManager):  # pragma: no cover
+    """Redis based client manager for asyncio servers.
+
+    This class implements a Redis backend for event sharing across multiple
+    processes. Only kept here as one more example of how to build a custom
+    backend, since the kombu backend is perfectly adequate to support a Redis
+    message queue.
+
+    To use a Redis backend, initialize the :class:`Server` instance as
+    follows::
+
+        server = socketio.Server(client_manager=socketio.AsyncRedisManager(
+            'redis://hostname:port/0'))
+
+    :param url: The connection URL for the Redis server. For a default Redis
+                store running on the same host, use ``redis://``.
+    :param channel: The channel name on which the server sends and receives
+                    notifications. Must be the same in all the servers.
+    :param write_only: If set ot ``True``, only initialize to emit events. The
+                       default of ``False`` initializes the class for emitting
+                       and receiving.
+    """
+    name = 'aioredis'
+
+    def __init__(self, url='redis://localhost:6379/0', channel='socketio',
+                 write_only=False):
+        if aioredis is None:
+            raise RuntimeError('Redis package is not installed '
+                               '(Run "pip install aioredis" in your '
+                               'virtualenv).')
+        self.host, self.port, self.db = _parse_redis_url(url)
+        self.pub = None
+        self.sub = None
+        super().__init__(channel=channel, write_only=write_only)
+
+    async def _publish(self, data):
+        if self.pub is None:
+            self.pub = await aioredis.create_redis((self.host, self.port),
+                                                   db=self.db)
+        return await self.pub.publish(self.channel, pickle.dumps(data))
+
+    async def _listen(self):
+        if self.sub is None:
+            self.sub = await aioredis.create_redis((self.host, self.port),
+                                                   db=self.db)
+            self.ch = (await self.sub.subscribe(self.channel))[0]
+        while True:
+            return await self.ch.get()