Wrap names with double quotes

trocco-io · Jun 11, 2024 · 7fc9db4 · 7fc9db4
1 parent 5fd4e49
commit 7fc9db4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/main/java/org/embulk/output/SnowflakeOutputPlugin.java b/src/main/java/org/embulk/output/SnowflakeOutputPlugin.java
@@ -167,8 +167,12 @@ protected JdbcOutputConnector getConnector(PluginTask task, boolean retryableMet
         throw new ConfigException(e);
       }
     }
+    if (t.getWarehouse().startsWith("\"") && t.getWarehouse().endsWith("\"")) {
+      props.setProperty("warehouse", t.getWarehouse());
+    } else {
+      props.setProperty("warehouse", "\"" + t.getWarehouse() + "\"");
+    }
 
-    props.setProperty("warehouse", t.getWarehouse());
     props.setProperty("db", t.getDatabase());
     props.setProperty("schema", t.getSchema());
     if (!t.getRole().isEmpty()) {