only check PK

airbytehq · Aug 21, 2023 · 1b49b18 · 1b49b18
1 parent f7d5809
commit 1b49b18
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.../io/airbyte/integrations/destination/snowflake/typing_deduping/SnowflakeSqlGenerator.java b/.../io/airbyte/integrations/destination/snowflake/typing_deduping/SnowflakeSqlGenerator.java
@@ -246,7 +246,7 @@ WHEN TYPEOF("_airbyte_data":"${column_name}") != 'ARRAY'
   String validatePrimaryKeys(final StreamId id,
                              final List<ColumnId> primaryKeys,
                              final LinkedHashMap<ColumnId, AirbyteType> streamColumns) {
-    if (streamColumns.keySet().stream().anyMatch(c -> c.originalName().contains("`"))) {
+    if (primaryKeys.stream().anyMatch(c -> c.originalName().contains("`"))) {
       // TODO why is snowflake throwing a bizarre error when we try to use a column with a backtick in it?
       // E.g. even this trivial procedure fails: (it should return the string `'foo`bar')
       // execute immediate 'BEGIN RETURN \'foo`bar\'; END;'