Extend configuration to support disable_during_db_migrate

README.md

@@ -69,6 +69,10 @@ If you do not want to rewrite default `#to_sql` method you may specify
 You can also disable log formatting by specifying `PpSql.add_rails_logger_formatting=false`
 in initializers.
 
+## Disable during `db:migrate`
+
+Formatting the sql messages can add significant overhead to the schema dump portion of the database migration process. This overhead can be avoided by setting `PpSql.disable_during_db_migrate = true`.
+
 ### Add to Application record
 
 I found usefull this trick:

lib/pp_sql.rb

@@ -6,10 +6,11 @@ module PpSql
   # if you do not want to rewrite AR native method #to_sql
   # you may switch this setting to false in initializer
   class << self
-    attr_accessor :rewrite_to_sql_method, :add_rails_logger_formatting
+    attr_accessor :rewrite_to_sql_method, :add_rails_logger_formatting, :disable_during_db_migrate
   end
   self.rewrite_to_sql_method = true
   self.add_rails_logger_formatting = true
+  self.disable_during_db_migrate = false
 
   module Formatter
     private
@@ -66,6 +67,11 @@ class Railtie < Rails::Railtie
           ActiveRecord::LogSubscriber.prepend LogSubscriberPrettyPrint
         end
       end
+
+      rake_tasks do
+        path = File.expand_path(__dir__)
+        Dir.glob("#{path}/pp_sql/tasks/**/*.rake").each { |f| load f }
+      end
     end
   end
 end

lib/pp_sql/tasks/db_hooks.rake

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+namespace :pp_sql do
+  desc 'Hook that is ran before rails db:migrate that will disable pp_sql when disable_during_db_migrate is set to true'
+
+  task :disable_during_db_migrate do
+    if PpSql.disable_during_db_migrate
+      PpSql.add_rails_logger_formatting = false
+      PpSql.rewrite_to_sql_method = false
+    end
+  end
+end
+
+Rake::Task['db:migrate'].enhance(['pp_sql:disable_during_db_migrate'])