2014-05-20 13:09:45 -04:00
|
|
|
require 'active_support/core_ext/string/filters'
|
2014-08-15 05:50:45 -04:00
|
|
|
require 'active_support/tagged_logging'
|
|
|
|
require 'active_support/logger'
|
2014-05-20 10:02:36 -04:00
|
|
|
|
2014-05-20 07:41:14 -04:00
|
|
|
module ActiveJob
|
|
|
|
module Logging
|
2014-05-22 13:33:23 -04:00
|
|
|
extend ActiveSupport::Concern
|
2014-05-22 16:38:01 -04:00
|
|
|
|
2014-05-22 13:33:23 -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
|
2014-08-25 10:34:50 -04:00
|
|
|
payload = {adapter: job.class.queue_adapter, job: job}
|
2014-05-22 16:38:01 -04:00
|
|
|
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
|
|
|
|
|
2014-05-22 13:33:23 -04:00
|
|
|
before_enqueue do |job|
|
2014-08-25 10:34:50 -04:00
|
|
|
if job.scheduled_at
|
2014-05-22 16:38:01 -04:00
|
|
|
ActiveSupport::Notifications.instrument "enqueue_at.active_job",
|
2014-08-25 10:34:50 -04:00
|
|
|
adapter: job.class.queue_adapter, job: job
|
2014-05-22 13:33:23 -04:00
|
|
|
else
|
|
|
|
ActiveSupport::Notifications.instrument "enqueue.active_job",
|
2014-08-25 10:34:50 -04:00
|
|
|
adapter: job.class.queue_adapter, job: job
|
2014-05-22 13:33:23 -04:00
|
|
|
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-09-17 15:46:53 -04:00
|
|
|
class LogSubscriber < ActiveSupport::LogSubscriber #:nodoc:
|
2014-05-20 13:09:45 -04:00
|
|
|
def enqueue(event)
|
2014-09-04 22:33:28 -04:00
|
|
|
info do
|
|
|
|
job = event.payload[:job]
|
|
|
|
"Enqueued #{job.class.name} (Job ID: #{job.job_id}) to #{queue_name(event)}" + args_info(job)
|
|
|
|
end
|
2014-05-20 13:09:45 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def enqueue_at(event)
|
2014-09-04 22:33:28 -04:00
|
|
|
info do
|
|
|
|
job = event.payload[:job]
|
|
|
|
"Enqueued #{job.class.name} (Job ID: #{job.job_id}) to #{queue_name(event)} at #{scheduled_at(event)}" + args_info(job)
|
|
|
|
end
|
2014-05-20 18:26:19 -04:00
|
|
|
end
|
2014-05-22 07:19:48 -04:00
|
|
|
|
2014-05-22 16:38:01 -04:00
|
|
|
def perform_start(event)
|
2014-09-04 22:33:28 -04:00
|
|
|
info do
|
|
|
|
job = event.payload[:job]
|
|
|
|
"Performing #{job.class.name} from #{queue_name(event)}" + args_info(job)
|
|
|
|
end
|
2014-05-20 13:09:45 -04:00
|
|
|
end
|
|
|
|
|
2014-05-22 16:38:01 -04:00
|
|
|
def perform(event)
|
2014-09-04 22:33:28 -04:00
|
|
|
info do
|
|
|
|
job = event.payload[:job]
|
|
|
|
"Performed #{job.class.name} from #{queue_name(event)} in #{event.duration.round(2).to_s}ms"
|
|
|
|
end
|
2014-05-22 16:38:01 -04:00
|
|
|
end
|
2014-05-22 13:33:23 -04:00
|
|
|
|
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
|
|
|
|
|
2014-08-25 10:34:50 -04:00
|
|
|
def args_info(job)
|
|
|
|
job.arguments.any? ? " with arguments: #{job.arguments.map(&:inspect).join(", ")}" : ""
|
2014-05-20 13:09:45 -04:00
|
|
|
end
|
2014-05-22 07:19:48 -04:00
|
|
|
|
2014-08-25 10:34:50 -04:00
|
|
|
def scheduled_at(event)
|
|
|
|
Time.at(event.payload[:job].scheduled_at).utc
|
2014-05-20 14:29:29 -04:00
|
|
|
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
|