2018-10-20 18:00:19 +00:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
|
|
|
# Don't handle sidekiq configuration as it
|
|
|
|
# has its own special active record configuration here
|
2019-12-22 09:07:51 +00:00
|
|
|
if defined?(ActiveRecord::Base) && !Gitlab::Runtime.sidekiq?
|
2018-10-20 18:00:19 +00:00
|
|
|
Gitlab::Cluster::LifecycleEvents.on_worker_start do
|
|
|
|
ActiveSupport.on_load(:active_record) do
|
2022-05-12 03:08:58 +00:00
|
|
|
# rubocop:disable Database/MultipleDatabases
|
|
|
|
ActiveRecord::Base.establish_connection # rubocop:disable Database/EstablishConnection
|
|
|
|
# rubocop:enable Database/MultipleDatabases
|
2018-10-20 18:00:19 +00:00
|
|
|
|
2020-09-09 12:08:22 +00:00
|
|
|
Gitlab::AppLogger.debug("ActiveRecord connection established")
|
2018-10-20 18:00:19 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
if defined?(ActiveRecord::Base)
|
|
|
|
Gitlab::Cluster::LifecycleEvents.on_before_fork do
|
2021-11-15 15:10:57 +00:00
|
|
|
raise 'ActiveRecord connection not established. Unable to start.' unless ApplicationRecord.database.exists?
|
2020-02-20 18:08:51 +00:00
|
|
|
|
2018-10-20 18:00:19 +00:00
|
|
|
# the following is highly recommended for Rails + "preload_app true"
|
|
|
|
# as there's no need for the master process to hold a connection
|
2022-05-12 03:08:58 +00:00
|
|
|
ActiveRecord::Base.clear_all_connections! # rubocop:disable Database/MultipleDatabases
|
2018-10-20 18:00:19 +00:00
|
|
|
|
2022-05-12 03:08:58 +00:00
|
|
|
Gitlab::AppLogger.debug("ActiveRecord connections disconnected")
|
2018-10-20 18:00:19 +00:00
|
|
|
end
|
|
|
|
end
|