1af6cf28c0
This allows measuring of timings of arbitrary Ruby blocks, this allows for more fine grained performance monitoring. Custom values and tags can also be attached to a block.
111 lines
3.1 KiB
Ruby
111 lines
3.1 KiB
Ruby
module Gitlab
|
|
module Metrics
|
|
extend Gitlab::CurrentSettings
|
|
|
|
RAILS_ROOT = Rails.root.to_s
|
|
METRICS_ROOT = Rails.root.join('lib', 'gitlab', 'metrics').to_s
|
|
PATH_REGEX = /^#{RAILS_ROOT}\/?/
|
|
|
|
def self.settings
|
|
@settings ||= {
|
|
enabled: current_application_settings[:metrics_enabled],
|
|
pool_size: current_application_settings[:metrics_pool_size],
|
|
timeout: current_application_settings[:metrics_timeout],
|
|
method_call_threshold: current_application_settings[:metrics_method_call_threshold],
|
|
host: current_application_settings[:metrics_host],
|
|
port: current_application_settings[:metrics_port],
|
|
sample_interval: current_application_settings[:metrics_sample_interval] || 15
|
|
}
|
|
end
|
|
|
|
def self.enabled?
|
|
settings[:enabled] || false
|
|
end
|
|
|
|
def self.mri?
|
|
RUBY_ENGINE == 'ruby'
|
|
end
|
|
|
|
def self.method_call_threshold
|
|
# This is memoized since this method is called for every instrumented
|
|
# method. Loading data from an external cache on every method call slows
|
|
# things down too much.
|
|
@method_call_threshold ||= settings[:method_call_threshold]
|
|
end
|
|
|
|
def self.pool
|
|
@pool
|
|
end
|
|
|
|
def self.submit_metrics(metrics)
|
|
prepared = prepare_metrics(metrics)
|
|
|
|
pool.with do |connection|
|
|
prepared.each do |metric|
|
|
begin
|
|
connection.write_points([metric])
|
|
rescue StandardError
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def self.prepare_metrics(metrics)
|
|
metrics.map do |hash|
|
|
new_hash = hash.symbolize_keys
|
|
|
|
new_hash[:tags].each do |key, value|
|
|
if value.blank?
|
|
new_hash[:tags].delete(key)
|
|
else
|
|
new_hash[:tags][key] = escape_value(value)
|
|
end
|
|
end
|
|
|
|
new_hash
|
|
end
|
|
end
|
|
|
|
def self.escape_value(value)
|
|
value.to_s.gsub('=', '\\=')
|
|
end
|
|
|
|
# Measures the execution time of a block.
|
|
#
|
|
# Example:
|
|
#
|
|
# Gitlab::Metrics.measure(:find_by_username_timings) do
|
|
# User.find_by_username(some_username)
|
|
# end
|
|
#
|
|
# series - The name of the series to store the data in.
|
|
# values - A Hash containing extra values to add to the metric.
|
|
# tags - A Hash containing extra tags to add to the metric.
|
|
#
|
|
# Returns the value yielded by the supplied block.
|
|
def self.measure(series, values = {}, tags = {})
|
|
return yield unless Transaction.current
|
|
|
|
start = Time.now.to_f
|
|
retval = yield
|
|
duration = (Time.now.to_f - start) * 1000.0
|
|
values = values.merge(duration: duration)
|
|
|
|
Transaction.current.add_metric(series, values, tags)
|
|
|
|
retval
|
|
end
|
|
|
|
# When enabled this should be set before being used as the usual pattern
|
|
# "@foo ||= bar" is _not_ thread-safe.
|
|
if enabled?
|
|
@pool = ConnectionPool.new(size: settings[:pool_size], timeout: settings[:timeout]) do
|
|
host = settings[:host]
|
|
port = settings[:port]
|
|
|
|
InfluxDB::Client.
|
|
new(udp: { host: host, port: port })
|
|
end
|
|
end
|
|
end
|
|
end
|