kiba-plus 0.1.14 → 0.1.15

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ccfeb1e0e0d0597a534850d23ce04f5cee5114ec
-  data.tar.gz: c40ac9e7aefeaea658074cc2000978476ad4f549
+  metadata.gz: 0e1dce9dada36e8dfe05f8fb763290c09fefdddb
+  data.tar.gz: 5cebb5aac9d4e0c5dde7c1a8a75a821e8d47b2d5
 SHA512:
-  metadata.gz: 5b5fc63b73b49e96f1a82bdd8be7b763c891787fb357d8fb37e3e98f4393a91ea8d65f23250cb26dca83fae05d6e79f3e3a0359044d1ab9e9e39b2a31728814f
-  data.tar.gz: 96f7bb6eb6c0273f24f0d71d04417d50f9e7ab548309503d56201c2a78c63b2117a6f42e9f79ea9a5c0b4ebd86385ba4989ea2e45015d1e65ce7f4241fc776cb
+  metadata.gz: 47f0b2212d7d203048424a3d20bdf93b7673bf7fbfd88de8812413998b08464b5a157db45c81b5ed4163051e57e73ff97e2ba4cbac3ff0077f463718cd952c3f
+  data.tar.gz: 71e19684a0fd7893dff572d03afb3af1fe75af5379592b6f0e7d50873abd19b8f7abb127dc2d53a79ab8bdaf07228ac06445332a5adb10c3fd3b0c50e3884339

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

@@ -17,7 +17,8 @@ module Kiba
         @options.assert_valid_keys(
           :connect_url,
           :schema,
-          :query
+          :query,
+          :stream
         )
         @client = PG.connect(connect_url)
         @client.exec "SET search_path TO %s" % [ options[:schema] ] if options[:schema]
@@ -25,9 +26,22 @@ module Kiba
 
       def each
         Kiba::Plus.logger.info query
-        results = client.query(query)
-        results.each do |row|
-          yield(row)
+        if stream
+          # http://www.rubydoc.info/github/ged/ruby-pg/PG%2FConnection%3Aset_single_row_mode
+          client.send_query(query)
+          client.set_single_row_mode
+          loop do
+            res = client.get_result or break
+            res.check
+            res.each do |row|
+              yield(row)
+            end
+          end
+        else
+          results = client.query(query)
+          results.each do |row|
+            yield(row)
+          end
         end
       end
 
@@ -40,6 +54,10 @@ module Kiba
       def query
         options.fetch(:query)
       end
+
+      def stream
+        options.fetch(:stream, false)
+      end
     end
   end
 end

data/lib/kiba/plus/version.rb

@@ -1,5 +1,5 @@
 module Kiba
   module Plus
-    VERSION = "0.1.14"
+    VERSION = "0.1.15"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: kiba-plus
 version: !ruby/object:Gem::Version
-  version: 0.1.14
+  version: 0.1.15
 platform: ruby
 authors:
 - Hooopo
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-08-21 00:00:00.000000000 Z
+date: 2017-08-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: kiba