IronMQ broker

django_q/brokers/__init__.py

@@ -153,6 +153,9 @@ def get_broker(list_key=Conf.PREFIX):
     if Conf.DISQUE_NODES:
         from brokers import disque
         return disque.Disque(list_key=list_key)
+    elif Conf.IRON_MQ:
+        from brokers import ironmq
+        return ironmq.IronMQBroker(list_key=list_key)
     # default to redis
     else:
         from brokers import redis_broker

django_q/brokers/ironmq.py

@@ -0,0 +1,44 @@
+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 info(self):
+        return 'IronMQ'
+
+    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()
+
+    def delete(self, task_id):
+        return self.connection.delete(task_id)['msg']
+
+    def fail(self, task_id):
+        self.delete(task_id)
+
+    def acknowledge(self, task_id):
+        return self.delete(task_id)
+
+    @staticmethod
+    def get_connection(list_key=Conf.PREFIX):
+        ironmq = IronMQ(name=None, **Conf.IRON_MQ)
+        return ironmq.queue(queue_name=list_key)