cloud_pubsub_max_messages

roticagas · Jan 14, 2019 · a14d878 · a14d878
1 parent 45136d9
commit a14d878
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/PubSubRunner/runner_application.py b/PubSubRunner/runner_application.py
@@ -94,7 +94,8 @@ def run(self, job=None, internal_server=True):
         CloudUtil.subscribe_message(self.config.cloud_project,
                                     self.config.cloud_pubsub_subscribe_subscription,
                                     self.subscribe_processing,
-                                    self.config.cloud_pubsub_max_lease_duration)
+                                    self.config.cloud_pubsub_max_lease_duration,
+                                    self.config.cloud_pubsub_max_messages)
         if job is not None and callable(job):
             job()
         if internal_server:

diff --git a/PubSubRunner/runner_config.py b/PubSubRunner/runner_config.py
@@ -14,6 +14,7 @@ class RunnerConfig:
         ('CLOUD_PUBSUB_CHECK', 'true'),
         ('CLOUD_PUBSUB_ACK', 'true'),
         ('CLOUD_PUBSUB_DEAD_LETTER_ACK', 'false'),
+        ('CLOUD_PUBSUB_MAX_MESSAGES', '10'),
     ]
 
     def __init__(self):
@@ -29,3 +30,4 @@ def __init__(self):
         self.cloud_pubsub_check = env[8] in ['True', 'true', 'TRUE']
         self.cloud_pubsub_ack = env[9] in ['True', 'true', 'TRUE']
         self.cloud_pubsub_dead_letter_ack = env[10] in ['True', 'true', 'TRUE']
+        self.cloud_pubsub_max_messages = env[11]