set max retry for trino connection (#4898)

project-koku · Feb 7, 2024 · cd391cc · cd391cc
1 parent 0cdebf7
commit cd391cc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/koku/koku/trino_database.py b/koku/koku/trino_database.py
@@ -106,6 +106,7 @@ def connect(**connect_args):
         ),
         "schema": connect_args["schema"],
         "legacy_primitive_types": connect_args.get("legacy_primitive_types", False),
+        "max_attempts": 9,  # based on exponential backoff used in the trino lib, 9 max retries with take ~100 seconds
     }
     return trino.dbapi.connect(**trino_connect_args)