Merge pull request #11 from seeker25/add_ordering

Add in config for publish ordering

python/gcp-queue/src/gcp_queue/gcp_queue.py

@@ -62,6 +62,8 @@ def __init__(self, app: Flask = None):
         self.audience = None
         self.credentials_pub = None
         self.gcp_auth_key = None
+        self.enable_message_ordering = False
+        self.publisher_options = ()
         self.publisher_audience = None
         self.service_account_info = None
         self._publisher = None
@@ -73,6 +75,11 @@ def init_app(self, app: Flask):
         """Initializes the application"""
         self.app = app
         self.gcp_auth_key = app.config.get("GCP_AUTH_KEY")
+        self.enable_message_ordering = app.config.get("PUB_ENABLE_MESSAGE_ORDERING")
+        if self.enable_message_ordering:
+            self.publisher_options = pubsub_v1.types.PublisherOptions(
+                enable_message_ordering=True
+            )
         if self.gcp_auth_key:
             try:
                 audience = app.config.get(
@@ -102,12 +109,13 @@ def init_app(self, app: Flask):
     def publisher(self):
         """Returns the publisher"""
 
-        if not self._publisher and self.credentials_pub:
-            self._publisher = pubsub_v1.PublisherClient(
-                credentials=self.credentials_pub
-            )
-        elif not self._publisher:
-            self._publisher = pubsub_v1.PublisherClient()
+        if self._publisher:
+            return self._publisher
+
+        self._publisher = pubsub_v1.PublisherClient(
+            credentials=self.credentials_pub,
+            publisher_options=self.publisher_options
+        )
         return self._publisher
 
     @staticmethod