2019-07-09 01:06:24 +00:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2021-05-27 06:10:47 +00:00
|
|
|
if Gitlab::Database::LoadBalancing.enable?
|
|
|
|
Gitlab::Database.disable_prepared_statements
|
2019-07-08 19:46:34 +00:00
|
|
|
|
2021-05-27 06:10:47 +00:00
|
|
|
Gitlab::Application.configure do |config|
|
|
|
|
config.middleware.use(Gitlab::Database::LoadBalancing::RackMiddleware)
|
|
|
|
end
|
2019-07-08 19:46:34 +00:00
|
|
|
|
2021-05-27 06:10:47 +00:00
|
|
|
Gitlab::Database::LoadBalancing.configure_proxy
|
2019-07-08 19:46:34 +00:00
|
|
|
|
2021-05-27 06:10:47 +00:00
|
|
|
# This needs to be executed after fork of clustered processes
|
|
|
|
Gitlab::Cluster::LifecycleEvents.on_worker_start do
|
2021-06-08 12:10:09 +00:00
|
|
|
# For Host-based LB, we need to re-connect as Rails discards connections on fork
|
|
|
|
Gitlab::Database::LoadBalancing.configure_proxy
|
|
|
|
|
2021-05-27 06:10:47 +00:00
|
|
|
# Service discovery must be started after configuring the proxy, as service
|
|
|
|
# discovery depends on this.
|
|
|
|
Gitlab::Database::LoadBalancing.start_service_discovery
|
2019-07-08 19:46:34 +00:00
|
|
|
end
|
|
|
|
end
|