db-purger 0.1.3 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 47c73f3315ea20145d1aa9b443fa2b53082170c8dea927fb6635d1e921afe8eb
-  data.tar.gz: 0ce2fbd7a325b6399dfce5bcc3767a0e01fb1775984ce2e33350f72353228c62
+  metadata.gz: f85bace783d6c2c4bf0526a608326a9d15ee3dc0babd439a2dd86c22ab304343
+  data.tar.gz: fa1f889ecf967b73357d2764764e01a3ba7a52a9dea5737b7008e9c87edf953b
 SHA512:
-  metadata.gz: 2e6e74e4de36039bab64bf9dff3f5887f032ab7f3f045ecfcf18b1a62e8c891e28366ef9378f8f508512b5913ac3244ea3b2ce12404c60df41b491dbcdd482a6
-  data.tar.gz: b9480ec3c2c581161904cc569eba5ee3fb046385bacddc7b7be0311b698c316018085cf29c02b351a7fa0cfdec33488409da747a2395e200a70c73be803ec5e4
+  metadata.gz: cc7a13db98536483613543736ed8a0bd03dc5dd1076060a4b1735e7b5499634d1e80e955cd714ce67785dcf42b0841b47ebefbaecae9143b4366c7d27be80786
+  data.tar.gz: bf6d5fa8f36eaf89087056673d1e49bfa74d4141321c078c5a0b0414fe58b8d64f1409b80838a2d05a7ae163cf11f0cf345dd2d812020a8adb4a1e5020ed1202

data/lib/db-purger/dynamic_plan_builder.rb

@@ -0,0 +1,98 @@
+# frozen_string_literal: true
+
+module DBPurger
+  # DBPurger::DynamicPlanBuilder generates a purge plan based on the database relations
+  class DynamicPlanBuilder
+    INDENT = '  '
+
+    attr_reader :output
+
+    def initialize(database)
+      @database = database
+      @output = ''.dup
+      @indent_depth = 0
+      write('DBPurger::PlanBuilder.build do')
+      @indent_depth += 1
+    end
+
+    def build(base_table_name, field)
+      write("base_table(#{base_table_name.to_sym.inspect}, #{field.to_sym.inspect})")
+      line_break
+      model = find_model_for_table(base_table_name)
+      foreign_key = foreign_key_name(model)
+      add_parent_tables(base_table_name, foreign_key)
+      unless (child_models = find_child_models(model, foreign_key)).empty?
+        line_break
+        add_child_tables(child_models, foreign_key, 0)
+      end
+      @indent_depth -= 1
+      write('end')
+      @output
+    end
+
+    private
+
+    def find_model_for_table(base_table_name)
+      @database.models.detect { |m| m.table_name == base_table_name.to_s }
+    end
+
+    def write(str)
+      @output << (INDENT * @indent_depth) + str + "\n"
+    end
+
+    def line_break
+      @output << "\n"
+    end
+
+    def add_parent_tables(base_table_name, field)
+      @database.models.each do |model|
+        next if model.table_name == base_table_name.to_s
+        next unless column?(model, field)
+
+        foreign_key = foreign_key_name(model)
+        write_table('parent', model.table_name, field, find_child_models(model, foreign_key), foreign_key)
+      end
+    end
+
+    def add_child_tables(child_models, field, change_indent_by = 1)
+      @indent_depth += change_indent_by
+      child_models.each do |model|
+        add_child_table(model, field)
+      end
+      @indent_depth -= change_indent_by
+    end
+
+    def add_child_table(model, field)
+      foreign_key = foreign_key_name(model)
+      write_table('child', model.table_name, field, find_child_models(model, foreign_key), foreign_key)
+    end
+
+    def find_child_models(model, field)
+      model_has_many_associations(model).map(&:klass).select { |m| column?(m, field) }
+    end
+
+    def model_has_many_associations(model)
+      model.reflect_on_all_associations.select do |assoc|
+        assoc.is_a?(ActiveRecord::Reflection::HasManyReflection)
+      end
+    end
+
+    def foreign_key_name(model)
+      model.table_name.singularize + '_id'
+    end
+
+    def column?(model, field)
+      model.columns.detect { |c| c.name == field.to_s } != nil
+    end
+
+    def write_table(table_type, table_name, field, child_models, foreign_key)
+      if child_models.empty?
+        write("#{table_type}_table(#{table_name.to_sym.inspect}, #{field.to_sym.inspect})")
+      else
+        write("#{table_type}_table(#{table_name.to_sym.inspect}, #{field.to_sym.inspect}) do")
+        add_child_tables(child_models, foreign_key)
+        write('end')
+      end
+    end
+  end
+end

data/lib/db-purger.rb

@@ -3,6 +3,7 @@
 # DBPurger is a tool to delete data from tables based on a initial purge value
 module DBPurger
   autoload :Config, 'db-purger/config'
+  autoload :DynamicPlanBuilder, 'db-purger/dynamic_plan_builder'
   autoload :Metrics, 'db-purger/metrics'
   autoload :MetricSubscriber, 'db-purger/metric_subscriber'
   autoload :PurgeTable, 'db-purger/purge_table'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: db-purger
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.2.0
 platform: ruby
 authors:
 - Doug Youch
@@ -46,6 +46,7 @@ extra_rdoc_files: []
 files:
 - lib/db-purger.rb
 - lib/db-purger/config.rb
+- lib/db-purger/dynamic_plan_builder.rb
 - lib/db-purger/metric_subscriber.rb
 - lib/db-purger/metrics.rb
 - lib/db-purger/plan.rb