python redis搭建队列服务

# -*- coding:utf-8 -*-

import redis

# from redis_queue
r = redis.Redis(host='127.0.0.1', port=6379, password='liang')


class RedisQueue(object):
    def __init__(self, name, namespace='queue'):
        # redis的默认参数为：host='localhost', port=6379, db=0， 其中db为定义redis database的数量
        self.__db = redis.Redis(host='127.0.0.1', port=6379, password='liang',db=0)  # 链接数据库
        self.key = '%s:%s' % (namespace, name)

    def qsize(self):
        return self.__db.llen(self.key)  # 返回队列里面list内元素的数量

    def put(self, item):
        self.__db.rpush(self.key, item)  # 添加新元素到队列最右方

    def get_wait(self, timeout=None):
        # 返回队列第一个元素，如果为空则等待至有元素被加入队列（超时时间阈值为timeout，如果为None则一直等待）
        item = self.__db.blpop(self.key, timeout=timeout)
        # if item:
        #     item = item[1]  # 返回值为一个tuple
        return item

    def get_nowait(self):
        # 直接返回队列第一个元素，如果队列为空返回的是None
        item = self.__db.lpop(self.key)
        return item

入队列

# -*- coding:utf-8 -*-
from redis_queue import RedisQueue
import time

q = RedisQueue('rq')  # 新建队列名为rq
for i in range(5):
    q.put(i)
    print "input.py: data {} enqueue {}".format(i, time.strftime("%c"))
    # time.sleep(1)


print q.qsize()



出队列

from redis_queue import RedisQueue
import time

q = RedisQueue('rq')
while 1:
    result = q.get_wait()
    # if not result:
    #     break
    # print "output.py: data {} out of queue {}".format(result, time.strftime("%c"))
    # time.sleep(2)
    print result