2021-02-03 21:09:17 +00:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2022-03-02 15:16:07 +00:00
|
|
|
return if Feature.feature_flags_available? && Feature.enabled?(:prometheus_initializer_refactor, default_enabled: :yaml)
|
|
|
|
|
2019-07-02 07:36:33 +00:00
|
|
|
# Keep separate directories for separate processes
|
|
|
|
def prometheus_default_multiproc_dir
|
|
|
|
return unless Rails.env.development? || Rails.env.test?
|
|
|
|
|
2019-12-22 09:07:51 +00:00
|
|
|
if Gitlab::Runtime.sidekiq?
|
2019-07-02 07:36:33 +00:00
|
|
|
Rails.root.join('tmp/prometheus_multiproc_dir/sidekiq')
|
2019-12-22 09:07:51 +00:00
|
|
|
elsif Gitlab::Runtime.puma?
|
2019-07-02 07:36:33 +00:00
|
|
|
Rails.root.join('tmp/prometheus_multiproc_dir/puma')
|
|
|
|
else
|
|
|
|
Rails.root.join('tmp/prometheus_multiproc_dir')
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2021-06-18 15:10:16 +00:00
|
|
|
::Prometheus::Client.configure do |config|
|
2020-09-09 12:08:22 +00:00
|
|
|
config.logger = Gitlab::AppLogger
|
2017-07-19 08:54:39 +00:00
|
|
|
|
|
|
|
config.initial_mmap_file_size = 4 * 1024
|
|
|
|
|
2019-07-02 07:36:33 +00:00
|
|
|
config.multiprocess_files_dir = ENV['prometheus_multiproc_dir'] || prometheus_default_multiproc_dir
|
2017-08-29 13:45:19 +00:00
|
|
|
|
2021-06-18 15:10:16 +00:00
|
|
|
config.pid_provider = ::Prometheus::PidProvider.method(:worker_id)
|
2017-07-19 08:54:39 +00:00
|
|
|
end
|
2017-08-07 17:13:02 +00:00
|
|
|
|
2017-11-22 23:26:50 +00:00
|
|
|
Gitlab::Application.configure do |config|
|
|
|
|
# 0 should be Sentry to catch errors in this middleware
|
2021-06-08 21:10:05 +00:00
|
|
|
config.middleware.insert_after(Labkit::Middleware::Rack, Gitlab::Metrics::RequestsRackMiddleware)
|
2017-11-22 23:26:50 +00:00
|
|
|
end
|
|
|
|
|
2021-11-18 12:13:46 +00:00
|
|
|
if Gitlab::Runtime.sidekiq? && (!ENV['SIDEKIQ_WORKER_ID'] || ENV['SIDEKIQ_WORKER_ID'] == '0')
|
|
|
|
# The single worker outside of a sidekiq-cluster, or the first worker (sidekiq_0)
|
|
|
|
# in a cluster of processes, is responsible for serving health checks.
|
2021-11-23 15:11:19 +00:00
|
|
|
#
|
|
|
|
# Do not clean the metrics directory here - the supervisor script should
|
|
|
|
# have already taken care of that.
|
2021-11-18 12:13:46 +00:00
|
|
|
Sidekiq.configure_server do |config|
|
|
|
|
config.on(:startup) do
|
2021-11-23 15:11:19 +00:00
|
|
|
# In https://gitlab.com/gitlab-org/gitlab/-/issues/345804 we are looking to
|
|
|
|
# only serve health-checks from a worker process; for backwards compatibility
|
|
|
|
# we still go through the metrics exporter server, but start to configure it
|
|
|
|
# with the new settings keys.
|
|
|
|
exporter_settings = Settings.monitoring.sidekiq_health_checks
|
|
|
|
Gitlab::Metrics::Exporter::SidekiqExporter.instance(exporter_settings).start
|
2021-11-18 12:13:46 +00:00
|
|
|
end
|
2019-06-14 01:41:29 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2018-05-24 13:02:39 +00:00
|
|
|
if !Rails.env.test? && Gitlab::Metrics.prometheus_metrics_enabled?
|
2021-02-03 15:09:24 +00:00
|
|
|
# When running Puma in a Single mode, `on_master_start` and `on_worker_start` are the same.
|
|
|
|
# Thus, we order these events to run `reinitialize_on_pid_change` with `force: true` first.
|
|
|
|
Gitlab::Cluster::LifecycleEvents.on_master_start do
|
|
|
|
::Prometheus::Client.reinitialize_on_pid_change(force: true)
|
2020-05-20 00:08:20 +00:00
|
|
|
|
2021-05-20 12:10:26 +00:00
|
|
|
if Gitlab::Runtime.puma?
|
2021-02-03 15:09:24 +00:00
|
|
|
Gitlab::Metrics::Samplers::PumaSampler.instance.start
|
2020-09-09 09:08:40 +00:00
|
|
|
end
|
|
|
|
|
2021-07-07 00:07:23 +00:00
|
|
|
Gitlab::Metrics.gauge(:deployments, 'GitLab Version', {}, :max).set({ version: Gitlab::VERSION, revision: Gitlab.revision }, 1)
|
2021-02-03 15:09:24 +00:00
|
|
|
|
2022-03-02 15:16:07 +00:00
|
|
|
if Gitlab::Runtime.puma?
|
2021-02-03 15:09:24 +00:00
|
|
|
Gitlab::Metrics::RequestsRackMiddleware.initialize_metrics
|
2020-04-21 15:21:10 +00:00
|
|
|
end
|
2021-01-22 21:09:10 +00:00
|
|
|
|
|
|
|
Gitlab::Ci::Parsers.instrument!
|
2020-01-09 12:08:03 +00:00
|
|
|
rescue IOError => e
|
|
|
|
Gitlab::ErrorTracking.track_exception(e)
|
|
|
|
Gitlab::Metrics.error_detected!
|
2017-11-02 20:23:23 +00:00
|
|
|
end
|
2019-05-29 12:27:44 +00:00
|
|
|
|
2021-02-03 15:09:24 +00:00
|
|
|
Gitlab::Cluster::LifecycleEvents.on_worker_start do
|
2021-06-18 15:10:16 +00:00
|
|
|
defined?(::Prometheus::Client.reinitialize_on_pid_change) && ::Prometheus::Client.reinitialize_on_pid_change
|
2021-12-20 15:12:25 +00:00
|
|
|
logger = Gitlab::AppLogger
|
|
|
|
Gitlab::Metrics::Samplers::RubySampler.initialize_instance(logger: logger).start
|
|
|
|
Gitlab::Metrics::Samplers::DatabaseSampler.initialize_instance(logger: logger).start
|
|
|
|
Gitlab::Metrics::Samplers::ThreadsSampler.initialize_instance(logger: logger).start
|
2019-10-02 21:06:22 +00:00
|
|
|
|
2022-03-02 15:16:07 +00:00
|
|
|
if Gitlab::Runtime.puma?
|
2021-12-20 15:12:25 +00:00
|
|
|
Gitlab::Metrics::Samplers::ActionCableSampler.instance(logger: logger).start
|
2021-02-03 15:09:24 +00:00
|
|
|
end
|
2020-05-15 21:08:21 +00:00
|
|
|
|
2021-02-03 15:09:24 +00:00
|
|
|
if Gitlab.ee? && Gitlab::Runtime.sidekiq?
|
2021-12-20 15:12:25 +00:00
|
|
|
Gitlab::Metrics::Samplers::GlobalSearchSampler.instance(logger: logger).start
|
2020-10-01 09:09:54 +00:00
|
|
|
end
|
2021-01-22 21:09:10 +00:00
|
|
|
|
|
|
|
Gitlab::Ci::Parsers.instrument!
|
2020-01-09 12:08:03 +00:00
|
|
|
rescue IOError => e
|
|
|
|
Gitlab::ErrorTracking.track_exception(e)
|
|
|
|
Gitlab::Metrics.error_detected!
|
2019-05-29 12:27:44 +00:00
|
|
|
end
|
2018-10-20 18:00:19 +00:00
|
|
|
end
|
2019-10-04 06:06:05 +00:00
|
|
|
|
2022-03-02 15:16:07 +00:00
|
|
|
if Gitlab::Runtime.puma?
|
2019-10-04 06:06:05 +00:00
|
|
|
Gitlab::Cluster::LifecycleEvents.on_master_start do
|
|
|
|
Gitlab::Metrics::Exporter::WebExporter.instance.start
|
|
|
|
end
|
|
|
|
|
2019-10-30 09:27:58 +00:00
|
|
|
Gitlab::Cluster::LifecycleEvents.on_before_graceful_shutdown do
|
|
|
|
# We need to ensure that before we re-exec or shutdown server
|
2019-10-16 15:06:17 +00:00
|
|
|
# we do stop the exporter
|
|
|
|
Gitlab::Metrics::Exporter::WebExporter.instance.stop
|
|
|
|
end
|
|
|
|
|
2019-10-04 06:06:05 +00:00
|
|
|
Gitlab::Cluster::LifecycleEvents.on_before_master_restart do
|
|
|
|
# We need to ensure that before we re-exec server
|
|
|
|
# we do stop the exporter
|
2019-10-16 15:06:17 +00:00
|
|
|
#
|
|
|
|
# We do it again, for being extra safe,
|
|
|
|
# but it should not be needed
|
2019-10-04 06:06:05 +00:00
|
|
|
Gitlab::Metrics::Exporter::WebExporter.instance.stop
|
|
|
|
end
|
|
|
|
|
|
|
|
Gitlab::Cluster::LifecycleEvents.on_worker_start do
|
|
|
|
# The `#close_on_exec=` takes effect only on `execve`
|
|
|
|
# but this does not happen for Ruby fork
|
|
|
|
#
|
|
|
|
# This does stop server, as it is running on master.
|
|
|
|
Gitlab::Metrics::Exporter::WebExporter.instance.stop
|
|
|
|
end
|
|
|
|
end
|