From 59ae4058e6107f276bdb22bcd018f36927d9da4a Mon Sep 17 00:00:00 2001
From: Morten Tokle <mortent@verizonmedia.com>
Date: Tue, 14 Dec 2021 08:35:52 +0100
Subject: Provide feed client builder implementation

---
 .../hosted/cd/commons/EndpointAuthenticator.java   |  7 ++++--
 .../vespa/hosted/cd/commons/FeedClientBuilder.java | 25 ++++++++++++++++++++++
 2 files changed, 30 insertions(+), 2 deletions(-)
 create mode 100644 tenant-cd-commons/src/main/java/ai/vespa/hosted/cd/commons/FeedClientBuilder.java

(limited to 'tenant-cd-commons/src/main/java')

diff --git a/tenant-cd-commons/src/main/java/ai/vespa/hosted/cd/commons/EndpointAuthenticator.java b/tenant-cd-commons/src/main/java/ai/vespa/hosted/cd/commons/EndpointAuthenticator.java
index 90775ab76af..5317cf01e21 100644
--- a/tenant-cd-commons/src/main/java/ai/vespa/hosted/cd/commons/EndpointAuthenticator.java
+++ b/tenant-cd-commons/src/main/java/ai/vespa/hosted/cd/commons/EndpointAuthenticator.java
@@ -4,6 +4,7 @@ package ai.vespa.hosted.cd.commons;
 import javax.net.ssl.SSLContext;
 import java.net.http.HttpRequest;
 import java.security.NoSuchAlgorithmException;
+import java.util.Map;
 
 /**
  * Adds environment dependent authentication to HTTP request against Vespa deployments.
@@ -20,8 +21,7 @@ public interface EndpointAuthenticator {
     default SSLContext sslContext() {
         try {
             return SSLContext.getDefault();
-        }
-        catch (NoSuchAlgorithmException e) {
+        } catch (NoSuchAlgorithmException e) {
             throw new RuntimeException(e);
         }
     }
@@ -31,4 +31,7 @@ public interface EndpointAuthenticator {
         return request;
     }
 
+    default Map<String, String> authorizationHeaders() {
+        return Map.of();
+    }
 }
diff --git a/tenant-cd-commons/src/main/java/ai/vespa/hosted/cd/commons/FeedClientBuilder.java b/tenant-cd-commons/src/main/java/ai/vespa/hosted/cd/commons/FeedClientBuilder.java
new file mode 100644
index 00000000000..892c8b63645
--- /dev/null
+++ b/tenant-cd-commons/src/main/java/ai/vespa/hosted/cd/commons/FeedClientBuilder.java
@@ -0,0 +1,25 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+package ai.vespa.hosted.cd.commons;
+
+import ai.vespa.feed.client.impl.FeedClientBuilderImpl;
+
+import java.util.Objects;
+import java.util.concurrent.atomic.AtomicReference;
+
+/**
+ * @author mortent
+ */
+public class FeedClientBuilder extends FeedClientBuilderImpl {
+
+    static AtomicReference<EndpointAuthenticator> endpointAuthenticator = new AtomicReference<>();
+
+    public static void setEndpointAuthenticator(EndpointAuthenticator authenticator) {
+        endpointAuthenticator.set(authenticator);
+    }
+
+    public FeedClientBuilder() {
+        super.setSslContext(Objects.requireNonNull(endpointAuthenticator.get(), FeedClientBuilder.class.getName() + " is not initialized").sslContext());
+        endpointAuthenticator.get().authorizationHeaders().forEach(super::addRequestHeader);
+    }
+}
-- 
cgit v1.2.3