fix fabric8io#4471: kubernetesclientbuilder method for the httpclient…

….builder

CHANGELOG.md

@@ -16,6 +16,7 @@
 * Fix #4426: [java-generator] Encode an `AnyType` instead of an Object if `x-kubernetes-preserve-unknown-fields` is present and the type is null.
 
 #### Improvements
+* Fix #4471: Adding KubernetesClientBuilder.withHttpClientBuilderConsumer to further customize the HttpClient for any implementation.
 * Fix #4348: Introduce specific annotations for the generators
 * Refactor #4441: refactoring `TokenRefreshInterceptor`
 * Fix #4365: The Watch retry logic will handle more cases, as well as perform an exceptional close for events that are not properly handled.  Informers can directly provide those exceptional outcomes via the SharedIndexInformer.stopped CompletableFuture.

httpclient-jdk/src/main/java/io/fabric8/kubernetes/client/jdkhttp/JdkHttpClientFactory.java

@@ -18,7 +18,6 @@
 
 import io.fabric8.kubernetes.client.Config;
 import io.fabric8.kubernetes.client.http.HttpClient;
-import io.fabric8.kubernetes.client.utils.HttpClientUtils;
 import io.fabric8.kubernetes.client.utils.Utils;
 
 import java.util.concurrent.Executor;
@@ -46,15 +45,6 @@ public void shutdownNow() {
 
   }
 
