activerecord-prunable 0.4.3 → 0.4.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8dd903bc39672f786b325d8907bfba3b4ba1d47e
-  data.tar.gz: 6377d77ab1c31b4d8ee4c38cd104c8e091df35a9
+  metadata.gz: d40d467ab85540ecaf695bab2974816e41161e09
+  data.tar.gz: 70488ff303ea76a6bfc5c997523c76fef3d33a3c
 SHA512:
-  metadata.gz: 502ae14d858b2b7f0997b70e1a752d165a1e553d511772a3412cbece5ce18b0b2becbc40e0a44238e6955241fa85ca5c19527da7ba12270d355ee5be245d194f
-  data.tar.gz: 5776ff5903da67cbf0b55c2b36f6c5b4c8605e9aa5ba6570e5160541bb2322ea7d2487fb79650a4d0d006e9a36a53d3645efc826f97a0c88e00402a859a1ba70
+  metadata.gz: d4cddff27a65443463d0f8e0d7ed0ee089e3dfa7b6e0903fe894a15d1a6155593760451ef48c03cab8a233b78ba4ba062d118c9373194d444160a846bdfb2a17
+  data.tar.gz: 51e402061a8349eeb1206931b5bb309e8c1959bdd8081d52da730d14312ff1242a63a5c5d69ee4ab2ed1aba3629552f5abb9092328a93c3c0c5d43e7bef09407

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    activerecord-prunable (0.4.2)
+    activerecord-prunable (0.4.3)
       activerecord (>= 3.0, < 6.0)
       activesupport (>= 3.0, < 6.0)
 

data/activerecord-prunable.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |spec|
   spec.name          = "activerecord-prunable"
-  spec.version       = "0.4.3"
+  spec.version       = "0.4.4"
   spec.authors       = ["dr2m"]
   spec.email         = ["maletin@maletin.work"]
 

data/lib/active_record/prunable.rb

@@ -30,14 +30,20 @@ module ActiveRecord
         class_variable_set(:@@prune_method, method)
       end
 
-      def prune_created_after(duration)
+      def prune_created_after(duration, batch_removal: nil, remove_in_batches: nil)
         class_variable_set(:@@prune_created_after, duration)
+
+        batch_removal_arg = batch_removal || remove_in_batches
+        batch_removal(batch_removal_arg) if batch_removal_arg
       end
 
       alias prune_after prune_created_after
 
-      def prune_updated_after(duration)
+      def prune_updated_after(duration, batch_removal: nil, remove_in_batches: nil)
         class_variable_set(:@@prune_updated_after, duration)
+
+        batch_removal_arg = batch_removal || remove_in_batches
+        batch_removal(batch_removal_arg) if batch_removal_arg
       end
 
       def prune!(*params, prune_method: nil, current_time: nil, batch_size: nil, in_batches: false)
@@ -58,9 +64,11 @@ module ActiveRecord
         destroyed_records
       end
 
-      def batch_removal(batch_size = 1000)
+      def batch_removal(batch_size = nil)
+        batch_size = 1000 unless batch_size.is_a?(Integer)
         class_variable_set(:@@prunable_batch_size, batch_size)
       end
+      alias remove_in_batches batch_removal
 
       private
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: activerecord-prunable
 version: !ruby/object:Gem::Version
-  version: 0.4.3
+  version: 0.4.4
 platform: ruby
 authors:
 - dr2m