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
|
# frozen_string_literal: true
module TestProf
# Forces ActiveRecord to use the same connection between threads
module ActiveRecordSharedConnection # :nodoc: all
class << self
attr_reader :connection
def enable!
self.connection = ActiveRecord::Base.connection
end
def disable!
self.connection = nil
end
def ignore(&block)
raise ArgumentError, "Block is required" unless block_given?
@ignores ||= []
ignores << block
end
def ignored?(config)
!ignores.nil? && ignores.any? { |clbk| clbk.call(config) }
end
private
attr_reader :ignores
def connection=(conn)
@connection = conn
connection.singleton_class.prepend Connection
connection
end
end
module Connection
def shared_lock
@shared_lock ||= Mutex.new
end
def exec_cache(*)
shared_lock.synchronize { super }
end
def exec_no_cache(*)
shared_lock.synchronize { super }
end
def execute(*)
shared_lock.synchronize { super }
end
end
module Ext
def connection
return super if ActiveRecordSharedConnection.ignored?(connection_config)
ActiveRecordSharedConnection.connection || super
end
end
end
end
ActiveSupport.on_load(:active_record) do
if ::ActiveRecord::Base.connection.pool.respond_to?(:lock_thread=)
TestProf.log :warn, "You activated ActiveRecordSharedConnection patch for the Rails version,\n" \
"which has a built-in support for the same functionality.\n" \
"Consider removing it, 'cause this could result in unexpected behaviour.\n\n" \
"Read more in the docs: https://test-prof.evilmartians.io/#/active_record_shared_connection"
end
TestProf::ActiveRecordSharedConnection.enable!
ActiveRecord::Base.singleton_class.prepend TestProf::ActiveRecordSharedConnection::Ext
end
|
