able 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 375a99d24d70203258f5d235ade3088182097302
-  data.tar.gz: afecad7c5c6e6b015825221d5a0a3ac56625d28f
+  metadata.gz: 06e95a0cad498159cb8aa84963ba3536943be99d
+  data.tar.gz: f37f1cd9bd4420fc9c193530774ef31e05a44785
 SHA512:
-  metadata.gz: ca3a53d44651d77710fd9bb45b43c172009d64bf47787c5952c089a156385cdb337fb9eff6dd8bae2fb62d4b858a891e0e2612beb7e00f409db583d52050afdc
-  data.tar.gz: 006295e17471daa7be685f12d5f404e012a30964a9b12313b01a06807e5cf59376b184b6c735aae69448d6cff6e371c57df0614ea252bec2eefa1b0ece0eaa59
+  metadata.gz: 226482a2b4644e5e694ab81956804e722e04d317223c91bc9d77741fa0779f3f13a74ffc9b9be5b099e25ac3e1350386449b9b47cd7c355f30a62e6330184577
+  data.tar.gz: 17b35e2c493948f739074113dd91c7b4132a558ec2240173cdb0acef0bb851c31175a21b8492dc66fa05223c2121297e0f798ae01026ead2fdd93380d5377439

data/lib/able/version.rb

@@ -1,3 +1,3 @@
 module Able
-  VERSION = '0.1.4'
+  VERSION = '0.1.5'
 end

data/lib/ext/kernel.rb

@@ -23,17 +23,20 @@ module Kernel
   end
 
   # Is good for rake tasks, migrations etc.
-  def each_with_progress(collection)
+  def each_with_progress(collection, sql_logging: false, batch: true)
     progress = 0
 
     all_count = collection.count
     puts_with_time "Found before: #{all_count} items."
     i = 0
 
-    collection.find_each(batch_size: 1000) do |item|
-      yield item
+    unless sql_logging
+      old_logger = ActiveRecord::Base.logger
+      ActiveRecord::Base.logger = nil
+    end
 
-      i = i + 1
+    iterator_proc = Proc.new do |item|
+      yield item, i
 
       current_progress = ((i * 100) / all_count.to_f).floor
       if current_progress > progress
@@ -41,6 +44,15 @@ module Kernel
         puts_with_time "Done #{progress}%. #{i} of #{all_count} records."
       end
 
+      i = i + 1
+    end
+
+    batch ?
+        collection.find_each(batch_size: 1000) { |item| iterator_proc.call(item) } :
+        collection.each { |item| iterator_proc.call(item) }
+
+    unless sql_logging
+      ActiveRecord::Base.logger = old_logger
     end
 
     puts_with_time "Left after: #{collection.count} items."

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: able
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - entity
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-08-18 00:00:00.000000000 Z
+date: 2016-09-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler