summaryrefslogtreecommitdiff
path: root/app/models/prometheus_metric.rb
blob: 9ddf66cd388f9b20f91bd2a7cd0adaaf337abba3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
# frozen_string_literal: true

class PrometheusMetric < ApplicationRecord
  belongs_to :project, validate: true, inverse_of: :prometheus_metrics
  has_many :prometheus_alerts, inverse_of: :prometheus_metric

  enum group: Enums::PrometheusMetric.groups

  validates :title, presence: true
  validates :query, presence: true
  validates :group, presence: true
  validates :y_label, presence: true
  validates :unit, presence: true
  validates :identifier, uniqueness: { scope: :project_id }, allow_nil: true

  validates :project, presence: true, unless: :common?
  validates :project, absence: true, if: :common?

  scope :for_dashboard_path, -> (dashboard_path) { where(dashboard_path: dashboard_path) }
  scope :for_project, -> (project) { where(project: project) }
  scope :for_group, -> (group) { where(group: group) }
  scope :for_title, -> (title) { where(title: title) }
  scope :for_y_label, -> (y_label) { where(y_label: y_label) }
  scope :for_identifier, -> (identifier) { where(identifier: identifier) }
  scope :not_identifier, -> (identifier) { where.not(identifier: identifier) }
  scope :common, -> { where(common: true) }
  scope :ordered, -> { reorder(created_at: :asc) }

  def priority
    group_details(group).fetch(:priority)
  end

  def group_title
    group_details(group).fetch(:group_title)
  end

  def required_metrics
    group_details(group).fetch(:required_metrics, []).map(&:to_s)
  end

  def to_query_metric
    Gitlab::Prometheus::Metric.new(id: id, title: title, required_metrics: required_metrics, weight: 0, y_label: y_label, queries: queries)
  end

  def to_metric_hash
    queries.first.merge(metric_id: id)
  end

  def queries
    [
      {
        query_range: query,
        unit: unit,
        label: legend,
        series: query_series
      }.compact
    ]
  end

  def query_series
    case legend
    when 'Status Code'
      [{
        label: 'status_code',
        when: [
          { value: '2xx', color: 'green' },
          { value: '4xx', color: 'orange' },
          { value: '5xx', color: 'red' }
        ]
      }]
    end
  end

  private

  def group_details(group)
    Enums::PrometheusMetric.group_details.fetch(group.to_sym)
  end
end