codeninjasllc/discourse
1
0
Fork 0
mirror of https://github.com/codeninjasllc/discourse.git synced 2024-11-27 17:46:05 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Internally AR::Base.establish_connection removes the current connection.

Browse source
This commit is contained in:
Guo Xiang Tan 2016-02-04 11:06:02 +08:00
parent a08496bb1a
commit c532d7d1ae
2 changed files with 19 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
lib/active_record/connection_adapters/postgresql_fallback_adapter.rb
View file

@ -24,7 +24,6 @@ class PostgreSQLFallbackHandler
if connection.active?
logger.info "#{self.class}: Master server is active. Reconnecting..."
ActiveRecord::Base.remove_connection
ActiveRecord::Base.establish_connection(config)
Discourse.disable_readonly_mode
@master = true

43
spec/components/active_record/connection_adapters/postgresql_fallback_adapter_spec.rb
View file

@ -27,40 +27,35 @@ describe ActiveRecord::ConnectionHandling do
end
it 'should failover to a replica server' do
begin
ActiveRecord::Base.expects(:postgresql_connection).with(config).raises(PG::ConnectionBad)
ActiveRecord::Base.expects(:verify_replica).with(@replica_connection)
ActiveRecord::Base.expects(:postgresql_connection).with(config).raises(PG::ConnectionBad)
ActiveRecord::Base.expects(:verify_replica).with(@replica_connection)
ActiveRecord::Base.expects(:postgresql_connection).with(config.merge({
host: "localhost", port: "6432"
})).returns(@replica_connection)
ActiveRecord::Base.expects(:postgresql_connection).with(config.merge({
host: "localhost", port: "6432"
})).returns(@replica_connection)
expect { ActiveRecord::Base.postgresql_fallback_connection(config) }
.to raise_error(PG::ConnectionBad)
expect { ActiveRecord::Base.postgresql_fallback_connection(config) }
.to raise_error(PG::ConnectionBad)
expect{ ActiveRecord::Base.postgresql_fallback_connection(config) }
.to change{ Discourse.readonly_mode? }.from(false).to(true)
expect{ ActiveRecord::Base.postgresql_fallback_connection(config) }
.to change{ Discourse.readonly_mode? }.from(false).to(true)
ActiveRecord::Base.unstub(:postgresql_connection)
ActiveRecord::Base.unstub(:postgresql_connection)
current_threads = Thread.list
current_threads = Thread.list
expect{ ActiveRecord::Base.connection_pool.checkout }
.to change{ Thread.list.size }.by(1)
expect{ ActiveRecord::Base.connection_pool.checkout }
.to change{ Thread.list.size }.by(1)
# Wait for the thread to finish execution
threads = (Thread.list - current_threads).each(&:join)
# Wait for the thread to finish execution
threads = (Thread.list - current_threads).each(&:join)
expect(Discourse.readonly_mode?).to eq(false)
expect(Discourse.readonly_mode?).to eq(false)
expect(ActiveRecord::Base.connection_pool.connections.count).to eq(0)
expect(ActiveRecord::Base.connection_pool.connections.count).to eq(0)
expect(ActiveRecord::Base.connection)
.to be_an_instance_of(ActiveRecord::ConnectionAdapters::PostgreSQLAdapter)
ensure
# threads.each { |t| Thread.kill(t) } if threads
ActiveRecord::Base.establish_connection(:test)
end
expect(ActiveRecord::Base.connection)
.to be_an_instance_of(ActiveRecord::ConnectionAdapters::PostgreSQLAdapter)
end
end