set statement_timeout for non-pooled connection

neondatabase · Feb 28, 2025 · de6bfde · de6bfde
1 parent d823de2
commit de6bfde
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test_runner/performance/test_perf_oltp_large_tenant.py b/test_runner/performance/test_perf_oltp_large_tenant.py
@@ -31,6 +31,10 @@ def run_test_pgbench(env: PgCompare, custom_scripts: str, duration: int):
     options = env.pg.default_options.get("options", "")
     # drop password from the connection string by passing password=None and set password separately
     connstr = env.pg.connstr(password=None, options=options)
+    # if connstr does not contain pooler we can set statement_timeout to 0
+    if "pooler" not in connstr:
+        options = "-cstatement_timeout=0 " + env.pg.default_options.get("options", "")
+        connstr = env.pg.connstr(password=None, options=options)
 
     script_args = [
         "pgbench",