-
Notifications
You must be signed in to change notification settings - Fork 21.6k
/
log_subscriber.rb
160 lines (132 loc) · 4.25 KB
/
log_subscriber.rb
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
# frozen_string_literal: true
module ActiveRecord
class LogSubscriber < ActiveSupport::LogSubscriber
IGNORE_PAYLOAD_NAMES = ["SCHEMA", "EXPLAIN"]
class_attribute :backtrace_cleaner, default: ActiveSupport::BacktraceCleaner.new
def self.runtime=(value)
ActiveRecord.deprecator.warn(<<-MSG.squish)
ActiveRecord::LogSubscriber.runtime= is deprecated and will be removed in Rails 7.2.
MSG
ActiveRecord::RuntimeRegistry.sql_runtime = value
end
def self.runtime
ActiveRecord.deprecator.warn(<<-MSG.squish)
ActiveRecord::LogSubscriber.runtime is deprecated and will be removed in Rails 7.2.
MSG
ActiveRecord::RuntimeRegistry.sql_runtime
end
def self.reset_runtime
ActiveRecord.deprecator.warn(<<-MSG.squish)
ActiveRecord::LogSubscriber.reset_runtime is deprecated and will be removed in Rails 7.2.
MSG
ActiveRecord::RuntimeRegistry.reset
end
def strict_loading_violation(event)
debug do
owner = event.payload[:owner]
reflection = event.payload[:reflection]
color(reflection.strict_loading_violation_message(owner), RED)
end
end
subscribe_log_level :strict_loading_violation, :debug
def sql(event)
payload = event.payload
return if IGNORE_PAYLOAD_NAMES.include?(payload[:name])
name = if payload[:async]
"ASYNC #{payload[:name]} (#{payload[:lock_wait].round(1)}ms) (db time #{event.duration.round(1)}ms)"
else
"#{payload[:name]} (#{event.duration.round(1)}ms)"
end
name = "CACHE #{name}" if payload[:cached]
sql = payload[:sql]
binds = nil
if payload[:binds]&.any?
casted_params = type_casted_binds(payload[:type_casted_binds])
binds = []
payload[:binds].each_with_index do |attr, i|
attribute_name = if attr.respond_to?(:name)
attr.name
elsif attr.respond_to?(:[]) && attr[i].respond_to?(:name)
attr[i].name
else
nil
end
filtered_params = filter(attribute_name, casted_params[i])
binds << render_bind(attr, filtered_params)
end
binds = binds.inspect
binds.prepend(" ")
end
name = colorize_payload_name(name, payload[:name])
sql = color(sql, sql_color(sql), bold: true) if colorize_logging
debug " #{name} #{sql}#{binds}"
end
subscribe_log_level :sql, :debug
private
def type_casted_binds(casted_binds)
casted_binds.respond_to?(:call) ? casted_binds.call : casted_binds
end
def render_bind(attr, value)
case attr
when ActiveModel::Attribute
if attr.type.binary? && attr.value
value = "<#{attr.value_for_database.to_s.bytesize} bytes of binary data>"
end
when Array
attr = attr.first
else
attr = nil
end
[attr&.name, value]
end
def colorize_payload_name(name, payload_name)
if payload_name.blank? || payload_name == "SQL" # SQL vs Model Load/Exists
color(name, MAGENTA, bold: true)
else
color(name, CYAN, bold: true)
end
end
def sql_color(sql)
case sql
when /\A\s*rollback/mi
RED
when /select .*for update/mi, /\A\s*lock/mi
WHITE
when /\A\s*select/i
BLUE
when /\A\s*insert/i
GREEN
when /\A\s*update/i
YELLOW
when /\A\s*delete/i
RED
when /transaction\s*\Z/i
CYAN
else
MAGENTA
end
end
def logger
ActiveRecord::Base.logger
end
def debug(progname = nil, &block)
return unless super
if ActiveRecord.verbose_query_logs
log_query_source
end
end
def log_query_source
source = extract_query_source_location(caller)
if source
logger.debug(" ↳ #{source}")
end
end
def extract_query_source_location(locations)
backtrace_cleaner.clean(locations.lazy).first
end
def filter(name, value)
ActiveRecord::Base.inspection_filter.filter_param(name, value)
end
end
end
ActiveRecord::LogSubscriber.attach_to :active_record