Rooba
/
kombu
mirror of https://github.com/celery/kombu.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Redis: Adds missing method _handle_message

This commit is contained in:
Ask Solem 2011-01-28 15:45:51 +01:00
parent 495fd9cba4
commit 45e0086aed
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
kombu/transport/pyredis.py
View File

@ -160,6 +160,16 @@ class Channel(virtual.Channel):
self.subclient.subscribe(keys) self.subclient.subscribe(keys)
self._in_listen = True self._in_listen = True
def _handle_message(self, client, r):
if r[0] == "unsubscribe" and r[2] == 0:
client.subscribed = False
elif r[0] == "pmessage":
return {"type": r[0], "pattern": r[1],
"channel": r[2], "data": r[3]}
else:
return {"type": r[0], "pattern": None,
"channel": r[1], "data": r[2]}
def _receive(self): def _receive(self):
c = self.subclient c = self.subclient
response = None response = None
@ -168,7 +178,7 @@ class Channel(virtual.Channel):
except self.connection.connection_errors: except self.connection.connection_errors:
self._in_listen = False self._in_listen = False
if response is not None: if response is not None:
payload = c._handle_message(response) payload = self._handle_message(client, response)
if payload["type"] == "message": if payload["type"] == "message":
return (deserialize(payload["data"]), return (deserialize(payload["data"]),
self._fanout_to_queue[payload["channel"]]) self._fanout_to_queue[payload["channel"]])