fork/django-q2
1
0
Fork 0
mirror of https://github.com/django-q2/django-q2.git synced 2026-10-05 04:48:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

SQS only supports max 10 messages in bulk mode

Browse Source 
* capped bulk at 10
* added tests
This commit is contained in:
Ilan Steemers
2015-09-09 19:34:06 +02:00
parent e4a5951f12
commit 0fe6c4cbde
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
django_q/brokers/aws_sqs.py
+3
View File
@@ -16,6 +16,9 @@ class Sqs(Broker):
return m.id return m.id
def dequeue(self): def dequeue(self):
# sqs supports max 10 messages in bulk
if Conf.BULK > 10:
Conf.BULK = 10
t = None t = None
if len(self.task_cache) > 0: if len(self.task_cache) > 0:
t = self.task_cache.pop() t = self.task_cache.pop()
django_q/tests/test_brokers.py
+4 -3
View File
@@ -187,10 +187,10 @@ def test_sqs():
task_id = broker.dequeue()[0] task_id = broker.dequeue()[0]
broker.fail(task_id) broker.fail(task_id)
# bulk test # bulk test
for i in range(5): for i in range(10):
broker.enqueue('test') broker.enqueue('test')
Conf.BULK = 5 Conf.BULK = 12
for i in range(5): for i in range(10):
task = broker.dequeue() task = broker.dequeue()
assert task is not None assert task is not None
broker.acknowledge(task[0]) broker.acknowledge(task[0])
@@ -202,4 +202,5 @@ def test_sqs():
broker.delete_queue() broker.delete_queue()
# back to django-redis # back to django-redis
Conf.SQS = None Conf.SQS = None
Conf.BULK = 1
Conf.DJANGO_REDIS = 'default' Conf.DJANGO_REDIS = 'default'