debian-mirror-gitlab/lib/gitlab/metrics/dashboard/stages/endpoint_inserter.rb
2019-09-04 21:01:54 +05:30

42 lines
1 KiB
Ruby

# frozen_string_literal: true
module Gitlab
module Metrics
module Dashboard
module Stages
class EndpointInserter < BaseStage
MissingQueryError = Class.new(DashboardProcessingError)
def transform!
for_metrics do |metric|
metric[:prometheus_endpoint_path] = endpoint_for_metric(metric)
end
end
private
def endpoint_for_metric(metric)
Gitlab::Routing.url_helpers.prometheus_api_project_environment_path(
project,
environment,
proxy_path: query_type(metric),
query: query_for_metric(metric)
)
end
def query_type(metric)
metric[:query] ? :query : :query_range
end
def query_for_metric(metric)
query = metric[query_type(metric)]
raise MissingQueryError.new('Each "metric" must define one of :query or :query_range') unless query
query
end
end
end
end
end
end