Handle exception when checking for closed connection (#314)

vertica · Feb 9, 2022 · 71b1e9c · 71b1e9c
1 parent 98b5621
commit 71b1e9c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/connector/src/main/scala/com/vertica/spark/datasource/jdbc/VerticaJdbcLayer.scala b/connector/src/main/scala/com/vertica/spark/datasource/jdbc/VerticaJdbcLayer.scala
@@ -307,7 +307,11 @@ class VerticaJdbcLayer(cfg: JDBCConfig) extends JdbcLayerInterface {
 
   def isClosed(): Boolean = {
     logger.debug("Checking if connection is closed.")
-    this.connection.fold(_ => true, conn => conn.isClosed())
+    try {
+      this.connection.fold(_ => true, conn => conn.isClosed())
+    } catch {
+      case _ : Throwable => true
+    }
   }
 
   def configureSession(fileStoreLayer: FileStoreLayerInterface): ConnectorResult[Unit] = {