diff --git a/ci/templates/tox.ini b/ci/templates/tox.ini index 56ceb3e..3871d42 100644 --- a/ci/templates/tox.ini +++ b/ci/templates/tox.ini @@ -8,6 +8,9 @@ envlist = report [testenv] +basepython = + {docs,spell}: python2.7 + {bootstrap,clean,check,report,extension-coveralls,coveralls,codecov}: python3.4 setenv = PYTHONPATH={toxinidir}/tests PYTHONUNBUFFERED=yes @@ -76,7 +79,8 @@ commands = [testenv:codecov] deps = - codecov + codecov==1.6.3 + coverage==4.0.3 skip_install = true usedevelop = false commands = diff --git a/src/python_sqs_consumer/sqs_consumer_pool.py b/src/python_sqs_consumer/sqs_consumer_pool.py index eba7689..a4ba52c 100644 --- a/src/python_sqs_consumer/sqs_consumer_pool.py +++ b/src/python_sqs_consumer/sqs_consumer_pool.py @@ -10,12 +10,15 @@ class SQSConsumerPool(object): workers_pool = None - def __init__(self, max_threads, processor_class, queue_name): + def __init__( + self, max_threads, processor_class, queue_name, max_number_of_messages=1, wait_time_seconds=10): self.max_threads = max_threads self.workers_pool = [None] * self.max_threads self.processor_class = processor_class self.queue_name = queue_name + self.max_number_of_messages = max_number_of_messages + self.wait_time_seconds = wait_time_seconds def run(self): while True: @@ -39,5 +42,6 @@ def consume(self, queue): logger.debug('Consumer waiting for incomming message...') if queue is not None: - messages = queue.receive_messages(MaxNumberOfMessages=1, WaitTimeSeconds=10) + messages = queue.receive_messages( + MaxNumberOfMessages=self.max_number_of_messages, WaitTimeSeconds=self.wait_time_seconds) processor.process(messages) diff --git a/tox.ini b/tox.ini index 76dcdd9..b3873f5 100644 --- a/tox.ini +++ b/tox.ini @@ -7,6 +7,9 @@ envlist = report [testenv] +basepython = + {docs,spell}: python2.7 + {bootstrap,clean,check,report,extension-coveralls,coveralls,codecov}: python3.4 setenv = PYTHONPATH={toxinidir}/tests PYTHONUNBUFFERED=yes @@ -75,7 +78,8 @@ commands = [testenv:codecov] deps = - codecov + codecov==1.6.3 + coverage==4.0.3 skip_install = true usedevelop = false commands =