Loading files and removing whole users, repos and orgs

.gitignore

@@ -5,3 +5,4 @@ dump/*
 !log/.keep
 !dump/.keep
 *.gem
+.byebug_history

README.md

@@ -1,6 +1,6 @@
 # README
 
-*travis-backup* is an application that helps with housekeeping and backup for Travis CI database v2.2 and with migration to v3.0 database. By default it removes requests and builds with their corresponding jobs and logs, as long as they are older than given threshold says (and backups them in files, if this option is active). Although it can be also run with special modes: `move_logs`, for moving logs from one database to another, and `remove_orphans`, for deleting all orphaned data.
+*travis-backup* is an application that helps with housekeeping and backup for Travis CI database v2.2 and with migration to v3.0 database. By default it removes requests and builds with their corresponding jobs and logs, as long as they are older than given threshold says (and backups them in files, if this option is active). Although it can be also run in special modes to perform other specific tasks.
 
 ### Installation and run
 
@@ -29,6 +29,9 @@ All arguments:
   --move_logs                # run in move logs mode - move all logs to database at destination_db_url URL
   --destination_db_url URL   # URL for moving logs to
   --remove_orphans           # run in remove orphans mode
+  --orphans_table            # name of the table we will remove orphans from (if not defined, all tables are considered)
+  --load_from_files          # loads files stored in files_location to the database
+  --id_gap                   # concerns file loading - the gap between the biggest id in database and the lowest one that will be set to loaded data (that's for data inserted by other users during the load being performed; equals 1000 by default)
 ```
 
 Or inside your app:
@@ -60,9 +63,13 @@ backup.run(repo_id: 1)
 
 Using `--move_logs` flag you can move all logs to database at `destination_db_url` URL (which is required in this case). When you run gem in this mode no files are created and no other tables are being touched.
 
-Using `--remove_orphans` flag you can remove all orphaned data from tables. When you run gem in this mode no files are created.
+Using `--remove_orphans` flag you can remove all orphaned data from the tables. You can pick a specific table using `--orphans_table` flag or, by leaving it undefined, let all tables to be processed in the removing orphans procedure. It can be combined with `--backup` flag in order to save removed data in files.
 
-Using `--dry_run` flag you can check which data would be removed by gem, but without removing them actually. Instead of that reports will be printed on standard output. This flag can be also combined with `--move_logs` or `--remove_orphans`.
+Using `--user_id`, `--org_id` or `--repo_id` flag without setting `--threshold` results in removing the specified user/organization/repository with all its dependencies. It can be combined with `--backup` flag in order to save removed data in files.
+
+Using `--load_from_files` flag you can restore dumped data from files located at path given by `--files_location`. The distance defined by `--id_gap` is going to be kept between biggest ids in the database and the lowest ones from the data loaded from files (and it equals 1000 by default).
+
+Using `--dry_run` flag you can check which data would be removed by gem, but without removing them actually. Instead of that reports will be printed on standard output. This flag can be also combined with special modes.
 
 ### Configuration options
 
@@ -80,9 +87,12 @@ backup:
   repo_id: 1                # run only for given repository
   move_logs: false          # run in move logs mode - move all logs to database at destination_db_url URL
   remove_orphans: false     # run in remove orphans mode
+  orphans_table: 'builds'   # name of the table we will remove orphans from (if not defined, all tables are considered)
+  load_from_files: false    # loads files stored in files_location to the database
+  id_gap: 1500              # concerns file loading - the gap between the biggest id in database and the lowest one that will be set to loaded data (that's for data inserted by other users during the load being performed; equals 1000 by default)
 ```
 
-You can also set these properties using env vars corresponding to them: `IF_BACKUP`, `BACKUP_DRY_RUN`, `BACKUP_LIMIT`, `BACKUP_THRESHOLD`, `BACKUP_FILES_LOCATION`, `BACKUP_USER_ID`, `BACKUP_ORG_ID`, `BACKUP_REPO_ID`, `BACKUP_MOVE_LOGS`, `BACKUP_REMOVE_ORPHANS`.
+You can also set these properties using env vars corresponding to them: `IF_BACKUP`, `BACKUP_DRY_RUN`, `BACKUP_LIMIT`, `BACKUP_THRESHOLD`, `BACKUP_FILES_LOCATION`, `BACKUP_USER_ID`, `BACKUP_ORG_ID`, `BACKUP_REPO_ID`, `BACKUP_MOVE_LOGS`, `BACKUP_REMOVE_ORPHANS`, `BACKUP_ORPHANS_TABLE`, `BACKUP_LOAD_FROM_FILES`, `BACKUP_ID_GAP`.
 
 You should also specify your database url. You can do this the standard way in `config/database.yml` file, setting the `database_url` hash argument while creating `Backup` instance or using the `DATABASE_URL` env var. Your database should be consistent with the Travis 2.2 database schema.
 

lib/backup/load_from_files.rb

@@ -0,0 +1,245 @@
+# frozen_string_literal: true
+
+require 'id_hash'
+
+class Backup
+  class LoadFromFiles
+    class JsonContent < String
+      def hash
+        @hash ||= JSON.parse(self).symbolize_keys
+      end
+    end
+
+    class DataFile
+      attr_accessor :content
+
+      def initialize(json_content)
+        @content = json_content
+      end
+
+      def table_name
+        @content.match(/"table_name":\s?"(\w+)"/)[1]
+      end
+
+      def table_name_sym
+        table_name.to_sym
+      end
+
+      def full_hash
+        @content.hash
+      end
+    end
+
+    class EntryFile < DataFile
+      def ids
+        @content.scan(/"id":\s?(\d+)/).flatten.map(&:to_i)
+      end
+
+      def lowest_id
+        ids.min
+      end
+
+      def entries
+        full_hash[:data]
+      end
+    end
+
+    class RelationshipFile < DataFile
+      def relationships
+        @relationships ||= full_hash[:nullified_relationships].map do |rel|
+          rel.symbolize_keys
+        end
+      end
+    end
+
+    def initialize(config, dry_run_reporter=nil)
+      @config = config
+      @dry_run_reporter = dry_run_reporter
+      @touched_models = []
+    end
+
+    def run
+      set_id_offsets
+      load_data_with_offsets
+      cancel_offset_for_foreign_data
+      set_id_sequences
+      load_nullified_relationships
+    end
+
+    private
+
+    def load_nullified_relationships
+      relationship_files.each do |file|
+        file.relationships.each do |rel|
+          offset = @id_offsets[file.table_name.to_sym]
+
+          ActiveRecord::Base.connection.execute(%{
+            update #{rel[:related_table]}
+            set #{rel[:foreign_key]} = #{rel[:parent_id].to_i + offset}
+            where id = #{rel[:related_id].to_i};
+          })
+        end
+      end
+    end
+
+    def set_id_sequences
+      @touched_models.each do |model|
+        value = model.last.id + 1
+        seq = model.table_name + '_id_seq'
+        set_sequence(seq, value)
+      end
+
+      set_shared_builds_tasks_seq
+    end
+
+    def set_shared_builds_tasks_seq
+      value = [Build.last&.id || -1, Job.last&.id || -1].max + 1
+
+      if value > 0
+        set_sequence("shared_builds_tasks_seq", value)
+      end
+    end
+
+    def set_sequence(seq, value)
+      ActiveRecord::Base.connection.execute("alter sequence #{seq} restart with #{value};")
+    end
+
+    def cancel_offset_for_foreign_data
+      @loaded_entries.each do |entry|
+        entry.class.reflect_on_all_associations.select { |a| a.macro == :belongs_to }.each do |association|
+          foreign_key = association.foreign_key.to_sym
+          if entry.send(association.name).nil? && entry.send(foreign_key).present?
+            entry_hash = entry.attributes.symbolize_keys
+            table_name = get_table_name(entry_hash, association)
+            offset = @id_offsets[table_name.to_sym]
+            next if offset.nil?
+
+            proper_id = entry.send(foreign_key) - offset
+            entry.update(foreign_key => proper_id)
+          end
+        end
+      end
+    end
+
+    def load_data_with_offsets
+      @repository_files = []
+
+      @loaded_entries = entry_files.map do |data_file|
+        model = Model.get_model_by_table_name(data_file.table_name)
+
+        if model == Repository
+          @repository_files << data_file
+          next
+        end
+
+        load_file(model, data_file)
+      end.flatten.compact
+
+      repository_entries = @repository_files.map do |data_file|
+        load_file(Repository, data_file)
+      end.flatten.compact
+
+      @loaded_entries.concat(repository_entries)
+    end
+
+    def load_file(model, data_file)
+      @touched_models << model
+
+      data_file.entries&.map do |entry_hash|
+        load_entry(model, entry_hash)
+      end
+    end
+
+    def load_entry(model, entry_hash)
+      entry_hash.symbolize_keys!
+      entry_hash[:id] += @id_offsets[model.table_name.to_sym]
+      add_offset_to_foreign_keys!(model, entry_hash)
+      model.create(entry_hash)
+    end
+
+    def add_offset_to_foreign_keys!(model, entry_hash)
+      model.reflect_on_all_associations.select { |a| a.macro == :belongs_to }.each do |association|
+        foreign_key_sym = association.foreign_key.to_sym
+        next unless entry_hash[foreign_key_sym]
+
+        table_name = get_table_name(entry_hash, association)
+        entry_hash[foreign_key_sym] += @id_offsets[table_name.to_sym] || 0
+      end
+    end
+
+    def get_table_name(entry_hash, association)
+      if association.polymorphic?
+        type_symbol = association.foreign_key.gsub(/_id$/, '_type').to_sym
+        class_name = entry_hash[type_symbol]
+      else
+        class_name = association.class_name
+      end
+
+      Model.get_model(class_name).table_name
+    end
+
+    def file_contents
+      @file_contents ||= Dir["#{@config.files_location}/**/*.json"].map do |file_path|
+        JsonContent.new(File.read(file_path))
+      end
+    end
+
+    def entry_files
+      @entry_files ||= file_contents.map do |content|
+        next unless content.hash[:data]
+
+        EntryFile.new(content)
+      end.compact
+    end
+
+    def relationship_files
+      @relationship_files ||= file_contents.map do |content|
+        next if content.hash[:data]
+
+        RelationshipFile.new(content)
+      end.compact
+    end
+
+    def find_lowest_ids_from_files
+      @lowest_ids_from_files = HashOfArrays.new
+
+      entry_files.each do |data_file|
+        table_name = data_file.table_name_sym
+        min_id = data_file.lowest_id
+        @lowest_ids_from_files.add(table_name, min_id) if min_id
+      end
+
+      @lowest_ids_from_files = @lowest_ids_from_files.map { |k, arr| [k, arr.min] }.to_h
+    end
+
+    def find_highest_ids_from_db
+      @highest_ids_from_db = {}
+
+      Model.subclasses.each do |model|
+        table_name = model.table_name.to_sym
+        @highest_ids_from_db[table_name] = model.order(:id).last&.id || 0
+      end
+    end
+
+    def set_id_offsets
+      find_lowest_ids_from_files
+      find_highest_ids_from_db
+
+      @id_offsets = @lowest_ids_from_files.map do |key, file_min|
+        db_max = @highest_ids_from_db[key]
+        offset = db_max - file_min + @config.id_gap
+        [key, offset]
+      end.to_h
+
+      make_offset_common_for_builds_and_jobs
+    end
+
+    def make_offset_common_for_builds_and_jobs
+      if @id_offsets[:builds] && @id_offsets[:jobs]
+        offset = [@id_offsets[:builds], @id_offsets[:jobs]].max
+        @id_offsets[:builds] = offset
+        @id_offsets[:jobs] = offset
+      end
+    end
+  end
+end