src/derails/active_job/instrumentation.rb
# frozen_string_literal: true

module ActiveJob
  class << self
    private
      def instrument_enqueue_all(queue_adapter, jobs)
        payload = { adapter: queue_adapter, jobs: jobs }
        ActiveSupport::Notifications.instrument("enqueue_all.active_job", payload) do
          result = yield payload
          payload[:enqueued_count] = result
          result
        end
      end
  end

  module Instrumentation # :nodoc:
    extend ActiveSupport::Concern

    included do
      around_enqueue do |_, block|
        scheduled_at ? instrument(:enqueue_at, &block) : instrument(:enqueue, &block)
      end
    end

    def perform_now
      instrument(:perform) { super }
    end

    def instrument(operation, payload = {}, &block) # :nodoc:
      payload[:job] = self
      payload[:adapter] = queue_adapter

      ActiveSupport::Notifications.instrument("#{operation}.active_job", payload) do |payload|
        value = block.call(payload) if block
        payload[:aborted] = @_halted_callback_hook_called if defined?(@_halted_callback_hook_called)
        @_halted_callback_hook_called = nil
        value
      end
    end

    private
      def _perform_job
        instrument(:perform_start)
        super
      end

      def halted_callback_hook(*)
        super
        @_halted_callback_hook_called = true
      end
  end
end
