Skip to content

Commit 8124a10

Browse files
committed
add: role
1 parent 3816031 commit 8124a10

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

src/main/java/org/embulk/output/SnowflakeOutputPlugin.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,10 @@ public interface SnowflakePluginTask extends PluginTask {
5454
@ConfigDefault("\"public\"")
5555
public String getSchema();
5656

57+
@Config("role")
58+
@ConfigDefault("\"\"")
59+
public String getRole();
60+
5761
@Config("delete_stage")
5862
@ConfigDefault("false")
5963
public boolean getDeleteStage();
@@ -113,6 +117,9 @@ protected JdbcOutputConnector getConnector(PluginTask task, boolean retryableMet
113117
props.setProperty("warehouse", t.getWarehouse());
114118
props.setProperty("db", t.getDatabase());
115119
props.setProperty("schema", t.getSchema());
120+
if(!t.getRole().isEmpty()) {
121+
props.setProperty("role", t.getRole());
122+
}
116123

117124
// When CLIENT_METADATA_REQUEST_USE_CONNECTION_CTX is false (default),
118125
// getMetaData().getColumns() returns columns of the tables which table name is

src/test/java/org/embulk/output/snowflake/TestSnowflakeOutputPlugin.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -208,6 +208,7 @@ public void testConfigDefault() throws Exception {
208208
assertEquals("", task.getUser());
209209
assertEquals("", task.getPassword());
210210
assertEquals("public", task.getSchema());
211+
assertEquals("", task.getRole());
211212
assertEquals(false, task.getDeleteStage());
212213
}
213214

0 commit comments

Comments
 (0)