Browse Source

apply similar fix to other client managers

pull/1476/head
Miguel Grinberg 5 days ago
parent
commit
bddd8e3c0a
Failed to extract signature
  1. 2
      src/socketio/async_aiopika_manager.py
  2. 2
      src/socketio/async_redis_manager.py
  3. 2
      src/socketio/redis_manager.py
  4. 2
      src/socketio/zmq_manager.py

2
src/socketio/async_aiopika_manager.py

@ -43,12 +43,12 @@ class AsyncAioPikaManager(AsyncPubSubManager): # pragma: no cover
raise RuntimeError('aio_pika package is not installed ' raise RuntimeError('aio_pika package is not installed '
'(Run "pip install aio_pika" in your ' '(Run "pip install aio_pika" in your '
'virtualenv).') 'virtualenv).')
super().__init__(channel=channel, write_only=write_only, logger=logger)
self.url = url self.url = url
self._lock = asyncio.Lock() self._lock = asyncio.Lock()
self.publisher_connection = None self.publisher_connection = None
self.publisher_channel = None self.publisher_channel = None
self.publisher_exchange = None self.publisher_exchange = None
super().__init__(channel=channel, write_only=write_only, logger=logger)
async def _connection(self): async def _connection(self):
return await aio_pika.connect_robust(self.url) return await aio_pika.connect_robust(self.url)

2
src/socketio/async_redis_manager.py

@ -52,9 +52,9 @@ class AsyncRedisManager(AsyncPubSubManager): # pragma: no cover
'(Run "pip install redis" in your virtualenv).') '(Run "pip install redis" in your virtualenv).')
if not hasattr(aioredis.Redis, 'from_url'): if not hasattr(aioredis.Redis, 'from_url'):
raise RuntimeError('Version 2 of aioredis package is required.') raise RuntimeError('Version 2 of aioredis package is required.')
super().__init__(channel=channel, write_only=write_only, logger=logger)
self.redis_url = url self.redis_url = url
self.redis_options = redis_options or {} self.redis_options = redis_options or {}
super().__init__(channel=channel, write_only=write_only, logger=logger)
self._redis_connect() self._redis_connect()
def _redis_connect(self): def _redis_connect(self):

2
src/socketio/redis_manager.py

@ -75,10 +75,10 @@ class RedisManager(PubSubManager): # pragma: no cover
raise RuntimeError('Redis package is not installed ' raise RuntimeError('Redis package is not installed '
'(Run "pip install redis" in your ' '(Run "pip install redis" in your '
'virtualenv).') 'virtualenv).')
super().__init__(channel=channel, write_only=write_only, logger=logger)
self.redis_url = url self.redis_url = url
self.redis_options = redis_options or {} self.redis_options = redis_options or {}
self._redis_connect() self._redis_connect()
super().__init__(channel=channel, write_only=write_only, logger=logger)
def initialize(self): def initialize(self):
super().initialize() super().initialize()

2
src/socketio/zmq_manager.py

@ -57,6 +57,7 @@ class ZmqManager(PubSubManager): # pragma: no cover
if not (url.startswith('zmq+tcp://') and r.search(url)): if not (url.startswith('zmq+tcp://') and r.search(url)):
raise RuntimeError('unexpected connection string: ' + url) raise RuntimeError('unexpected connection string: ' + url)
super().__init__(channel=channel, write_only=write_only, logger=logger)
url = url.replace('zmq+', '') url = url.replace('zmq+', '')
(sink_url, sub_port) = url.split('+') (sink_url, sub_port) = url.split('+')
sink_port = sink_url.split(':')[-1] sink_port = sink_url.split(':')[-1]
@ -72,7 +73,6 @@ class ZmqManager(PubSubManager): # pragma: no cover
self.sink = sink self.sink = sink
self.sub = sub self.sub = sub
self.channel = channel self.channel = channel
super().__init__(channel=channel, write_only=write_only, logger=logger)
def _publish(self, data): def _publish(self, data):
pickled_data = pickle.dumps( pickled_data = pickle.dumps(

Loading…
Cancel
Save