2018-11-17 00:37:17 +00:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2018-01-19 14:04:37 +00:00
|
|
|
# rubocop:disable Style/ClassVars
|
|
|
|
|
2018-01-12 16:55:53 +00:00
|
|
|
module Gitlab
|
|
|
|
module Metrics
|
2018-01-29 11:33:08 +00:00
|
|
|
module Methods
|
2018-01-12 16:55:53 +00:00
|
|
|
extend ActiveSupport::Concern
|
2018-01-19 14:04:37 +00:00
|
|
|
|
|
|
|
included do
|
2018-01-22 15:43:04 +00:00
|
|
|
@@_metric_provider_mutex ||= Mutex.new
|
2018-01-22 19:52:41 +00:00
|
|
|
@@_metrics_provider_cache = {}
|
2018-01-19 14:04:37 +00:00
|
|
|
end
|
2018-01-16 13:06:56 +00:00
|
|
|
|
2018-01-12 16:55:53 +00:00
|
|
|
class_methods do
|
2021-10-18 21:12:56 +00:00
|
|
|
def reload_metric!(name = nil)
|
|
|
|
if name.nil?
|
|
|
|
@@_metrics_provider_cache = {}
|
|
|
|
else
|
|
|
|
@@_metrics_provider_cache.delete(name)
|
|
|
|
end
|
2018-01-16 14:47:07 +00:00
|
|
|
end
|
|
|
|
|
2018-01-12 16:55:53 +00:00
|
|
|
private
|
|
|
|
|
2018-01-16 13:06:56 +00:00
|
|
|
def define_metric(type, name, opts = {}, &block)
|
2018-01-17 21:16:25 +00:00
|
|
|
if respond_to?(name)
|
2018-01-19 13:54:40 +00:00
|
|
|
raise ArgumentError, "method #{name} already exists"
|
2018-01-12 16:55:53 +00:00
|
|
|
end
|
|
|
|
|
2018-01-16 14:47:07 +00:00
|
|
|
define_singleton_method(name) do
|
2018-01-22 19:55:48 +00:00
|
|
|
# inlining fetch_metric method to avoid method call overhead when instrumenting hot spots
|
2018-01-22 19:52:41 +00:00
|
|
|
@@_metrics_provider_cache[name] || init_metric(type, name, opts, &block)
|
2018-01-16 13:06:56 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def fetch_metric(type, name, opts = {}, &block)
|
2018-01-22 19:52:41 +00:00
|
|
|
@@_metrics_provider_cache[name] || init_metric(type, name, opts, &block)
|
2018-01-19 13:54:40 +00:00
|
|
|
end
|
2018-01-16 13:06:56 +00:00
|
|
|
|
2018-01-19 13:54:40 +00:00
|
|
|
def init_metric(type, name, opts = {}, &block)
|
2020-06-30 15:08:48 +00:00
|
|
|
options = ::Gitlab::Metrics::Methods::MetricOptions.new(opts)
|
2018-01-16 13:06:56 +00:00
|
|
|
options.evaluate(&block)
|
|
|
|
|
2018-01-18 19:22:49 +00:00
|
|
|
if disabled_by_feature(options)
|
2021-02-10 03:09:31 +00:00
|
|
|
synchronized_cache_fill(name) { ::Gitlab::Metrics::NullMetric.instance }
|
2018-01-18 19:22:49 +00:00
|
|
|
else
|
|
|
|
synchronized_cache_fill(name) { build_metric!(type, name, options) }
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def synchronized_cache_fill(key)
|
2018-01-19 17:12:19 +00:00
|
|
|
@@_metric_provider_mutex.synchronize do
|
2018-01-22 19:52:41 +00:00
|
|
|
@@_metrics_provider_cache[key] ||= yield
|
2018-01-16 13:06:56 +00:00
|
|
|
end
|
2018-01-18 19:22:49 +00:00
|
|
|
end
|
2018-01-16 13:06:56 +00:00
|
|
|
|
2018-01-18 19:22:49 +00:00
|
|
|
def disabled_by_feature(options)
|
2022-05-11 00:08:02 +00:00
|
|
|
options.with_feature && !::Feature.enabled?(options.with_feature,
|
|
|
|
type: :undefined, default_enabled_if_undefined: false)
|
2018-01-16 13:06:56 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def build_metric!(type, name, options)
|
2018-01-12 16:55:53 +00:00
|
|
|
case type
|
2018-01-17 01:55:52 +00:00
|
|
|
when :gauge
|
2019-02-14 18:05:35 +00:00
|
|
|
::Gitlab::Metrics.gauge(name, options.docstring, options.base_labels, options.multiprocess_mode)
|
2018-01-17 01:55:52 +00:00
|
|
|
when :counter
|
2019-02-14 18:05:35 +00:00
|
|
|
::Gitlab::Metrics.counter(name, options.docstring, options.base_labels)
|
2018-01-17 01:55:52 +00:00
|
|
|
when :histogram
|
2019-02-14 18:05:35 +00:00
|
|
|
::Gitlab::Metrics.histogram(name, options.docstring, options.base_labels, options.buckets)
|
2018-01-17 01:55:52 +00:00
|
|
|
when :summary
|
|
|
|
raise NotImplementedError, "summary metrics are not currently supported"
|
|
|
|
else
|
|
|
|
raise ArgumentError, "uknown metric type #{type}"
|
2018-01-12 16:55:53 +00:00
|
|
|
end
|
2018-01-16 13:06:56 +00:00
|
|
|
end
|
2018-01-12 16:55:53 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|