Discourse/discourse
Discourse/discourse
2
0
Fork 0
mirror of https://github.com/discourse/discourse.git synced 2025-09-04 08:47:37 +08:00
Code Packages Activity Actions 11

Expose replica_postgresql_connection to ActiveRecord::Base.

Browse source
This commit is contained in:
Guo Xiang Tan 2017-11-23 09:52:42 +08:00
parent 82222e8d18
commit a509f466a0
2 changed files with 24 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -131,13 +131,7 @@ module ActiveRecord
if fallback_handler.master_down?
Discourse.enable_readonly_mode(Discourse::PG_READONLY_MODE_KEY)
fallback_handler.verify_master
connection = postgresql_connection(config.dup.merge(
host: config[:replica_host],
port: config[:replica_port]
))
verify_replica(connection)
connection = replica_postgresql_connection(config)
else
begin
connection = postgresql_connection(config)
@ -158,10 +152,21 @@ module ActiveRecord
connection
end
def replica_postgresql_connection(config)
config = config.dup.merge(
host: config[:replica_host],
port: config[:replica_port]
)
connection = postgresql_connection(config)
verify_replica(connection)
connection
end
private
def verify_replica(connection)
value = connection.raw_connection.exec("SELECT pg_is_in_recovery()").values[0][0]
value = connection.exec_query("SELECT pg_is_in_recovery()").rows[0][0]
raise "Replica database server is not in recovery mode." if !value
end
end