Fix AmazonTimestreamQuery client object leak (#48)

awslabs · Jan 6, 2024 · 269b4f2 · 269b4f2
1 parent f88ab6b
commit 269b4f2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/jdbc/src/main/java/software/amazon/timestream/jdbc/TimestreamConnection.java b/jdbc/src/main/java/software/amazon/timestream/jdbc/TimestreamConnection.java
@@ -322,6 +322,8 @@ public boolean isValid(int timeout) throws SQLException {
     } catch (Exception e) {
       LOGGER.error("Connection is no longer valid: {}", e.getMessage());
       return false;
+    } finally {
+      client.shutdown();
     }
   }