lostapathy/db_subsetter

module DbSubsetter
  module Dialect
    # Dialect to subset to/from database without explicit support
    class Generic
      INSERT_BATCH_SIZE = 500

      def self.import
        ActiveRecord::Base.connection.disable_referential_integrity do
          yield
        end
      end

      def self.integrity_problems
        raise NotImplementedError, 'integrity_problems not implemented for this dialect'
      end

      def self.truncate_table(table)
        ActiveRecord::Base.connection.truncate(table)
      rescue NotImplementedError
        table = ActiveRecord::Base.connection.quote_table_name(table)
        ActiveRecord::Base.connection.execute("DELETE FROM #{table}")
      end
    end
  end
end