/
redis_broker.py
62 lines (47 loc) · 1.72 KB
/
redis_broker.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
import redis
from django_q.brokers import Broker
from django_q.conf import Conf, logger
try:
import django_redis
except ImportError:
django_redis = None
class Redis(Broker):
def __init__(self, list_key=Conf.PREFIX):
super(Redis, self).__init__(list_key='django_q:{}:q'.format(list_key))
def enqueue(self, task):
return self.connection.rpush(self.list_key, task)
def dequeue(self):
task = self.connection.blpop(self.list_key, 1)
if task:
return [(None, task[1])]
def queue_size(self):
return self.connection.llen(self.list_key)
def delete_queue(self):
return self.connection.delete(self.list_key)
def purge_queue(self):
return self.connection.ltrim(self.list_key, 1, 0)
def ping(self):
try:
return self.connection.ping()
except redis.ConnectionError as e:
logger.error('Can not connect to Redis server.')
raise e
def info(self):
if not self._info:
info = self.connection.info('server')
self._info = 'Redis {}'.format(info['redis_version'])
return self._info
def set_stat(self, key, value, timeout):
self.connection.set(key, value, timeout)
def get_stat(self, key):
if self.connection.exists(key):
return self.connection.get(key)
def get_stats(self, pattern):
keys = self.connection.keys(pattern=pattern)
if keys:
return self.connection.mget(keys)
@staticmethod
def get_connection(list_key=Conf.PREFIX):
if django_redis and Conf.DJANGO_REDIS:
return django_redis.get_redis_connection(Conf.DJANGO_REDIS)
return redis.StrictRedis(**Conf.REDIS)