search-api-v2/app/services/metrics/quality_monitoring.rb at 9b8ce2aa09e19cf6cb70d41c44d97a0a539c6742 · alphagov/search-api-v2 · GitHub

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
module Metrics
  class QualityMonitoring
    def initialize(registry)
      @quality_monitoring_score = registry.gauge(
        :search_api_v2_quality_monitoring_score,
        docstring: "Quality monitoring scores for Search API v2 (between 0 and 1)",
        labels: %i[dataset_type dataset_name],
      )
      @quality_monitoring_failures = registry.gauge(
        :search_api_v2_quality_monitoring_failures,
        docstring: "Quality monitoring failure count for Search API v2",
        labels: %i[dataset_type dataset_name],
      )
      @quality_monitoring_total = registry.gauge(
        :search_api_v2_quality_monitoring_total,
        docstring: "Quality monitoring total count for Search API v2",
        labels: %i[dataset_type dataset_name],
      )
    end

    def record_score(dataset_type, dataset_name, score)
      quality_monitoring_score.set(score, labels: {
        dataset_type: dataset_type.to_s,
        dataset_name: dataset_name.to_s,
      })
    end

    def record_failure_count(dataset_type, dataset_name, count)
      quality_monitoring_failures.set(count, labels: {
        dataset_type: dataset_type.to_s,
        dataset_name: dataset_name.to_s,
      })
    end

    def record_total_count(dataset_type, dataset_name, count)
      quality_monitoring_total.set(count, labels: {
        dataset_type: dataset_type.to_s,
        dataset_name: dataset_name.to_s,
      })
    end

  private

    attr_reader :quality_monitoring_score, :quality_monitoring_failures, :quality_monitoring_total
  end
end