/
connection_proxy.rb
84 lines (70 loc) · 2.37 KB
/
connection_proxy.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
module ActiveReload
class MasterDatabase < ActiveRecord::Base
self.abstract_class = true
establish_connection :master_database
end
class ConnectionProxy
def initialize(master, slave)
@slave = slave.connection
@master = master.connection
@current = @slave
end
attr_accessor :slave, :master
def self.setup!
setup_for ActiveReload::MasterDatabase
end
def self.setup_for(master, slave = nil)
slave ||= ActiveRecord::Base
slave.send :include, ActiveRecordConnectionMethods
ActiveRecord::Observer.send :include, ActiveReload::ObserverExtensions
ActiveRecord::Base.active_connections[slave.name] = new(master, slave)
end
def with_master
set_to_master!
yield
ensure
set_to_slave!
end
def set_to_master!
@current = @master
end
def set_to_slave!
@current = @slave
end
delegate :insert, :update, :delete, :create_table, :rename_table, :drop_table, :add_column, :remove_column,
:change_column, :change_column_default, :rename_column, :add_index, :remove_index, :initialize_schema_information,
:dump_schema_information, :execute, :to => :master
def transaction(start_db_transaction = true, &block)
with_master { @current.transaction(start_db_transaction, &block) }
end
def method_missing(method, *args, &block)
@current.send(method, *args, &block)
end
end
module ActiveRecordConnectionMethods
def self.included(base)
base.alias_method_chain :reload, :master
end
def reload_with_master(*args, &block)
connection.with_master { reload_without_master }
end
end
# extend observer to always use the master database
# observers only get triggered on writes, so shouldn't be a performance hit
# removes a race condition if you are using conditionals in the observer
module ObserverExtensions
def self.included(base)
base.alias_method_chain :update, :masterdb
end
# Send observed_method(object) if the method exists.
def update_with_masterdb(observed_method, object) #:nodoc:
if object.class.connection.respond_to?(:with_master)
object.class.connection.with_master do
update_without_masterdb(observed_method, object)
end
else
update_without_masterdb(observed_method, object)
end
end
end
end