revil-O/platypush
1
0
Fork 0
forked from platypush/platypush
Code Pull requests Activity

Making sure that Redis constructor only initializes the attributes without creating the object and performing a connection

Browse source
This commit is contained in:
Fabio Manganiello 2018-10-25 20:06:29 +02:00
parent 6ce348365f
commit 964c7b5cf0
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
platypush/backend/redis.py
View file

@ -31,19 +31,21 @@ class RedisBackend(Backend):
self.queue = queue self.queue = queue
self.redis_args = redis_args self.redis_args = redis_args
self.redis = Redis(**self.redis_args) self.redis = None
def _get_redis(self):
return Redis(**self.redis_args)
def send_message(self, msg, queue_name=None): def send_message(self, msg, queue_name=None):
if queue_name: if queue_name:
self.redis.rpush(queue_name, msg) self._get_redis().rpush(queue_name, msg)
else: else:
self.redis.rpush(self.queue, msg) self._get_redis().rpush(self.queue, msg)
def get_message(self, queue_name=None): def get_message(self, queue_name=None):
queue = queue_name or self.queue queue = queue_name or self.queue
msg = self.redis.blpop(queue)[1].decode('utf-8') msg = self._get_redis().blpop(queue)[1].decode('utf-8')
try: try:
msg = Message.build(json.loads(msg)) msg = Message.build(json.loads(msg))