adds pluggable brokers

* added redis broker
 * added django_redis broker
 * added task acknowledgement
This commit is contained in:
Ilan Steemers
2015-08-29 18:54:07 +02:00
parent 0de742b417
commit 0740c90148
12 changed files with 247 additions and 163 deletions
+47
View File
@@ -0,0 +1,47 @@
from django_q.conf import Conf
class Broker(object):
def __init__(self, list_key=Conf.Q_LIST):
self.connection = self.get_connection()
self.list_key = list_key
def enqueue(self, task):
pass
def dequeue(self):
pass
def queue_size(self):
pass
def delete_queue(self, list_key=None):
pass
def acknowledge(self, ack_id):
pass
def ping(self):
pass
def set(self, key, value, timeout):
pass
def get(self, key):
pass
def get_pattern(self, pattern):
pass
@staticmethod
def get_connection():
return 0
def get_broker(list_key=Conf.Q_LIST):
if Conf.REDIS:
from brokers import redis
return redis.Redis(list_key=list_key)
elif Conf.DJANGO_REDIS:
from brokers import django_redis
return django_redis.DjangoRedis(list_key=list_key)
+10
View File
@@ -0,0 +1,10 @@
import django_redis
from django_q.brokers import redis
from django_q.conf import Conf
class DjangoRedis(redis.Redis):
@staticmethod
def get_connection():
return django_redis.get_redis_connection(Conf.DJANGO_REDIS)
+43
View File
@@ -0,0 +1,43 @@
import redis
from django_q.brokers import Broker
from django_q.conf import Conf, logger
class Redis(Broker):
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, list_key=None):
list_key = list_key if list_key else self.list_key
return self.connection.delete(list_key)
def ping(self):
try:
return self.connection.ping()
except Exception as e:
logger.error('Can not connect to Redis server.')
raise e
def set(self, key, value, timeout):
self.connection.set(key, value, timeout)
def get(self, key):
if self.connection.exists(key):
return self.connection.get(key)
def get_pattern(self, pattern):
keys = self.connection.keys(pattern=pattern)
if keys:
return self.connection.mget(keys)
@staticmethod
def get_connection():
return redis.StrictRedis(**Conf.REDIS)