mirror of
https://github.com/django-q2/django-q2.git
synced 2026-10-02 20:18:11 +08:00
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
import redis
|
|
from django_q.brokers import Broker
|
|
from django_q.conf import Conf
|
|
|
|
|
|
class Disque(Broker):
|
|
def enqueue(self, task):
|
|
return self.connection.execute_command(
|
|
'ADDJOB {} {} 500 RETRY {}'.format(self.list_key, task, Conf.RETRY)).decode()
|
|
|
|
def dequeue(self):
|
|
task = self.connection.execute_command('GETJOB TIMEOUT 1000 FROM {}'.format(self.list_key))
|
|
if task:
|
|
return task[0][1].decode(), task[0][2].decode()
|
|
|
|
def queue_size(self):
|
|
return self.connection.execute_command('QLEN {}'.format(self.list_key))
|
|
|
|
def acknowledge(self, ack_id):
|
|
return self.connection.execute_command('ACKJOB {}'.format(ack_id))
|
|
|
|
def ping(self):
|
|
return self.connection.ping()
|
|
|
|
@staticmethod
|
|
def get_connection():
|
|
for node in Conf.DISQUE:
|
|
host, port = node.split(':')
|
|
redis_client = redis.Redis(host, int(port))
|
|
try:
|
|
redis_client.ping()
|
|
redis_client.decode_responses = True
|
|
return redis_client
|
|
except redis.exceptions.ConnectionError:
|
|
pass
|
|
raise ConnectionError('Could not connect to any Disque nodes')
|