fix: Allow customizing the rake task regex to avoid starting the repo…

…rter (#220)

judoscale-rails/lib/judoscale/rails/config.rb

@@ -3,11 +3,12 @@
 module Judoscale
   module Rails
     module Config
-      attr_accessor :start_reporter_after_initialize
+      attr_accessor :start_reporter_after_initialize, :rake_task_ignore_regex
 
       def reset
         super
         @start_reporter_after_initialize = true
+        @rake_task_ignore_regex = /assets:|db:/
       end
     end
 

judoscale-rails/lib/judoscale/rails/railtie.rb

@@ -38,7 +38,7 @@ def judoscale_config
       config.after_initialize do
         if in_rails_console_or_runner?
           logger.debug "No reporting since we're in a Rails console or runner process"
-        elsif in_rake_task?(/assets:|db:/)
+        elsif in_rake_task?(judoscale_config.rake_task_ignore_regex)
           logger.debug "No reporting since we're in a build process"
         elsif judoscale_config.start_reporter_after_initialize
           Reporter.start

judoscale-rails/test/config_test.rb

@@ -8,5 +8,9 @@ module Judoscale
     it "adds the start_reporter_after_initialize config option" do
       _(::Judoscale::Config.instance.start_reporter_after_initialize).must_equal true
     end
+
+    it "adds the rake_task_ignore_regex config option" do
+      _(::Judoscale::Config.instance.rake_task_ignore_regex).must_equal(/assets:|db:/)
+    end
   end
 end

sample-apps/rails-sample/config/initializers/judoscale.rb

@@ -3,5 +3,6 @@
 Judoscale.configure do |config|
   # Open https://judoscale-adapter-mock.requestcatcher.com in a browser to monitor requests
   config.api_base_url = ENV["JUDOSCALE_URL"] || "https://judoscale-adapter-mock.requestcatcher.com"
+  config.rake_task_ignore_regex = /assets:|db:|middleware/
   # config.start_reporter_after_initialize = false
 end