kiba-plus 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 6f1a1cdf8bd14192dc38eff8bf739b08291c89f2
-  data.tar.gz: ccdc4706b15ef09c52c991bb1a3043fbe86d72b6
+  metadata.gz: bd0ab521f8e00c7b382e0837b43ac63d46fa04d4
+  data.tar.gz: 6a0116b43920b28f6cd4c5f69f108c3d1c554006
 SHA512:
-  metadata.gz: b84053566ac09a919756801fd9edc193b06decdafec0d671eda0663a0a4c2ed658581e778c734c77c5afd56e72d0991297b6c76def3bcbab9f628be86d5d0994
-  data.tar.gz: 95036ece396cbee056b591279baeb05d52d255677d664a83078ffde6a815a2fe61012080bf2822466b2ae77da4069ea025c46cff3f94ccfea75b4c8645e04aed
+  metadata.gz: 67cc2a0d8b6c00951bacdfde6b9452a8057a5f7c274aa5e055ae1c0c759becd9bae7b816f380b96bb5483a4683881bd8956242d28792c3d5c28fa1418a690afc
+  data.tar.gz: f417fe27d5b199ef50f76e1a1441d7a1a44e41af3bfd92542b6fb4fe3e27dd423b75aa72026de781492924993b0ae867a2b3ff574f3d7235f4995cd465558bc0

data/lib/kiba/plus/destination/pg.rb

@@ -9,12 +9,11 @@ module Kiba::Plus::Destination
       @options.assert_valid_keys(
         :connect_url,
         :schema,
-        :prepare_name,
-        :prepare_sql,
+        :table_name,
         :columns
       )
       @conn = PG.connect(connect_url)
-      @conn.exec "SET search_path TO %s" % [ options.fetch(:schema) ] unless options.fetch(:schema).empty?
+      @conn.exec "SET search_path TO %s" % [ options[:schema] ] if options[:schema]
       init
     end
 
@@ -43,16 +42,22 @@ module Kiba::Plus::Destination
       options.fetch(:connect_url)
     end
 
+    def table_name
+      options.fetch(:table_name)
+    end
+
     def prepare_name
-      options.fetch(:prepare_name, self.class.to_s.downcase + "stmt")
+      options.fetch(:prepare_name, table_name + "_stmt")
     end
 
     def prepare_sql
-      options.fetch(:prepare_sql)
+      sql = <<-SQL
+        INSERT INTO #{table_name} (#{columns.join(', ') }) VALUES (#{columns.each_with_index.map { |_, i| "$#{i + 1}" }.join(', ')});
+      SQL
     end
 
     def columns
       options.fetch(:columns)
     end
   end
-end
+end

data/lib/kiba/plus/job.rb

@@ -18,7 +18,7 @@ module Kiba
           @client = Mysql2::Client.new(mysql2_connect_hash(connect_url))
         elsif url.scheme =~ /postgres/i
           @client = PG.connect(connect_url)
-          @client.exec "SET search_path TO %s" % [ options.fetch(:schema) ] unless options.fetch(:schema).empty?
+          @client.exec "SET search_path TO %s" % [ options[:schema] ] if options[:schema]
         else
           raise 'No Imp!'
         end
@@ -147,4 +147,4 @@ module Kiba
       end
     end
   end
-end
+end

data/lib/kiba/plus/source/pg.rb

@@ -15,7 +15,7 @@ module Kiba
           :query
         )
         @client = PG.connect(connect_url)
-        @client.exec "SET search_path TO %s" % [ options.fetch(:schema) ] unless options.fetch(:schema).empty?
+        @client.exec "SET search_path TO %s" % [ options[:schema] ] if options[:schema]
       end
 
       def each
@@ -37,4 +37,4 @@ module Kiba
       end
     end
   end
-end
+end

data/lib/kiba/plus/version.rb

@@ -1,5 +1,5 @@
 module Kiba
   module Plus
-    VERSION = "0.1.4"
+    VERSION = "0.1.5"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: kiba-plus
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Hooopo