mirror of
https://github.com/django-q2/django-q2.git
synced 2026-09-30 07:18:11 +08:00
Ilan Steemers
8ce441bed5
* added IronMQ * added Aws SQS * removed SafeRedis. During tests the 'safe' part wasn't consistent enough
39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
from django_q.conf import Conf
|
|
from django_q.brokers import Broker
|
|
from iron_mq import IronMQ
|
|
|
|
|
|
class IronMQBroker(Broker):
|
|
|
|
def enqueue(self, task):
|
|
return self.connection.post(task)['ids'][0]
|
|
|
|
def dequeue(self):
|
|
timeout = Conf.RETRY or None
|
|
task = self.connection.get(timeout=timeout, wait=1)['messages']
|
|
if task:
|
|
return task[0]['id'], task[0]['body']
|
|
|
|
def ping(self):
|
|
return self.connection.name == self.list_key
|
|
|
|
def queue_size(self):
|
|
return self.connection.size()
|
|
|
|
def delete_queue(self):
|
|
return self.connection.delete_queue()['msg']
|
|
|
|
def purge_queue(self):
|
|
return self.connection.clear()['msg']
|
|
|
|
def delete(self, task_id):
|
|
return self.connection.delete(task_id)['msg']
|
|
|
|
def acknowledge(self, task_id):
|
|
return self.delete(task_id)
|
|
|
|
@staticmethod
|
|
def get_connection(list_key=Conf.PREFIX):
|
|
ironmq = IronMQ(name=None, **Conf.IRONMQ)
|
|
return ironmq.queue(queue_name=list_key)
|