diff --git a/lib/multidb/balancer.rb b/lib/multidb/balancer.rb index 75801fe..08d25e3 100644 --- a/lib/multidb/balancer.rb +++ b/lib/multidb/balancer.rb @@ -20,6 +20,7 @@ def connection class Balancer def initialize(configuration) + @default_candidate = Candidate.new(@configuration.default_adapter) @candidates = {}.with_indifferent_access @configuration = configuration @configuration.raw_configuration[:databases].each_pair do |name, config| @@ -31,7 +32,7 @@ def initialize(configuration) end end unless @candidates.include?(:default) - @candidates[:default] = [Candidate.new(@configuration.default_adapter)] + @candidates[:default] = [@default_candidate] end end @@ -64,7 +65,7 @@ def use(name, &block) end def current_connection - Thread.current[:multidb_connection] ||= ActiveRecord::Base.connection_pool.connection + Thread.current[:multidb_connection] ||= @default_candidate.connection end class << self