mirror of
https://github.com/django-q2/django-q2.git
synced 2026-10-08 06:58:11 +08:00
adds fail method to broker
This commit is contained in:
@@ -55,6 +55,13 @@ class Broker(object):
|
|||||||
"""
|
"""
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
def fail(self, task_id):
|
||||||
|
"""
|
||||||
|
Fails a task message
|
||||||
|
:param task_id:
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
|
||||||
def ping(self):
|
def ping(self):
|
||||||
"""
|
"""
|
||||||
Checks whether the broker connection is available
|
Checks whether the broker connection is available
|
||||||
|
|||||||
@@ -33,6 +33,9 @@ class Sqs(Broker):
|
|||||||
m.receipt_handle = task_id
|
m.receipt_handle = task_id
|
||||||
return self.queue.delete_message(m)
|
return self.queue.delete_message(m)
|
||||||
|
|
||||||
|
def fail(self, task_id):
|
||||||
|
self.delete(task_id)
|
||||||
|
|
||||||
def delete_queue(self):
|
def delete_queue(self):
|
||||||
self.connection.delete_queue(self.queue)
|
self.connection.delete_queue(self.queue)
|
||||||
|
|
||||||
|
|||||||
@@ -27,6 +27,9 @@ class Disque(Broker):
|
|||||||
def delete(self, task_id):
|
def delete(self, task_id):
|
||||||
return self.connection.execute_command('DELJOB {}'.format(task_id))
|
return self.connection.execute_command('DELJOB {}'.format(task_id))
|
||||||
|
|
||||||
|
def fail(self, task_id):
|
||||||
|
return self.delete(task_id)
|
||||||
|
|
||||||
def delete_queue(self):
|
def delete_queue(self):
|
||||||
jobs = self.connection.execute_command('JSCAN QUEUE {}'.format(self.list_key))[1]
|
jobs = self.connection.execute_command('JSCAN QUEUE {}'.format(self.list_key))[1]
|
||||||
if jobs:
|
if jobs:
|
||||||
|
|||||||
@@ -32,6 +32,9 @@ class IronMQBroker(Broker):
|
|||||||
def delete(self, task_id):
|
def delete(self, task_id):
|
||||||
return self.connection.delete(task_id)['msg']
|
return self.connection.delete(task_id)['msg']
|
||||||
|
|
||||||
|
def fail(self, task_id):
|
||||||
|
self.delete(task_id)
|
||||||
|
|
||||||
def acknowledge(self, task_id):
|
def acknowledge(self, task_id):
|
||||||
return self.delete(task_id)
|
return self.delete(task_id)
|
||||||
|
|
||||||
|
|||||||
@@ -311,6 +311,7 @@ def pusher(task_queue, event, broker=None):
|
|||||||
task = signing.SignedPackage.loads(task[1])
|
task = signing.SignedPackage.loads(task[1])
|
||||||
except (TypeError, signing.BadSignature) as e:
|
except (TypeError, signing.BadSignature) as e:
|
||||||
logger.error(e)
|
logger.error(e)
|
||||||
|
broker.fail(ack_id)
|
||||||
continue
|
continue
|
||||||
task['ack_id'] = ack_id
|
task['ack_id'] = ack_id
|
||||||
task_queue.put(task)
|
task_queue.put(task)
|
||||||
|
|||||||
Reference in New Issue
Block a user