mirror of
https://github.com/django-q2/django-q2.git
synced 2026-10-10 18:00:56 +08:00
Adds more brokers
* added IronMQ * added Aws SQS * removed SafeRedis. During tests the 'safe' part wasn't consistent enough
This commit is contained in:
@@ -0,0 +1,57 @@
|
||||
import os
|
||||
from django_q.conf import Conf
|
||||
from django_q.brokers import Broker
|
||||
import boto.sqs
|
||||
from boto.sqs.message import Message
|
||||
|
||||
|
||||
class Sqs(Broker):
|
||||
def __init__(self, list_key=Conf.PREFIX):
|
||||
super().__init__(list_key)
|
||||
self.queue = self.get_queue()
|
||||
|
||||
def enqueue(self, task):
|
||||
m = Message()
|
||||
m.set_body(task)
|
||||
self.queue.write(m)
|
||||
return m.id
|
||||
|
||||
def dequeue(self):
|
||||
rs = self.queue.get_messages(visibility_timeout=Conf.RETRY or 30)
|
||||
if rs:
|
||||
m = rs[0]
|
||||
return m.receipt_handle, m.get_body()
|
||||
|
||||
def acknowledge(self, task_id):
|
||||
return self.delete(task_id)
|
||||
|
||||
def queue_size(self):
|
||||
return self.queue.count()
|
||||
|
||||
def delete(self, task_id):
|
||||
m = Message()
|
||||
m.receipt_handle = task_id
|
||||
return self.queue.delete_message(m)
|
||||
|
||||
def delete_queue(self):
|
||||
self.connection.delete_queue(self.queue)
|
||||
|
||||
def purge_queue(self):
|
||||
self.queue.purge()
|
||||
|
||||
def ping(self):
|
||||
try:
|
||||
self.connection.get_all_queues()
|
||||
return True
|
||||
except Exception as e:
|
||||
raise e
|
||||
|
||||
@staticmethod
|
||||
def get_connection(list_key=Conf.PREFIX):
|
||||
conn = boto.sqs.connect_to_region(Conf.SQS['region'],
|
||||
aws_access_key_id=Conf.SQS['aws_access_key_id'],
|
||||
aws_secret_access_key=Conf.SQS['aws_secret_access_key'])
|
||||
return conn
|
||||
|
||||
def get_queue(self):
|
||||
return self.connection.create_queue(self.list_key)
|
||||
Reference in New Issue
Block a user