mirror of
https://github.com/django-q2/django-q2.git
synced 2026-10-06 09:18:11 +08:00
adds pluggable brokers
* added redis broker * added django_redis broker * added task acknowledgement
This commit is contained in:
@@ -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)
|
||||
@@ -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)
|
||||
@@ -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)
|
||||
Reference in New Issue
Block a user