-  @Override
-  public HttpClient createHttpClient(Config config) {
-    JdkHttpClientBuilderImpl builderWrapper = newBuilder();
-
-    HttpClientUtils.applyCommonConfiguration(config, builderWrapper, this);
-
-    return builderWrapper.build();
-  }
-
   @Override
   public JdkHttpClientBuilderImpl newBuilder() {
     return new JdkHttpClientBuilderImpl(this);

httpclient-jetty/src/main/java/io/fabric8/kubernetes/client/jetty/JettyHttpClientFactory.java

@@ -15,20 +15,10 @@
  */
 package io.fabric8.kubernetes.client.jetty;
 
-import io.fabric8.kubernetes.client.Config;
 import io.fabric8.kubernetes.client.http.HttpClient;
 
-import static io.fabric8.kubernetes.client.utils.HttpClientUtils.applyCommonConfiguration;
-
 public class JettyHttpClientFactory implements HttpClient.Factory {
 
-  @Override
-  public HttpClient createHttpClient(Config config) {
-    final var builder = newBuilder();
-    applyCommonConfiguration(config, builder, this);
-    return builder.build();
-  }
-
   @Override
   public JettyHttpClientBuilder newBuilder() {
     return new JettyHttpClientBuilder(this);

httpclient-jetty/src/test/java/io/fabric8/kubernetes/client/jetty/JettyHttpClientFactoryTest.java

@@ -27,7 +27,7 @@ class JettyHttpClientFactoryTest {
   @DisplayName("createHttpClient instantiates a JettyHttpClient")
   void createHttpClientInstantiatesJettyHttpClient() {
     // When
-    try (var result = new JettyHttpClientFactory().createHttpClient(Config.empty())) {
+    try (var result = new JettyHttpClientFactory().newBuilder(Config.empty()).build()) {
       // Then
       assertThat(result).isNotNull().isInstanceOf(JettyHttpClient.class);
     }

httpclient-okhttp/src/main/java/io/fabric8/kubernetes/client/okhttp/OkHttpClientFactory.java

@@ -59,13 +59,13 @@ public Builder newBuilder() {
   }
 
   /**
-   * Creates an HTTP client configured to access the Kubernetes API.
+   * Creates an HTTP client builder configured to access the Kubernetes API.
    *
    * @param config Kubernetes API client config
-   * @return returns an HTTP client
+   * @return returns an HTTP client builder
    */
   @Override
-  public OkHttpClientImpl createHttpClient(Config config) {
+  public OkHttpClientBuilderImpl newBuilder(Config config) {
     try {
       OkHttpClient.Builder httpClientBuilder = newOkHttpClientBuilder();
 
@@ -101,7 +101,7 @@ public OkHttpClientImpl createHttpClient(Config config) {
 
       additionalConfig(httpClientBuilder);
 
-      return builderWrapper.build();
+      return builderWrapper;
     } catch (Exception e) {
       throw KubernetesClientException.launderThrowable(e);
     }

kubernetes-client-api/src/main/java/io/fabric8/kubernetes/client/KubernetesClientBuilder.java

@@ -23,6 +23,7 @@
 import java.io.InputStream;
 import java.lang.reflect.InvocationTargetException;
 import java.util.concurrent.Executor;
+import java.util.function.Consumer;
 import java.util.function.Supplier;
 
 /**
@@ -44,6 +45,7 @@ default void onClose(Executor executor) {
   private HttpClient.Factory factory;
   private Class<KubernetesClient> clazz;
   private ExecutorSupplier executorSupplier;
+  private Consumer<HttpClient.Builder> builderConsumer;
 
   public KubernetesClientBuilder() {
     // basically the same logic as in KubernetesResourceUtil for finding list types
@@ -60,6 +62,10 @@ public KubernetesClientBuilder() {
     }
   }
 
+  KubernetesClientBuilder(Class<KubernetesClient> clazz) {
+    this.clazz = clazz;
+  }
+
   public KubernetesClient build() {
     if (config == null) {
       config = new ConfigBuilder().build();
@@ -68,7 +74,7 @@ public KubernetesClient build() {
       if (factory == null) {
         return clazz.getConstructor(Config.class).newInstance(config);
       }
-      HttpClient client = factory.createHttpClient(config);
+      HttpClient client = getHttpClient();
       return clazz.getConstructor(HttpClient.class, Config.class, ExecutorSupplier.class).newInstance(client, config,
           executorSupplier);
     } catch (InstantiationException | IllegalAccessException | IllegalArgumentException | InvocationTargetException
@@ -77,6 +83,14 @@ public KubernetesClient build() {
     }
   }
 
+  HttpClient getHttpClient() {
+    HttpClient.Builder builder = factory.newBuilder(config);
+    if (this.builderConsumer != null) {
+      this.builderConsumer.accept(builder);
+    }
+    return builder.build();
+  }
+
   public KubernetesClientBuilder withConfig(Config config) {
     this.config = config;
     return this;
@@ -125,4 +139,15 @@ public KubernetesClientBuilder withTaskExecutorSupplier(ExecutorSupplier executo
     return this;
   }
 
+  /**
+   * Provide additional configuration for the {@link HttpClient} that is created for this {@link KubernetesClient}.
+   *
+   * @param consumer to modify the {@link HttpClient.Builder}
+   * @return this builder
+   */
+  public KubernetesClientBuilder withHttpClientBuilderConsumer(Consumer<HttpClient.Builder> consumer) {
+    this.builderConsumer = consumer;
+    return this;
+  }
+
 }

kubernetes-client-api/src/main/java/io/fabric8/kubernetes/client/http/HttpClient.java

@@ -18,6 +18,7 @@
 
 import io.fabric8.kubernetes.client.Config;
 import io.fabric8.kubernetes.client.RequestConfig;
+import io.fabric8.kubernetes.client.utils.HttpClientUtils;
 
 import java.io.BufferedReader;
 import java.net.InetSocketAddress;
@@ -33,7 +34,18 @@ public interface HttpClient extends AutoCloseable {
 
   interface Factory {
 
-    HttpClient createHttpClient(Config config);
+    /**
+     * Create a builder that is customized by the {@link Config}. By default it
+     * will apply the common configuration {@link HttpClientUtils#applyCommonConfiguration(Config, Builder, Factory)}
+     *
+     * @param config the configuration to apply
+     * @return the configured {@link Builder}
+     */
+    default HttpClient.Builder newBuilder(Config config) {
+      Builder builder = newBuilder();
+      HttpClientUtils.applyCommonConfiguration(config, builder, this);
+      return builder;
+    }
 
     HttpClient.Builder newBuilder();
 

kubernetes-client-api/src/main/java/io/fabric8/kubernetes/client/utils/HttpClientUtils.java

@@ -159,7 +159,7 @@ public static HttpClient createHttpClient(Config config) {
       throw new KubernetesClientException(
           "No httpclient implementations found on the context classloader, please ensure your classpath includes an implementation jar");
     }
-    return factory.createHttpClient(config);
+    return factory.newBuilder(config).build();
   }
 
   public static void applyCommonConfiguration(Config config, HttpClient.Builder builder, HttpClient.Factory factory) {

kubernetes-client-api/src/test/java/io/fabric8/kubernetes/client/KubernetesClientBuilderTest.java

@@ -0,0 +1,37 @@
+/**
+ * Copyright (C) 2015 Red Hat, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *         http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.fabric8.kubernetes.client;
+
+import io.fabric8.kubernetes.client.http.HttpClient;
+import io.fabric8.kubernetes.client.http.HttpClient.Factory;
+import org.junit.jupiter.api.Test;
+import org.mockito.Mockito;
+
+class KubernetesClientBuilderTest {
+
+  @Test
+  void testHttpClientConfiguration() {
+    KubernetesClientBuilder builder = new KubernetesClientBuilder(null);
+    Factory mockFactory = Mockito.mock(HttpClient.Factory.class);
+    HttpClient.Builder mockBuilder = Mockito.mock(HttpClient.Builder.class);
+    Mockito.when(mockFactory.newBuilder(Mockito.any())).thenReturn(mockBuilder);
+    builder.withHttpClientFactory(mockFactory).withHttpClientBuilderConsumer(b -> b.proxyAuthorization("something"));
+    builder.getHttpClient();
+    Mockito.verify(mockBuilder).proxyAuthorization("something");
+  }
+
+}

kubernetes-client/src/test/java/io/fabric8/kubernetes/client/okhttp/OkHttpClientFactoryTest.java

@@ -37,29 +37,29 @@ class OkHttpClientFactoryTest {
 
   @Test
   void shouldRespectMaxRequests() {
-    OkHttpClientImpl client = new OkHttpClientFactory().createHttpClient(new ConfigBuilder().build());
+    OkHttpClientImpl client = new OkHttpClientFactory().newBuilder(new ConfigBuilder().build()).build();
 
     assertEquals(64, client.getOkHttpClient().dispatcher().getMaxRequests());
 
     Config config = new ConfigBuilder()
         .withMaxConcurrentRequests(120)
         .build();
 
-    client = new OkHttpClientFactory().createHttpClient(config);
+    client = new OkHttpClientFactory().newBuilder(config).build();
     assertEquals(120, client.getOkHttpClient().dispatcher().getMaxRequests());
   }
 
   @Test
   void shouldRespectMaxRequestsPerHost() {
-    OkHttpClientImpl client = new OkHttpClientFactory().createHttpClient(new ConfigBuilder().build());
+    OkHttpClientImpl client = new OkHttpClientFactory().newBuilder(new ConfigBuilder().build()).build();
 
     assertEquals(5, client.getOkHttpClient().dispatcher().getMaxRequestsPerHost());
 
     Config config = new ConfigBuilder()
         .withMaxConcurrentRequestsPerHost(20)
         .build();
 
-    client = new OkHttpClientFactory().createHttpClient(config);
+    client = new OkHttpClientFactory().newBuilder(config).build();
 
     assertEquals(20, client.getOkHttpClient().dispatcher().getMaxRequestsPerHost());
   }