Enable hot-reload via adding ActiveSupport::Reloader delegate

lib/shoryuken.rb

@@ -91,6 +91,10 @@ def self.healthy?
     :on,
     :cache_visibility_timeout?,
     :cache_visibility_timeout=,
+    :reloader,
+    :reloader=,
+    :enable_reloading,
+    :enable_reloading=,
     :delay
   )
 end

lib/shoryuken/environment_loader.rb

@@ -70,6 +70,13 @@ def initialize_rails
             ::Rails.application.config.eager_load = true
           end
         end
+        ::Rails::Application.initializer 'shoryuken.set_reloader_hook' do |app|
+          Shoryuken.reloader = proc do |&block|
+            app.reloader.wrap do
+              block.call
+            end
+          end
+        end
         if Shoryuken.active_job?
           require 'shoryuken/extensions/active_job_extensions'
           require 'shoryuken/extensions/active_job_adapter'

lib/shoryuken/options.rb

@@ -17,7 +17,7 @@ class Options
     }.freeze
 
     attr_accessor :active_job_queue_name_prefixing, :cache_visibility_timeout, :groups,
-                  :launcher_executor,
+                  :launcher_executor, :reloader, :enable_reloading,
                   :start_callback, :stop_callback, :worker_executor, :worker_registry, :exception_handlers
     attr_writer :default_worker_options, :sqs_client
     attr_reader :sqs_client_receive_message_opts
@@ -29,6 +29,8 @@ def initialize
       self.active_job_queue_name_prefixing = false
       self.worker_executor = Worker::DefaultExecutor
       self.cache_visibility_timeout = false
+      self.reloader = proc { |&block| block.call }
+      self.enable_reloading ||= false
       # this is needed for keeping backward compatibility
       @sqs_client_receive_message_opts ||= {}
     end

lib/shoryuken/processor.rb

@@ -14,12 +14,21 @@ def initialize(queue, sqs_msg)
     end
 
     def process
-      return logger.error { "No worker found for #{queue}" } unless worker
+      worker_perform = proc do
+        return logger.error { "No worker found for #{queue}" } unless worker
+        Shoryuken::Logging.with_context("#{worker_name(worker.class, sqs_msg, body)}/#{queue}/#{sqs_msg.message_id}") do
+          worker.class.server_middleware.invoke(worker, queue, sqs_msg, body) do
+            worker.perform(sqs_msg, body)
+          end
+        end
+      end
 
-      Shoryuken::Logging.with_context("#{worker_name(worker.class, sqs_msg, body)}/#{queue}/#{sqs_msg.message_id}") do
-        worker.class.server_middleware.invoke(worker, queue, sqs_msg, body) do
-          worker.perform(sqs_msg, body)
+      if Shoryuken.enable_reloading
+        Shoryuken.reloader.call do
+          worker_perform.call
         end
+      else
+        worker_perform.call
       end
     rescue Exception => ex
       Array(Shoryuken.exception_handlers).each { |handler| handler.call(ex, queue, sqs_msg) }

lib/shoryuken/version.rb

@@ -1,3 +1,3 @@
 module Shoryuken
-  VERSION = '6.1.2'.freeze
+  VERSION = '6.2.0'.freeze
 end

spec/shoryuken/processor_spec.rb

@@ -159,5 +159,43 @@ def perform(sqs_msg, body); end
         subject.process
       end
     end
+
+    context 'when specifying a reloader' do
+      before do
+        Shoryuken.reloader = proc do |_|
+          TestWorker.new.called
+        end
+      end
+
+      after do
+        Shoryuken.reloader = proc { |&block| block.call }
+      end
+
+      context 'when reloader is enabled' do
+        before do
+          Shoryuken.enable_reloading = true
+        end
+
+        it 'wraps execution in reloader' do
+          expect_any_instance_of(TestWorker).to receive(:called)
+          expect_any_instance_of(TestWorker).to_not receive(:perform)
+
+          subject.process
+        end
+
+        after do
+          Shoryuken.enable_reloading = false
+        end
+      end
+
+      context 'when reloader is disabled' do
+        it 'does not wrap execution in reloader' do
+          expect_any_instance_of(TestWorker).to_not receive(:called)
+          expect_any_instance_of(TestWorker).to receive(:perform).with(sqs_msg, sqs_msg.body)
+
+          subject.process
+        end
+      end
+    end
   end
 end