summaryrefslogtreecommitdiff
path: root/app/models/project_services
diff options
context:
space:
mode:
Diffstat (limited to 'app/models/project_services')
-rw-r--r--app/models/project_services/prometheus_service.rb22
1 files changed, 16 insertions, 6 deletions
diff --git a/app/models/project_services/prometheus_service.rb b/app/models/project_services/prometheus_service.rb
index 2fb94c1facb..580e2c01aaa 100644
--- a/app/models/project_services/prometheus_service.rb
+++ b/app/models/project_services/prometheus_service.rb
@@ -116,31 +116,41 @@ class PrometheusService < MonitoringService
{ success: false, result: err.message }
end
- def client(environment_id)
+ def client(environment_id = nil)
if manual_configuration?
Gitlab::PrometheusClient.new(RestClient::Resource.new(api_url))
else
cluster = cluster_with_prometheus(environment_id)
raise Gitlab::PrometheusError, "couldn't find cluster with Prometheus installed" unless cluster
+ rest_client = client_from_cluster(cluster)
- Gitlab::PrometheusClient.new(cluster.application_prometheus.proxy_client)
+ raise Gitlab::PrometheusError, "couldn't create proxy Prometheus client" unless rest_client
+ Gitlab::PrometheusClient.new(rest_client)
end
end
def prometheus_installed?
- cluster_with_prometheus.present?
+ project.clusters.enabled.any? { |cluster| cluster.application_prometheus&.installed? }
end
private
def cluster_with_prometheus(environment_id = nil)
clusters = if environment_id
- ::Environment.find_by(id: environment_id).try(:enabled_clusters) || []
+ ::Environment.find_by(id: environment_id).try do |env|
+ # sort results by descending order based on environment_scope being longer
+ # thus more closely matching environment slug
+ project.clusters.enabled.for_environment(env).sort_by { |c| c.environment_scope&.length }.reverse!
+ end
else
- project.clusters.enabled.select { |c| c.environment_scope == '*' || c.environment_scope == '' }
+ project.clusters.enabled.for_all_environments
end
- clusters.detect { |cluster| cluster.application_prometheus&.installed? }
+ clusters&.detect { |cluster| cluster.application_prometheus&.installed? }
+ end
+
+ def client_from_cluster(cluster)
+ cluster.application_prometheus.proxy_client
end
def rename_data_to_metrics(metrics)