Class: Agentlog::Notifications

Inherits:
Object
  • Object
show all
Defined in:
lib/agentlog/notifications.rb

Constant Summary collapse

SQL_NOISE = 
/\A(?:SCHEMA|TRANSACTION)\z/

Instance Method Summary collapse

Constructor Details

#initialize(configuration:, logger:) ⇒ Notifications

Returns a new instance of Notifications.



7
8
9
10
11
12
 
# File 'lib/agentlog/notifications.rb', line 7

def initialize(configuration:, logger:)
  @configuration = configuration
  @logger = logger
  @compactor = Compactor.new(configuration)
  @subscriptions = []
end

Instance Method Details

#enqueue(payload) ⇒ Object 



67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
 
# File 'lib/agentlog/notifications.rb', line 67

def enqueue(payload)
  return unless configuration.log_jobs

  context = current_context
  return unless context

  job = payload[:job]
  context.add_job(
    job: {
      class_name: job.class.name,
      queue_name: job.queue_name,
      arguments: @compactor.compact(job.arguments || [])
    }
  )
end

#install! ⇒ Object 



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
 
# File 'lib/agentlog/notifications.rb', line 14

def install!
  @subscriptions << ActiveSupport::Notifications.subscribe(
    "start_processing.action_controller"
  ) do |_name, start, _finish, _id, payload|
    start_processing(start, payload)
  end

  @subscriptions << ActiveSupport::Notifications.subscribe(
    "sql.active_record"
  ) do |_name, start, finish, _id, payload|
    sql(start, finish, payload)
  end

  @subscriptions << ActiveSupport::Notifications.subscribe(
    "enqueue.active_job"
  ) do |_name, _start, _finish, _id, payload|
    enqueue(payload)
  end

  @subscriptions << ActiveSupport::Notifications.subscribe(
    "process_action.action_controller"
  ) do |_name, start, finish, _id, payload|
    process_action(start, finish, payload)
  end
end

#process_action(start, finish, payload) ⇒ Object 



83
84
85
86
87
88
89
90
91
92
93
 
# File 'lib/agentlog/notifications.rb', line 83

def process_action(start, finish, payload)
  context = current_context || start_processing(start, payload)
  duration_ms = milliseconds(start, finish)

  emit_request_line(context, payload, duration_ms)
  emit_sql_lines(context)
  emit_job_lines(context)
  emit_exception_line(payload)
ensure
  Thread.current[:agentlog_context] = nil
end

#sql(start, finish, payload) ⇒ Object 



56
57
58
59
60
61
62
63
64
65
 
# File 'lib/agentlog/notifications.rb', line 56

def sql(start, finish, payload)
  return unless configuration.log_sql

  context = current_context
  return unless context
  return if payload[:cached]
  return if payload[:name].to_s.match?(SQL_NOISE)

  context.add_sql(name: payload[:name].to_s, duration_ms: milliseconds(start, finish))
end

#start_processing(started_at, payload) ⇒ Object 



45
46
47
48
49
50
51
52
53
54
 
# File 'lib/agentlog/notifications.rb', line 45

def start_processing(started_at, payload)
  Thread.current[:agentlog_context] = Context.new(
    path: payload[:path] || payload[:original_fullpath],
    method: payload[:method],
    controller: payload[:controller],
    action: payload[:action],
    format: payload[:format],
    params: filtered_params(payload[:params] || {})
  ).tap { |context| context.started_at = started_at }
end

#uninstall! ⇒ Object 



40
41
42
43
 
# File 'lib/agentlog/notifications.rb', line 40

def uninstall!
  @subscriptions.each { |subscription| ActiveSupport::Notifications.unsubscribe(subscription) }
  @subscriptions.clear
end