aboutsummaryrefslogtreecommitdiffstats
path: root/lib/active_job/logging.rb
blob: f4a33ffe19ed2df5318ce0c041e4d8b112c3a686 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
require 'active_support/core_ext/string/filters'

module ActiveJob
  module Logging
    extend ActiveSupport::Concern
    
    module ClassMethods
      mattr_accessor(:logger) { ActiveSupport::Logger.new(STDOUT) }
    end
    
    included do
      before_enqueue do |job|
        if job.enqueued_at
          ActiveSupport::Notifications.instrument "enqueue_at.active_job", 
            adapter: job.class.queue_adapter, job: job.class, args: job.arguments, timestamp: job.enqueued_at
        else
          ActiveSupport::Notifications.instrument "enqueue.active_job",
            adapter: job.class.queue_adapter, job: job.class, args: job.arguments
        end
      end
      
      before_perform do |job|
        ActiveSupport::Notifications.instrument "perform.active_job",
          adapter: job.class.queue_adapter, job: job.class, args: job.arguments
      end
    end


    class LogSubscriber < ActiveSupport::LogSubscriber
      def enqueue(event)
        info "Enqueued #{event.payload[:job].name} to #{queue_name(event)}" + args_info(event)
      end

      def enqueue_at(event)
        info "Enqueued #{event.payload[:job].name} to #{queue_name(event)} at #{enqueued_at(event)}" + args_info(event)
      end

      def perform(event)
        info "Performed #{event.payload[:job].name} from #{queue_name(event)}" + args_info(event)
      end


      private
        def queue_name(event)
          event.payload[:adapter].name.demodulize.remove('Adapter')
        end

        def args_info(event)
          event.payload[:args].any? ? ": #{event.payload[:args].inspect}" : ""
        end

        def enqueued_at(event)
          Time.at(event.payload[:timestamp]).utc
        end

        def logger
          ActiveJob::Base.logger
        end
    end
  end
end

ActiveJob::Logging::LogSubscriber.attach_to :active_job