Add latest changes from gitlab-org/gitlab@master

1 files changed, 34 insertions, 0 deletions

diff --git a/lib/gitlab/prometheus/adapter.rb b/lib/gitlab/prometheus/adapter.rb
new file mode 100644
index 00000000000..ed10ef2917f
--- /dev/null
+++ b/lib/gitlab/prometheus/adapter.rb
@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module Prometheus
+    class Adapter
+      attr_reader :project, :cluster
+
+      def initialize(project, cluster)
+        @project = project
+        @cluster = cluster
+      end
+
+      def prometheus_adapter
+        @prometheus_adapter ||= if service_prometheus_adapter.can_query?
+                                  service_prometheus_adapter
+                                else
+                                  cluster_prometheus_adapter
+                                end
+      end
+
+      def cluster_prometheus_adapter
+        application = cluster&.application_prometheus
+
+        application if application&.available?
+      end
+
+      private
+
+      def service_prometheus_adapter
+        project.find_or_initialize_service('prometheus')
+      end
+    end
+  end
+end