Add loadbalancer support

Signed-off-by: Eric D. Helms <[email protected]>

Author: ehelms

lib/smart_proxy_container_gateway/container_gateway.rb

@@ -41,7 +41,7 @@ class Plugin < ::Proxy::Plugin
         container_instance.singleton_dependency :container_gateway_main_impl, (lambda do
           Proxy::ContainerGateway::ContainerGatewayMain.new(
             database: container_instance.get_dependency(:database_impl),
-            **settings.slice(:pulp_endpoint, :pulp_client_ssl_ca, :pulp_client_ssl_cert, :pulp_client_ssl_key)
+            **settings.slice(:pulp_endpoint, :pulp_client_ssl_ca, :pulp_client_ssl_cert, :pulp_client_ssl_key, :loadbalancer)
           )
         end)
       end

lib/smart_proxy_container_gateway/container_gateway_api.rb

@@ -45,8 +45,9 @@ class Api < ::Sinatra::Base
           status pulp_response.code.to_i
           body pulp_response.body
         else
-          redirection_location = pulp_response['location']
-          redirect to(redirection_location)
+          redirection_uri = URI(pulp_response['location'])
+          redirection_uri.host = URI(container_gateway_main.loadbalancer).host
+          redirect(redirection_uri.to_s)
         end
       end
 
@@ -217,8 +218,9 @@ def head_or_get_blobs
           status pulp_response.code.to_i
           body pulp_response.body
         else
-          redirection_location = pulp_response['location']
-          redirect to(redirection_location)
+          redirection_uri = URI(pulp_response['location'])
+          redirection_uri.host = URI(container_gateway_main.loadbalancer).host
+          redirect(redirection_uri.to_s)
         end
       end
 

lib/smart_proxy_container_gateway/container_gateway_main.rb

@@ -9,17 +9,20 @@ module ContainerGateway
     extend ::Proxy::Log
 
     class ContainerGatewayMain
-      attr_reader :database
+      attr_reader :database, :loadbalancer
 
-      def initialize(database:, pulp_endpoint:, pulp_client_ssl_ca:, pulp_client_ssl_cert:, pulp_client_ssl_key:)
+      # rubocop:disable Metrics/ParameterLists, Layout/LineLength
+      def initialize(database:, pulp_endpoint:, pulp_client_ssl_ca:, pulp_client_ssl_cert:, pulp_client_ssl_key:, loadbalancer: nil)
         @database = database
         @pulp_endpoint = pulp_endpoint
+        @loadbalancer = loadbalancer || pulp_endpoint
         @pulp_client_ssl_ca = pulp_client_ssl_ca
         @pulp_client_ssl_cert = OpenSSL::X509::Certificate.new(File.read(pulp_client_ssl_cert))
         @pulp_client_ssl_key = OpenSSL::PKey::RSA.new(
           File.read(pulp_client_ssl_key)
         )
       end
+      # rubocop:enable Metrics/ParameterLists, Layout/LineLength
 
       def pulp_registry_request(uri, headers)
         http_client = Net::HTTP.new(uri.host, uri.port)