debian-mirror-gitlab/db/importers/common_metrics_importer.rb

# frozen_string_literal: true

module Importers
  class PrometheusMetric < ActiveRecord::Base
    enum group: {
      # built-in groups
      nginx_ingress_vts: -1,
      ha_proxy: -2,
      aws_elb: -3,
      nginx: -4,
      kubernetes: -5,
      nginx_ingress: -6,

      # custom groups
      business: 0,
      response: 1,
      system: 2
    }

    scope :common, -> { where(common: true) }

    GROUP_TITLES = {
      business: _('Business metrics (Custom)'),
      response: _('Response metrics (Custom)'),
      system: _('System metrics (Custom)'),
      nginx_ingress_vts: _('Response metrics (NGINX Ingress VTS)'),
      nginx_ingress: _('Response metrics (NGINX Ingress)'),
      ha_proxy: _('Response metrics (HA Proxy)'),
      aws_elb: _('Response metrics (AWS ELB)'),
      nginx: _('Response metrics (NGINX)'),
      kubernetes: _('System metrics (Kubernetes)')
    }.freeze
  end

  class CommonMetricsImporter
    MissingQueryId = Class.new(StandardError)

    attr_reader :content

    def initialize(filename = 'common_metrics.yml')
      @content = YAML.load_file(Rails.root.join('config', 'prometheus', filename))
    end

    def execute
      PrometheusMetric.reset_column_information

      process_content do |id, attributes|
        find_or_build_metric!(id)
          .update!(**attributes)
      end
    end

    private

    def process_content(&blk)
      content.map do |group|
        process_group(group, &blk)
      end
    end

    def process_group(group, &blk)
      attributes = {
        group: find_group_title_key(group['group'])
      }

      group['metrics'].map do |metric|
        process_metric(metric, attributes, &blk)
      end
    end

    def process_metric(metric, attributes, &blk)
      attributes = attributes.merge(
        title: metric['title'],
        y_label: metric['y_label'])

      metric['queries'].map do |query|
        process_metric_query(query, attributes, &blk)
      end
    end

    def process_metric_query(query, attributes, &blk)
      attributes = attributes.merge(
        legend: query['label'],
        query: query['query_range'],
        unit: query['unit'])

      yield(query['id'], attributes)
    end

    def find_or_build_metric!(id)
      raise MissingQueryId unless id

      PrometheusMetric.common.find_by(identifier: id) ||
        PrometheusMetric.new(common: true, identifier: id)
    end

    def find_group_title_key(title)
      PrometheusMetric.groups[find_group_title(title)]
    end

    def find_group_title(title)
      PrometheusMetric::GROUP_TITLES.invert[title]
    end
  end
end