db-purger 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5b7144cdb2cda2433339283a8ca98c35ace1015c30fce5642a4c064234f07e72
-  data.tar.gz: 8b199fe73457afb3dabf37a2e6746cc839d42514a0ff81b99e0bd0ab62a1d2b5
+  metadata.gz: 14ec62643b185d00f26b61c57db34008d36fc58f2b92bf2375ba8501d3dd065d
+  data.tar.gz: db80d215880b0b8d889a986ae6f75b524c0fe4297bee086f516a5f5a0fcc2da5
 SHA512:
-  metadata.gz: 3c56fb34ca9276ad3e3e258825f703631f80ab6a1711a8e703feb4fb2b2d5676672c221f2f86ca8c310712928eb23705d86cb7fbdca8e01e56e2e0b653e859a3
-  data.tar.gz: 7283b99020d4164eadba078b9560921339c4e87d467bad5b0870991f6ed00f42be464db212949e6f9e083cc3d8d79e8bedada03bb3450bf36c2d57cfd5776d64
+  metadata.gz: abeaae8ae9930615a84e9639de423cef35869abf5fe5fd0a47caef1682135fcf61fc9e2fc127531db2ee3c104e322d4c6a8ce00984f3efdb569934813c0ce26c
+  data.tar.gz: ecfcd9b6c3557a1ba9fa2933ec0a4dd57698e95b528bf613adff62121136ed2f5514e579bd56775b8e8886c75e6e6d93ce5e6e31ba34c95577c8b7c3dd04881f

data/lib/db-purger/config.rb

@@ -3,8 +3,11 @@
 module DBPurger
   # DBPurger::Config keeps track of global config options for the purge process
   class Config
+    DEFAULT_DATETIME_FORMAT = '%Y-%m-%d %H:%M:%S'
+
     attr_writer :explain,
-                :explain_file
+                :explain_file,
+                :datetime_format
 
     def explain?
       @explain == true
@@ -13,5 +16,9 @@ module DBPurger
     def explain_file
       (@explain_file || $stdout)
     end
+
+    def datetime_format
+      @datetime_format || DEFAULT_DATETIME_FORMAT
+    end
   end
 end

data/lib/db-purger/plan_builder.rb

@@ -65,6 +65,8 @@ module DBPurger
       table.foreign_key = options[:foreign_key]
       table.batch_size = options[:batch_size]
       table.conditions = options[:conditions]
+      table.mark_deleted_field = options[:mark_deleted_field]
+      table.mark_deleted_value = options[:mark_deleted_value]
       build_nested_plan(table, &block) if block
       table
     end

data/lib/db-purger/purge_table_helper.rb

@@ -54,9 +54,9 @@ module DBPurger
                                               num_records: num_records) do |payload|
         records_deleted =
           if ::DBPurger.config.explain?
-            delete_sql = scope.to_sql.sub(/SELECT .*?FROM/, 'DELETE FROM')
-            ::DBPurger.config.explain_file.puts(delete_sql)
-            scope.count
+            explain(scope)
+          elsif @table.mark_deleted_field
+            scope.update_all(@table.mark_deleted_field => @table.mark_deleted_value)
           else
             scope.delete_all
           end
@@ -66,6 +66,37 @@ module DBPurger
       end
     end
 
+    def explain(scope)
+      sql =
+        if @table.mark_deleted_field
+          explain_update_sql(scope)
+        else
+          scope.to_sql.sub(/SELECT .*?FROM/, 'DELETE FROM')
+        end
+      ::DBPurger.config.explain_file.puts(sql)
+      scope.count
+    end
+
+    def explain_update_sql(scope)
+      sql = scope.to_sql.dup
+      sql.sub!(/SELECT .*?FROM/, 'UPDATE')
+      sql.sub!('WHERE', "SET #{mark_deleted_field_quoted} = #{mark_deleted_value_quoted} WHERE")
+      sql
+    end
+
+    def mark_deleted_field_quoted
+      @mark_deleted_field_quoted ||= model.connection.quote_column_name(@table.mark_deleted_field)
+    end
+
+    def mark_deleted_value_quoted
+      @mark_deleted_value_quoted ||=
+        if @table.mark_deleted_value.is_a?(Time)
+          model.connection.quote(@table.mark_deleted_value.strftime(::DBPurger.config.datetime_format))
+        else
+          model.connection.quote(@table.mark_deleted_value)
+        end
+    end
+
     def delete_records(batch)
       if foreign_tables?
         delete_records_and_foreign_tables(batch)

data/lib/db-purger/table.rb

@@ -8,7 +8,10 @@ module DBPurger
     attr_accessor :foreign_key,
                   :batch_size,
                   :conditions,
-                  :search_proc
+                  :search_proc,
+                  :mark_deleted_field
+
+    attr_writer :mark_deleted_value
 
     attr_reader :name,
                 :field
@@ -40,5 +43,9 @@ module DBPurger
     def fields
       [@field] + foreign_keys
     end
+
+    def mark_deleted_value
+      @mark_deleted_value || 1
+    end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: db-purger
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Doug Youch
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-07-18 00:00:00.000000000 Z
+date: 2019-07-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: dynamic-active-model