Close StatementClient when request is timed out

takezoe · wendigo · commit 641f574b9f31 · 2025-01-29T11:38:00.000+01:00
diff --git a/client/trino-client/src/main/java/io/trino/client/StatementClientV1.java b/client/trino-client/src/main/java/io/trino/client/StatementClientV1.java
@@ -403,6 +403,7 @@ private boolean executeRequest(Request request, String taskName, OptionalLong ma
             if (attempts > 0) {
                 Duration sinceStart = Duration.nanosSince(start);
                 if (sinceStart.compareTo(requestTimeoutNanos) > 0) {
+                    close();
                     state.compareAndSet(State.RUNNING, State.CLIENT_ERROR);
                     throw new RuntimeException(format("Error fetching next (attempts: %s, duration: %s)", attempts, sinceStart), cause);
                 }

Original file line number	Diff line number	Diff line change
`@@ -403,6 +403,7 @@ private boolean executeRequest(Request request, String taskName, OptionalLong ma`
`403`	`403`	`if (attempts > 0) {`
`404`	`404`	`Duration sinceStart = Duration.nanosSince(start);`
`405`	`405`	`if (sinceStart.compareTo(requestTimeoutNanos) > 0) {`
	`406`	`+ close();`
`406`	`407`	`state.compareAndSet(State.RUNNING, State.CLIENT_ERROR);`
`407`	`408`	`throw new RuntimeException(format("Error fetching next (attempts: %s, duration: %s)", attempts, sinceStart), cause);`
`408`	`409`	`}`