Destination Snowflake: Use safe executeMetadataQuery call (#37857)

airbytehq · May 6, 2024 · 3cca1c0 · 3cca1c0
1 parent 3c62abe
commit 3cca1c0
Show file tree

Hide file tree

Showing 3 changed files with 198 additions and 191 deletions.
diff --git a/airbyte-integrations/connectors/destination-snowflake/metadata.yaml b/airbyte-integrations/connectors/destination-snowflake/metadata.yaml
@@ -5,7 +5,7 @@ data:
   connectorSubtype: database
   connectorType: destination
   definitionId: 424892c4-daac-4491-b35d-c6688ba547ba
-  dockerImageTag: 3.7.1
+  dockerImageTag: 3.7.2
   dockerRepository: airbyte/destination-snowflake
   documentationUrl: https://docs.airbyte.com/integrations/destinations/snowflake
   githubIssueLabel: destination-snowflake

diff --git a/...rbyte/integrations/destination/snowflake/typing_deduping/SnowflakeDestinationHandler.java b/...rbyte/integrations/destination/snowflake/typing_deduping/SnowflakeDestinationHandler.java
@@ -132,12 +132,18 @@ private InitialRawTableStatus getInitialRawTableState(final StreamId id, final D
     if (destinationSyncMode == DestinationSyncMode.OVERWRITE) {
       return new InitialRawTableStatus(false, false, Optional.empty());
     }
-    final ResultSet tables = database.getMetaData().getTables(
-        databaseName,
-        id.getRawNamespace(),
-        id.getRawName(),
-        null);
-    if (!tables.next()) {
+    final boolean tableExists = database.executeMetadataQuery(databaseMetaData -> {
+      LOGGER.info("Retrieving table from Db metadata: {} {}",
+          id.getRawNamespace(),
+          id.getRawName());
+      try (final ResultSet tables = databaseMetaData.getTables(databaseName, id.getRawNamespace(), id.getRawName(), null)) {
+        return tables.next();
+      } catch (SQLException e) {
+        LOGGER.error("Failed to retrieve table metadata", e);
+        throw new RuntimeException(e);
+      }
+    });
+    if (!tableExists) {
       return new InitialRawTableStatus(false, false, Optional.empty());
     }
     // Snowflake timestamps have nanosecond precision, so decrement by 1ns