fix: clear interrupted flag after cancel

Clear the interrupted flag after cancelling a statement when using a direct executor. Fixes #1879
googleapis · Dec 26, 2024 · 2b2ed5f · 2b2ed5f
1 parent 91f575e
commit 2b2ed5f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/main/java/com/google/cloud/spanner/jdbc/AbstractJdbcStatement.java b/src/main/java/com/google/cloud/spanner/jdbc/AbstractJdbcStatement.java
@@ -16,6 +16,7 @@
 
 package com.google.cloud.spanner.jdbc;
 
+import com.google.cloud.spanner.ErrorCode;
 import com.google.cloud.spanner.Options;
 import com.google.cloud.spanner.Options.QueryOption;
 import com.google.cloud.spanner.ReadContext.QueryAnalyzeMode;
@@ -259,6 +260,11 @@ private <T> T doWithStatementTimeout(
       connection.recordClientLibLatencyMetric(executionDuration.toMillis());
       return result;
     } catch (SpannerException spannerException) {
+      if (spannerException.getErrorCode() == ErrorCode.CANCELLED && this.executingLock != null) {
+        // Clear the interrupted flag of the thread.
+        //noinspection ResultOfMethodCallIgnored
+        Thread.interrupted();
+      }
       throw JdbcSqlExceptionFactory.of(spannerException);
     } finally {
       if (this.executingLock != null) {

diff --git a/src/test/java/com/google/cloud/spanner/jdbc/JdbcStatementTimeoutTest.java b/src/test/java/com/google/cloud/spanner/jdbc/JdbcStatementTimeoutTest.java
@@ -161,7 +161,6 @@ public void testCancel() throws Exception {
                         message instanceof ExecuteSqlRequest
                             && ((ExecuteSqlRequest) message).getSql().equals(sql),
                     5000L);
-                System.out.println("Cancelling statement");
                 statement.cancel();
                 return null;
               });