kiba-plus 0.1.15 → 0.1.16

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0e1dce9dada36e8dfe05f8fb763290c09fefdddb
-  data.tar.gz: 5cebb5aac9d4e0c5dde7c1a8a75a821e8d47b2d5
+  metadata.gz: 257ad79c9bf88386c966076e00b7ef9ae70c3f9a
+  data.tar.gz: 0a355add33a1a458b09ff2a68f09f2a7412fbf3d
 SHA512:
-  metadata.gz: 47f0b2212d7d203048424a3d20bdf93b7673bf7fbfd88de8812413998b08464b5a157db45c81b5ed4163051e57e73ff97e2ba4cbac3ff0077f463718cd952c3f
-  data.tar.gz: 71e19684a0fd7893dff572d03afb3af1fe75af5379592b6f0e7d50873abd19b8f7abb127dc2d53a79ab8bdaf07228ac06445332a5adb10c3fd3b0c50e3884339
+  metadata.gz: 12449db4db53d485251dde8fd664e2062fc82149825636ba55c37e7f2c853184cbfae4c90f7336e1666205e1b03737d7e029b84586f26526c5f6bd5abb4c1a40
+  data.tar.gz: 1ad1409f452190020f91ee7c83aa76dec01ca82c991088054225a1a4e8bddb246258f2ea09228d9ff9a3ff4e3244935845bb1a21a085403eca468d294517635a

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

@@ -18,7 +18,8 @@ module Kiba
           :connect_url,
           :schema,
           :query,
-          :stream
+          :stream,
+          :process_bar
         )
         @client = PG.connect(connect_url)
         @client.exec "SET search_path TO %s" % [ options[:schema] ] if options[:schema]
@@ -26,7 +27,7 @@ module Kiba
 
       def each
         Kiba::Plus.logger.info query
-        if stream
+        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
@@ -34,12 +35,14 @@ module Kiba
             res = client.get_result or break
             res.check
             res.each do |row|
+              print_process_bar if process_bar?
               yield(row)
             end
           end
         else
           results = client.query(query)
           results.each do |row|
+            print_process_bar if process_bar?
             yield(row)
           end
         end
@@ -55,9 +58,21 @@ module Kiba
         options.fetch(:query)
       end
 
-      def stream
+      def stream?
         options.fetch(:stream, false)
       end
+
+      def process_bar?
+        options.fetch(:process_bar, true)
+      end
+
+      def print_process_bar
+        @num ||= 1
+        $stdout.print "." * @num
+        $stdout.print "\r"
+        $stdout.flush
+        @num = @num + 1
+      end
     end
   end
 end

data/lib/kiba/plus/version.rb

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

metadata

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