summaryrefslogtreecommitdiff
path: root/lib/gitlab/prometheus/queries/knative_invocation_query.rb
blob: 2691abe46d6288bf9884c998a3e951b2a412bfdd (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
# frozen_string_literal: true

module Gitlab
  module Prometheus
    module Queries
      class KnativeInvocationQuery < BaseQuery
        include QueryAdditionalMetrics

        def query(serverless_function_id)
          PrometheusMetric
            .find_by_identifier(:system_metrics_knative_function_invocation_count)
            .to_query_metric.tap do |q|
            q.queries[0][:result] = run_query(q.queries[0][:query_range], context(serverless_function_id))
          end
        end

        protected

        def context(function_id)
          function = Serverless::Function.find_by_id(function_id)
          {
            function_name: function.name,
            kube_namespace: function.namespace
          }
        end

        def run_query(query, context)
          query %= context
          client_query_range(query, start: 8.hours.ago.to_f, stop: Time.now.to_f)
        end

        def self.transform_reactive_result(result)
          result[:metrics] = result.delete :data
          result
        end
      end
    end
  end
end