From b5578381cecdaa133021a79b3a343b188280da7a Mon Sep 17 00:00:00 2001
From: bmc-msft <41130664+bmc-msft@users.noreply.github.com>
Date: Wed, 4 Nov 2020 15:41:05 -0500
Subject: [PATCH] default TTL for queued messages to infinite (#259)

---
 src/api-service/__app__/onefuzzlib/azure/queue.py | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/api-service/__app__/onefuzzlib/azure/queue.py b/src/api-service/__app__/onefuzzlib/azure/queue.py
index fa4ef706a..95557b702 100644
--- a/src/api-service/__app__/onefuzzlib/azure/queue.py
+++ b/src/api-service/__app__/onefuzzlib/azure/queue.py
@@ -23,6 +23,8 @@ from .creds import get_storage_account_name_key
 
 QueueNameType = Union[str, UUID]
 
+DEFAULT_TTL = -1
+
 
 @cached(ttl=60)
 def get_queue_client(account_id: str) -> QueueServiceClient:
@@ -102,11 +104,17 @@ def send_message(
     message: bytes,
     *,
     account_id: str,
+    visibility_timeout: Optional[int] = None,
+    time_to_live: int = DEFAULT_TTL,
 ) -> None:
     queue = get_queue(name, account_id=account_id)
     if queue:
         try:
-            queue.send_message(base64.b64encode(message).decode())
+            queue.send_message(
+                base64.b64encode(message).decode(),
+                visibility_timeout=visibility_timeout,
+                time_to_live=time_to_live,
+            )
         except ResourceNotFoundError:
             pass
 
@@ -163,6 +171,7 @@ def queue_object(
     *,
     account_id: str,
     visibility_timeout: Optional[int] = None,
+    time_to_live: int = DEFAULT_TTL,
 ) -> bool:
     queue = get_queue(name, account_id=account_id)
     if not queue:
@@ -170,7 +179,9 @@ def queue_object(
 
     encoded = base64.b64encode(message.json(exclude_none=True).encode()).decode()
     try:
-        queue.send_message(encoded, visibility_timeout=visibility_timeout)
+        queue.send_message(
+            encoded, visibility_timeout=visibility_timeout, time_to_live=time_to_live
+        )
         return True
     except ResourceNotFoundError:
         return False