pythonasyncioeventletgeventlong-pollinglow-latencysocket-iosocketiosocketio-serverweb-serverwebsocket
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
2.7 KiB
78 lines
2.7 KiB
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()
|
|
|