From 7fc9db4adf538ea4eb31843ea549253404a063c4 Mon Sep 17 00:00:00 2001
From: mzumi <masazumi.m@gmail.com>
Date: Mon, 10 Jun 2024 08:27:27 +0000
Subject: [PATCH] Wrap names with double quotes

---
 src/main/java/org/embulk/output/SnowflakeOutputPlugin.java | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/main/java/org/embulk/output/SnowflakeOutputPlugin.java b/src/main/java/org/embulk/output/SnowflakeOutputPlugin.java
index 7f34019..aeae0d2 100644
--- 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()) {