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
|
# frozen_string_literal: true
module ActiveRecord
# This is a thread locals registry for Active Record. For example:
#
# ActiveRecord::RuntimeRegistry.sql_runtime
#
# returns the connection handler local to the current unit of execution (either thread of fiber).
module RuntimeRegistry # :nodoc:
extend self
def sql_runtime
ActiveSupport::IsolatedExecutionState[:active_record_sql_runtime] ||= 0.0
end
def sql_runtime=(runtime)
ActiveSupport::IsolatedExecutionState[:active_record_sql_runtime] = runtime
end
def async_sql_runtime
ActiveSupport::IsolatedExecutionState[:active_record_async_sql_runtime] ||= 0.0
end
def async_sql_runtime=(runtime)
ActiveSupport::IsolatedExecutionState[:active_record_async_sql_runtime] = runtime
end
def queries_count
ActiveSupport::IsolatedExecutionState[:active_record_queries_count] ||= 0
end
def queries_count=(count)
ActiveSupport::IsolatedExecutionState[:active_record_queries_count] = count
end
def cached_queries_count
ActiveSupport::IsolatedExecutionState[:active_record_cached_queries_count] ||= 0
end
def cached_queries_count=(count)
ActiveSupport::IsolatedExecutionState[:active_record_cached_queries_count] = count
end
def reset
reset_runtimes
reset_queries_count
reset_cached_queries_count
end
def reset_runtimes
rt, self.sql_runtime = sql_runtime, 0.0
self.async_sql_runtime = 0.0
rt
end
def reset_queries_count
qc = queries_count
self.queries_count = 0
qc
end
def reset_cached_queries_count
qc = cached_queries_count
self.cached_queries_count = 0
qc
end
end
end
ActiveSupport::Notifications.monotonic_subscribe("sql.active_record") do |name, start, finish, id, payload|
unless ["SCHEMA", "TRANSACTION"].include?(payload[:name])
ActiveRecord::RuntimeRegistry.queries_count += 1
ActiveRecord::RuntimeRegistry.cached_queries_count += 1 if payload[:cached]
end
runtime = (finish - start) * 1_000.0
if payload[:async]
ActiveRecord::RuntimeRegistry.async_sql_runtime += (runtime - payload[:lock_wait])
end
ActiveRecord::RuntimeRegistry.sql_runtime += runtime
end
|
