訊息佇列

某次在某乎上看到有人提到訊息佇列的問題，然後有人在回答裡提到了Redis，接著便有人在評論裡指出：Redis是快取，不是訊息佇列。

但不幸的是，Redis的確提供一個簡易的訊息佇列機制，可以用於一些要求不那麼高的場合。

方法就是利用Redis的列表型別的push和pop操作。

我對前文所介紹的Redis Cache作了一點簡單的擴充套件，增加了訊息佇列功能。

實現

程式碼基本就這麼點：

class RedisMQ(RedisCache):
    def __init__(self, dbname, host='localhost', port=6379, db=0
 
):
        super(RedisMQ, self).__init__(dbname, host, port, db)

    def push(self, channel, data):
        ch = self._getkey("channel", channel)
        self.db.lpush(ch, self.SERIALIZER.dumps(data))

    def pop(self, channel, timeout=5):
        ch = self._getkey("channel", channel)
        msg = self.db.brpop(ch, timeout)
        return
 
 self.SERIALIZER.loads(msg[1]) if msg else None


class Channel(object):
    MQ = RedisMQ("msgqueue")

    def __init__(self, channel):
        self.channel = channel

    def push(self, **kwargs):
        Channel.MQ.push(self.channel, kwargs)

    def pop(self):
        return Channel.MQ.pop(self.channel)
 

用法

訊息生產者

ch = Channel("test")
ch.push(a=123,b="hello")

訊息消費者，可能是另一個執行緒，甚至是另一個程序，甚至是另外一臺主機——只要它們共用同一個redis即可。

ch = Channel("test")
while msg=ch.pop():
    # msg: {"a": 123, "b": "hello"}
# ch is empty