1 files changed, 95 insertions, 0 deletions
diff --git a/gems/gitlab-http/lib/gitlab/http_v2/client.rb b/gems/gitlab-http/lib/gitlab/http_v2/client.rb
new file mode 100644
index 00000000000..8daf19d7351
--- /dev/null
+++ b/gems/gitlab-http/lib/gitlab/http_v2/client.rb
@@ -0,0 +1,95 @@
+# frozen_string_literal: true
+
+require 'httparty'
+require 'net/http'
+require 'active_support/all'
+require_relative 'new_connection_adapter'
+require_relative "exceptions"
+
+module Gitlab
+  module HTTP_V2
+    class Client
+      DEFAULT_TIMEOUT_OPTIONS = {
+        open_timeout: 10,
+        read_timeout: 20,
+        write_timeout: 30
+      }.freeze
+      DEFAULT_READ_TOTAL_TIMEOUT = 30.seconds
+
+      SILENT_MODE_ALLOWED_METHODS = [
+        Net::HTTP::Get,
+        Net::HTTP::Head,
+        Net::HTTP::Options,
+        Net::HTTP::Trace
+      ].freeze
+
+      include HTTParty # rubocop:disable Gitlab/HTTParty
+
+      class << self
+        alias_method :httparty_perform_request, :perform_request
+      end
+
+      connection_adapter NewConnectionAdapter
+
+      def self.perform_request(http_method, path, options, &block)
+        raise_if_blocked_by_silent_mode(http_method) if options.delete(:silent_mode_enabled)
+
+        log_info = options.delete(:extra_log_info)
+        options_with_timeouts =
+          if !options.has_key?(:timeout)
+            options.with_defaults(DEFAULT_TIMEOUT_OPTIONS)
+          else
+            options
+          end
+
+        if options[:stream_body]
+          httparty_perform_request(http_method, path, options_with_timeouts, &block)
+        else
+          begin
+            start_time = nil
+            read_total_timeout = options.fetch(:timeout, DEFAULT_READ_TOTAL_TIMEOUT)
+
+            httparty_perform_request(http_method, path, options_with_timeouts) do |fragment|
+              start_time ||= system_monotonic_time
+              elapsed = system_monotonic_time - start_time
+
+              raise ReadTotalTimeout, "Request timed out after #{elapsed} seconds" if elapsed > read_total_timeout
+
+              yield fragment if block
+            end
+          rescue HTTParty::RedirectionTooDeep
+            raise RedirectionTooDeep
+          rescue *HTTP_ERRORS => e
+            extra_info = log_info || {}
+            extra_info = log_info.call(e, path, options) if log_info.respond_to?(:call)
+            configuration.log_exception(e, extra_info)
+
+            raise e
+          end
+        end
+      end
+
+      def self.try_get(path, options = {}, &block)
+        self.get(path, options, &block) # rubocop:disable Style/RedundantSelf
+      rescue *HTTP_ERRORS
+        nil
+      end
+
+      def self.raise_if_blocked_by_silent_mode(http_method)
+        return if SILENT_MODE_ALLOWED_METHODS.include?(http_method)
+
+        configuration.silent_mode_log_info('Outbound HTTP request blocked', http_method.to_s)
+
+        raise SilentModeBlockedError, 'only get, head, options, and trace methods are allowed in silent mode'
+      end
+
+      def self.system_monotonic_time
+        Process.clock_gettime(Process::CLOCK_MONOTONIC, :float_second)
+      end
+
+      def self.configuration
+        Gitlab::HTTP_V2.configuration
+      end
+    end
+  end
+end