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

    private
      def _perform_job
        instrument(:perform_start)
        super
      end

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

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

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