set max retry for trino connection (#4898)

maskarb · web-flow · commit cd391ccc389f · 2024-02-07T21:16:38.000+02:00
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)
 

Original file line number	Diff line number	Diff line change
`@@ -106,6 +106,7 @@ def connect(**connect_args):`
`106`	`106`	`),`
`107`	`107`	`"schema": connect_args["schema"],`
`108`	`108`	`"legacy_primitive_types": connect_args.get("legacy_primitive_types", False),`
	`109`	`+ "max_attempts": 9, # based on exponential backoff used in the trino lib, 9 max retries with take ~100 seconds`
`109`	`110`	`}`
`110`	`111`	`return trino.dbapi.connect(**trino_connect_args)`
`111`	`112`