1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
rails--rails/activejob/lib/active_job/logging.rb

91 lines
2.9 KiB
Ruby
Raw Normal View History

2014-05-20 13:09:45 -04:00
require 'active_support/core_ext/string/filters'
require 'active_support/tagged_logging'
require 'active_support/logger'
2014-05-20 07:41:14 -04:00
module ActiveJob
module Logging
extend ActiveSupport::Concern
2014-05-22 16:38:01 -04:00
included do
2014-05-22 15:11:21 -04:00
cattr_accessor(:logger) { ActiveSupport::TaggedLogging.new(ActiveSupport::Logger.new(STDOUT)) }
2014-08-12 14:24:19 -04:00
around_enqueue do |_, block, _|
2014-05-22 16:38:01 -04:00
tag_logger do
block.call
end
end
around_perform do |job, block, _|
tag_logger(job.class.name, job.job_id) do
payload = {adapter: job.class.queue_adapter, job: job.class, args: job.arguments}
ActiveSupport::Notifications.instrument("perform_start.active_job", payload.dup)
2014-08-12 14:24:19 -04:00
ActiveSupport::Notifications.instrument("perform.active_job", payload) do
2014-05-22 16:38:01 -04:00
block.call
end
end
end
before_enqueue do |job|
if job.enqueued_at
2014-05-22 16:38:01 -04:00
ActiveSupport::Notifications.instrument "enqueue_at.active_job",
2014-05-30 19:19:30 -04:00
adapter: job.class.queue_adapter, job: job.class, job_id: job.job_id, args: job.arguments, timestamp: job.enqueued_at
else
ActiveSupport::Notifications.instrument "enqueue.active_job",
2014-05-30 19:19:30 -04:00
adapter: job.class.queue_adapter, job: job.class, job_id: job.job_id, args: job.arguments
end
end
end
2014-05-22 16:38:01 -04:00
private
def tag_logger(*tags)
if logger.respond_to?(:tagged)
tags.unshift "ActiveJob" unless logger_tagged_by_active_job?
ActiveJob::Base.logger.tagged(*tags){ yield }
else
yield
end
end
def logger_tagged_by_active_job?
logger.formatter.current_tags.include?("ActiveJob")
end
2014-05-20 13:09:45 -04:00
class LogSubscriber < ActiveSupport::LogSubscriber
def enqueue(event)
info { "Enqueued #{event.payload[:job].name} (Job ID: #{event.payload[:job_id]}) to #{queue_name(event)}" + args_info(event) }
2014-05-20 13:09:45 -04:00
end
def enqueue_at(event)
info { "Enqueued #{event.payload[:job].name} (Job ID: #{event.payload[:job_id]}) to #{queue_name(event)} at #{enqueued_at(event)}" + args_info(event) }
end
2014-05-22 07:19:48 -04:00
2014-05-22 16:38:01 -04:00
def perform_start(event)
info { "Performing #{event.payload[:job].name} from #{queue_name(event)}" + args_info(event) }
2014-05-20 13:09:45 -04:00
end
2014-05-22 16:38:01 -04:00
def perform(event)
info { "Performed #{event.payload[:job].name} from #{queue_name(event)} in #{event.duration.round(2).to_s}ms" }
2014-05-22 16:38:01 -04:00
end
2014-05-20 13:09:45 -04:00
private
def queue_name(event)
2014-05-30 19:19:30 -04:00
event.payload[:adapter].name.demodulize.remove('Adapter') + "(#{event.payload[:job].queue_name})"
2014-05-20 13:09:45 -04:00
end
def args_info(event)
2014-05-22 16:38:01 -04:00
event.payload[:args].any? ? " with arguments: #{event.payload[:args].map(&:inspect).join(", ")}" : ""
2014-05-20 13:09:45 -04:00
end
2014-05-22 07:19:48 -04:00
def enqueued_at(event)
Time.at(event.payload[:timestamp]).utc
end
2014-05-20 13:09:45 -04:00
def logger
ActiveJob::Base.logger
end
end
2014-05-20 07:41:14 -04:00
end
end
2014-05-20 13:09:45 -04:00
ActiveJob::Logging::LogSubscriber.attach_to :active_job