class GitLab::Exporter::Database::BloatProber

Prober class to gather bloat metrics

Constants

METRIC_KEYS

Attributes

bloat_types[R]
collector[R]
metrics[R]

Public Class Methods

new(metrics: PrometheusMetrics.new, logger: nil, **opts) click to toggle source
# File lib/gitlab_exporter/database/bloat.rb, line 42
def initialize(metrics: PrometheusMetrics.new,
               logger: nil,
               **opts)
  @metrics = metrics
  @collector = opts[:collector] || BloatCollector.new(**opts)
  @collector.logger = logger
  @bloat_types = opts[:bloat_types] || %i[btree table]
end

Public Instance Methods

probe_db() click to toggle source
# File lib/gitlab_exporter/database/bloat.rb, line 51
def probe_db
  bloat_types.each do |type|
    probe_for_type(type)
  end
end
write_to(target) click to toggle source
# File lib/gitlab_exporter/database/bloat.rb, line 57
def write_to(target)
  target.write(metrics.to_s)
end

Private Instance Methods

probe_for_type(type) click to toggle source
# File lib/gitlab_exporter/database/bloat.rb, line 63
def probe_for_type(type)
  collector.run(type).each do |query_name, data|
    METRIC_KEYS.each do |key|
      metrics.add("gitlab_database_bloat_#{type}_#{key}", data[key].to_f, query_name: query_name)
    end
  end

  self
rescue PG::ConnectionBad
  self
end