WIP

Author: artem-v

tokens/src/main/java/io/scalecube/security/tokens/jwt/JwksKeyLocator.java

@@ -45,7 +45,10 @@ private JwksKeyLocator(Builder builder) {
     this.connectTimeout = Objects.requireNonNull(builder.connectTimeout, "connectTimeout");
     this.requestTimeout = Objects.requireNonNull(builder.requestTimeout, "requestTimeout");
     this.keyTtl = builder.keyTtl;
-    this.httpClient = HttpClient.newBuilder().connectTimeout(connectTimeout).build();
+    this.httpClient =
+        builder.httpClient != null
+            ? builder.httpClient
+            : HttpClient.newBuilder().connectTimeout(connectTimeout).build();
   }
 
   public static Builder builder() {
@@ -160,6 +163,7 @@ public static class Builder {
     private Duration connectTimeout = Duration.ofSeconds(10);
     private Duration requestTimeout = Duration.ofSeconds(10);
     private int keyTtl = 60 * 1000;
+    private HttpClient httpClient;
 
     private Builder() {}
 
@@ -211,6 +215,17 @@ public Builder keyTtl(int keyTtl) {
       return this;
     }
 
+    /**
+     * Setter for optional {@link HttpClient}.
+     *
+     * @param httpClient httpClient
+     * @return this
+     */
+    public Builder httpClient(HttpClient httpClient) {
+      this.httpClient = httpClient;
+      return this;
+    }
+
     public JwksKeyLocator build() {
       return new JwksKeyLocator(this);
     }