mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
efa64bb295
Use inheritance to keep the behavior in the right modules. The order of Instrumentation and Logging had to change to be flipped to keep the current behavior.
44 lines
1 KiB
Ruby
44 lines
1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module ActiveJob
|
|
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)
|
|
enhanced_block = ->(event_payload) do
|
|
value = block.call if block
|
|
|
|
if defined?(@_halted_callback_hook_called) && @_halted_callback_hook_called
|
|
event_payload[:aborted] = true
|
|
@_halted_callback_hook_called = nil
|
|
end
|
|
|
|
value
|
|
end
|
|
|
|
ActiveSupport::Notifications.instrument \
|
|
"#{operation}.active_job", payload.merge(adapter: queue_adapter, job: self), &enhanced_block
|
|
end
|
|
|
|
def halted_callback_hook(*)
|
|
super
|
|
@_halted_callback_hook_called = true
|
|
end
|
|
end
|
|
end
|