redis消息队列的使用
# -*- coding: utf-8 -*-
"""
Created on Tue Mar 26 15:58:34 2019
@author: Admin
"""
import redis
class RedisQueue(object):
def __init__(self, name, namespace='queue', **redis_kwargs):
# redis的默认参数为:host='localhost', port=6379, db=0, 其中db为定义redis database的数量
self.__db= redis.Redis(**redis_kwargs)
self.key = '%s:%s' %(namespace, name)
def qsize(self):
return self.__db.llen(self.key) # 返回队列里面list内元素的数量
def put(self, message):
self.__db.rpush(self.key, message) # 添加新元素到队列最右方
def get_tuple(self, timeout=None):
# 返回队列第一个元素,如果为空则等待至有元素被加入队列(超时时间阈值为timeout,如果为None则一直等待)
message = self.__db.blpop(self.key, timeout=timeout)
# if item:
# item = item[1] # 返回值为一个tuple (self.key,item)
return message
def get_first_element(self):
# 直接返回队列第一个元素,如果队列为空返回的是None
item = self.__db.lpop(self.key).decode()
return item
q = RedisQueue("dog")
#发布任务到dog
messg = "ssssssssss"
q.put(messg)
#在dog获取任务
dd = q.get_first_element()
print(dd)