embulk-input-vertica 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 42c918e05a7a36c3d44401a4fcf2720612144b16
-  data.tar.gz: 4c2aa39ef52247884cd8a461ae4dd3c22e5207d2
+  metadata.gz: 859d1f2716cc5ff691ed1c040a0be20a00582c34
+  data.tar.gz: b3aee724019f0092802a73e783681f6a9d4f392c
 SHA512:
-  metadata.gz: e682ae3bdd721a275acbc348ba37d32b78b161f9ced536daf126bfd65c5910856fba1d1f841b88899143e9c6337dccfdd97432758d310e52162af324a1d052a4
-  data.tar.gz: 893067fb4e42f2e5b07c01b0a39f4325cb84ecbe91a866d06cdb728509a3b89612073a14fbd5067ebf4ecb9ebf72c307c3a9e18761427a032c3b4d7e80c5270c
+  metadata.gz: 5c4bd57db32969f155385b8fc88300297f87575bf44b049fcff832e8b96e30f5be7365938246349674c86e38b6caa456447a81d257f6bfe25aadf606a550bf31
+  data.tar.gz: 1d82ea31cc168a87283ca7e5db7243f8b6d0a5e02833781cc3323f7ac0ba34f1249a8e0655df14d8956a0f900209fbc957dbc934390e3670d8b5c9b385b24e52

data/.gitignore

@@ -10,3 +10,4 @@ build/
 .tags
 .ruby-version
 example2
+*.iml

data/CHANGELOG.md

@@ -1,3 +1,9 @@
+# 0.1.2
+
+Enhancements:
+
+* Add `resouce_pool` option
+
 # 0.1.1
 
 Changes:

data/README.md

@@ -18,6 +18,7 @@ Vertica input plugins for Embulk loads records from vertica.
 - **fetch_rows**: number of rows to fetch one time (used for java.sql.Statement#setFetchSize) (integer, default: 10000)
 - **connect_timeout**: timeout for establishment of a database connection. (integer (seconds), default: 300)
 - **socket_timeout**: timeout for socket read operations. 0 means no timeout. (integer (seconds), default: 1800)
+- **resource_pool**: set session resource pool (string, default: null)
 - **ssl**: ~~enables SSL. data will be encrypted but CA or certification will not be verified (boolean, default: false)~~ Not supported yet
 - **options**: extra JDBC properties (hash, default: {})
 - If you write SQL directly,

data/build.gradle

@@ -16,7 +16,7 @@ configurations {
     provided
 }
 
-version = "0.1.1"
+version = "0.1.2"
 
 sourceCompatibility = 1.7
 targetCompatibility = 1.7

data/settings.gradle

@@ -0,0 +1 @@
+rootProject.name = 'embulk-input-vertica'

data/src/main/java/org/embulk/input/VerticaInputPlugin.java

@@ -1,5 +1,9 @@
 package org.embulk.input;
 
+import com.google.common.base.Optional;
+
+import java.sql.PreparedStatement;
+import java.sql.Statement;
 import java.util.Properties;
 import java.sql.Connection;
 import java.sql.Driver;
@@ -7,14 +11,15 @@ import java.sql.SQLException;
 import org.embulk.config.Config;
 import org.embulk.config.ConfigDefault;
 import org.embulk.input.jdbc.AbstractJdbcInputPlugin;
-import org.embulk.input.jdbc.getter.ColumnGetterFactory;
 import org.embulk.input.vertica.VerticaInputConnection;
-import org.embulk.spi.PageBuilder;
-import org.joda.time.DateTimeZone;
+
+import org.embulk.spi.Exec;
+import org.slf4j.Logger;
 
 public class VerticaInputPlugin
         extends AbstractJdbcInputPlugin
 {
+    private final Logger logger = Exec.getLogger(VerticaInputPlugin.class);
     private static final Driver driver = new com.vertica.jdbc.Driver();
 
     public interface VerticaPluginTask
@@ -40,6 +45,10 @@ public class VerticaInputPlugin
         @Config("schema")
         @ConfigDefault("\"public\"")
         public String getSchema();
+
+        @Config("resource_pool")
+        @ConfigDefault("null")
+        public Optional<String> getResourcePool();
     }
 
     @Override
@@ -77,7 +86,16 @@ public class VerticaInputPlugin
         props.putAll(t.getOptions());
 
         Connection con = driver.connect(url, props);
+
         try {
+            if (t.getResourcePool().isPresent()) {
+                Statement stmt = con.createStatement();
+                String escapedResourcePool = t.getResourcePool().get().replaceAll("'", "''");
+                String sql = String.format("SET SESSION RESOURCE_POOL = '%s'", escapedResourcePool);
+                logger.info("SQL: " + sql);
+                stmt.execute(sql);
+            }
+
             VerticaInputConnection c = new VerticaInputConnection(con, t.getSchema());
             con = null;
             return c;

data/src/main/java/org/embulk/input/vertica/VerticaInputConnection.java

@@ -1,8 +1,6 @@
 package org.embulk.input.vertica;
 
 import java.sql.Connection;
-import java.sql.PreparedStatement;
-import java.sql.ResultSet;
 import java.sql.SQLException;
 import org.slf4j.Logger;
 import org.embulk.spi.Exec;

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: embulk-input-vertica
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Naotoshi Seo
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2016-03-29 00:00:00.000000000 Z
+date: 2016-06-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -60,11 +60,12 @@ files:
 - lib/embulk/input/vertica.rb
 - lib/vertica-jdbc-7.0.1-0.jar
 - lib/vertica-jdbc-7.2.1-0.jar
+- settings.gradle
 - src/main/java/org/embulk/input/VerticaInputPlugin.java
 - src/main/java/org/embulk/input/vertica/VerticaInputConnection.java
 - classpath/embulk-core-0.8.8.jar
 - classpath/embulk-input-jdbc-0.7.0.jar
-- classpath/embulk-input-vertica-0.1.1.jar
+- classpath/embulk-input-vertica-0.1.2.jar
 - classpath/joda-time-2.9.2.jar
 - classpath/jruby-complete-9.0.5.0.jar
 - classpath/msgpack-core-0.8.3.jar