1 files changed, 83 insertions, 0 deletions
diff --git a/qa/qa/support/influxdb_tools.rb b/qa/qa/support/influxdb_tools.rb
new file mode 100644
index 00000000000..e53b843ca87
--- /dev/null
+++ b/qa/qa/support/influxdb_tools.rb
@@ -0,0 +1,83 @@
+# frozen_string_literal: true
+
+require "active_support/core_ext/module/delegation"
+
+module QA
+  module Support
+    # Common tools for use with influxdb metrics setup
+    #
+    module InfluxdbTools
+      INFLUX_TEST_METRICS_BUCKET = "e2e-test-stats"
+      LIVE_ENVS = %w[staging staging-canary staging-ref canary preprod production].freeze
+
+      private
+
+      delegate :ci_project_name, to: "QA::Runtime::Env"
+
+      # Query client
+      #
+      # @return [QueryApi]
+      def query_api
+        @query_api ||= influx_client.create_query_api
+      end
+
+      # Write client
+      #
+      # @return [WriteApi]
+      def write_api
+        @write_api ||= influx_client.create_write_api
+      end
+
+      # InfluxDb client
+      #
+      # @return [InfluxDB2::Client]
+      def influx_client
+        @influx_client ||= InfluxDB2::Client.new(
+          ENV["QA_INFLUXDB_URL"] || raise("Missing QA_INFLUXDB_URL env variable"),
+          ENV["QA_INFLUXDB_TOKEN"] || raise("Missing QA_INFLUXDB_TOKEN env variable"),
+          bucket: INFLUX_TEST_METRICS_BUCKET,
+          org: "gitlab-qa",
+          precision: InfluxDB2::WritePrecision::NANOSECOND
+        )
+      end
+
+      # Test run type
+      # Automatically infer for staging (`gstg`, `gstg-cny`, `gstg-ref`), canary, preprod or production env
+      #
+      # @return [String, nil]
+      def run_type
+        @run_type ||= begin
+          return env('QA_RUN_TYPE') if env('QA_RUN_TYPE')
+          return unless LIVE_ENVS.include?(ci_project_name)
+
+          test_subset = if env('NO_ADMIN') == 'true'
+                          'sanity-no-admin'
+                        elsif env('SMOKE_ONLY') == 'true'
+                          'sanity'
+                        else
+                          'full'
+                        end
+
+          "#{ci_project_name}-#{test_subset}"
+        end
+      end
+
+      # Merge request iid
+      #
+      # @return [String]
+      def merge_request_iid
+        env('CI_MERGE_REQUEST_IID') || env('TOP_UPSTREAM_MERGE_REQUEST_IID')
+      end
+
+      # Return non empty environment variable value
+      #
+      # @param [String] name
+      # @return [String, nil]
+      def env(name)
+        return unless ENV[name] && !ENV[name].empty?
+
+        ENV[name]
+      end
+    end
+  end
+end