mirror of
https://github.com/django-q2/django-q2.git
synced 2026-09-27 12:38:12 +08:00
58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
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(Sqs, self).__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)
|