Adapter-Specific Enhancements

Author: railspulse

lib/rails_pulse/adapters/delayed_job_plugin.rb

@@ -0,0 +1,25 @@
+module RailsPulse
+  module Adapters
+    class DelayedJobPlugin < Delayed::Plugin
+      callbacks do |lifecycle|
+        lifecycle.around(:perform) do |worker, job_data, &block|
+          next block.call(worker, job_data) unless RailsPulse.configuration.enabled
+          next block.call(worker, job_data) unless RailsPulse.configuration.track_jobs
+
+          job_wrapper = JobWrapper.new(
+            job_id: job_data.id.to_s,
+            class_name: job_data.payload_object.class.name,
+            queue_name: job_data.queue,
+            arguments: job_data.payload_object.args,
+            enqueued_at: job_data.created_at,
+            executions: job_data.attempts
+          )
+
+          RailsPulse::JobRunCollector.track(job_wrapper, adapter: "delayed_job") do
+            block.call(worker, job_data)
+          end
+        end
+      end
+    end
+  end
+end

lib/rails_pulse/adapters/sidekiq_middleware.rb

@@ -0,0 +1,41 @@
+module RailsPulse
+  module Adapters
+    class SidekiqMiddleware
+      def call(worker, job_data, queue)
+        return yield unless RailsPulse.configuration.enabled
+        return yield unless RailsPulse.configuration.track_jobs
+
+        # Create ActiveJob-like wrapper for tracking
+        job_wrapper = JobWrapper.new(
+          job_id: job_data["jid"],
+          class_name: worker.class.name,
+          queue_name: queue,
+          arguments: job_data["args"],
+          enqueued_at: Time.at(job_data["enqueued_at"] || Time.current.to_f),
+          executions: job_data["retry_count"] || 0
+        )
+
+        RailsPulse::JobRunCollector.track(job_wrapper, adapter: "sidekiq") do
+          yield
+        end
+      end
+    end
+
+    class JobWrapper
+      attr_reader :job_id, :queue_name, :arguments, :enqueued_at, :executions
+
+      def initialize(job_id:, class_name:, queue_name:, arguments:, enqueued_at:, executions:)
+        @job_id = job_id
+        @class_name = class_name
+        @queue_name = queue_name
+        @arguments = arguments
+        @enqueued_at = enqueued_at
+        @executions = executions
+      end
+
+      def class
+        OpenStruct.new(name: @class_name)
+      end
+    end
+  end
+end

lib/rails_pulse/engine.rb

@@ -62,6 +62,24 @@ class Engine < ::Rails::Engine
       end
     end
 
+    initializer "rails_pulse.configure_sidekiq", after: "rails_pulse.active_job" do
+      if defined?(Sidekiq) && RailsPulse.configuration.job_adapters.dig(:sidekiq, :enabled)
+        require "rails_pulse/adapters/sidekiq_middleware"
+        Sidekiq.configure_server do |config|
+          config.server_middleware do |chain|
+            chain.add RailsPulse::Adapters::SidekiqMiddleware
+          end
+        end
+      end
+    end
+
+    initializer "rails_pulse.configure_delayed_job", after: "rails_pulse.active_job" do
+      if defined?(Delayed::Job) && RailsPulse.configuration.job_adapters.dig(:delayed_job, :enabled)
+        require "rails_pulse/adapters/delayed_job_plugin"
+        Delayed::Worker.plugins << RailsPulse::Adapters::DelayedJobPlugin
+      end
+    end
+
     initializer "rails_pulse.database_configuration", before: "active_record.initialize_timezone" do
       # Ensure database configuration is applied early in the initialization process
       # This allows models to properly connect to configured databases