activerecord-prunable 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 45b9f545fee642c67289799871a2d5da4054d84c
-  data.tar.gz: 5f1ee266dd3ede0271215b9fb0461d8c03d4ebbc
+  metadata.gz: 53737b5692d22d48a79f1b754a7541683e5e1226
+  data.tar.gz: 96d239d3b0809674c2af1095ed5674cfd7889df1
 SHA512:
-  metadata.gz: eb2e391664238c813f99c1741c43a060c5d9970f72b770079acceeba8bba911fe9ce2a4695cc6a82663a5e93307782c52ee618ea3354be9481b231bc6af7dfdc
-  data.tar.gz: b6d61ba7d11befb1d65bbda2e61be368ea638f3cd6c3ccc85d3302dd502567edd8a8b1764aec5c2d61ca1081f18276488487ea48fdd249977e76d930aaa5c729
+  metadata.gz: b1c9ad1d3c656a02e318910140a35765fb3b638340085f85cd3dcd0fd4005ea9327591d17d71d0cbdab4c8b95e6a2fe54825d651e373d7014bbd2b5866cad7bf
+  data.tar.gz: 1e49941cf2e63081ef6c2dff32d4207bf22666a084cf3918baac7fd4b983f0957d8beeda8b53cf028017a1336faa3758d87c7d0f2b23256ccf84d10e450b0f17

data/Gemfile.lock

@@ -1,21 +1,21 @@
 PATH
   remote: .
   specs:
-    activerecord-prunable (0.1.0)
+    activerecord-prunable (0.2.0)
       activerecord
       activesupport
 
 GEM
   remote: https://rubygems.org/
   specs:
-    activemodel (4.2.6)
-      activesupport (= 4.2.6)
+    activemodel (4.2.7)
+      activesupport (= 4.2.7)
       builder (~> 3.1)
-    activerecord (4.2.6)
-      activemodel (= 4.2.6)
-      activesupport (= 4.2.6)
+    activerecord (4.2.7)
+      activemodel (= 4.2.7)
+      activesupport (= 4.2.7)
       arel (~> 6.0)
-    activesupport (4.2.6)
+    activesupport (4.2.7)
       i18n (~> 0.7)
       json (~> 1.7, >= 1.7.7)
       minitest (~> 5.1)

data/README.md

@@ -24,6 +24,10 @@ gem "activerecord-prunable"
 
      scope :prunable, -> { where("created_at > ?", 1.month.ago) }
 
+     # You can also set type of removing records (:destroy or :delete).
+     # By default it's :destroy
+     prune_method :delete
+
    end
    ```
 
@@ -51,6 +55,12 @@ Pruning multiple models:
 Prunable.prune!(SomeModel, AnotherModel)
 ```
 
+Set default method of pruning (:destroy or :delete):
+
+```ruby
+Prunable.prune!(prune_method: :delete)
+```
+
 Getting an array of all models that include `ActiveRecord::Prunable`:
 
 ```ruby

data/activerecord-prunable.gemspec

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

data/lib/active_record/prunable.rb

@@ -18,28 +18,61 @@ module ActiveRecord
     end
 
     module ClassMethods
+      def prune_method(method)
+        unless [:destroy, :delete].include?(method)
+          logger.info "Incorrect prune method #{method} will be ignored"
+          return false
+        end
+
+        logger.info "Prune method #{method} was successfully setted"
+        class_variable_set(:@@prune_method, method)
+      end
+
       def prune!
         logger.info "Pruning old records of #{self}"
+        return false unless check_scope
+
+        destroyed = prune_by_method
 
+        if destroyed > 0
+          logger.info "#{destroyed} records have been pruned."
+        else
+          logger.info "Nothing to prune."
+        end
+
+        destroyed
+      end
+
+      private
+
+      def check_scope
         unless respond_to?(:prunable)
           logger.info "This model has no :prunable scope, nothing to prune."
-          return
+          return false
         end
 
         unless prunable.is_a?(::ActiveRecord::Relation)
           logger.info ":prunable is not a relation, nothing to prune."
-          return
+          return false
         end
 
-        destroyed = prunable.destroy_all
+        true
+      end
 
-        if destroyed.any?
-          logger.info "#{destroyed.size} records have been pruned."
+      def prune_by_method
+        unless class_variable_defined?(:@@prune_method)
+          prune_method = :destroy
         else
-          logger.info "Nothing to prune."
+          prune_method = class_variable_get(:@@prune_method)
         end
 
-        destroyed
+        logger.info "Prune method is #{prune_method}"
+
+        if prune_method == :delete
+          prunable.delete_all
+        else
+          prunable.destroy_all.size
+        end
       end
     end
   end

data/lib/activerecord-prunable.rb

@@ -9,9 +9,15 @@ module Prunable
       ActiveRecord::Prunable.includes
     end
 
-    def prune!(*models)
+    def prune!(*models, prune_method: nil)
       models = self.models if models.empty?
-      models.each(&:prune!)
+      models.each do |model|
+        if prune_method && !model.class_variable_defined?(:@@prune_method)
+          model.prune_method(prune_method)
+        end
+
+        model.prune!
+      end
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: activerecord-prunable
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - dr2m
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-07-20 00:00:00.000000000 Z
+date: 2016-08-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec