[#136690921] custom max_number_of_messages (#3)

admetricks · Dec 27, 2016 · e4e7e75 · e4e7e75
1 parent dbbefad
commit e4e7e75
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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 =