1 files changed, 6 insertions, 3 deletions
diff --git a/jaxrs_client_utils/src/main/java/com/yahoo/vespa/jaxrs/client/JaxRsStrategyFactory.java b/jaxrs_client_utils/src/main/java/com/yahoo/vespa/jaxrs/client/JaxRsStrategyFactory.java
index 722652557c7..6523a0c138f 100644
--- a/jaxrs_client_utils/src/main/java/com/yahoo/vespa/jaxrs/client/JaxRsStrategyFactory.java
+++ b/jaxrs_client_utils/src/main/java/com/yahoo/vespa/jaxrs/client/JaxRsStrategyFactory.java
@@ -34,13 +34,15 @@ import java.util.Set;
 public class JaxRsStrategyFactory {
     private final Set<HostName> hostNames;
     private int port;
+    private final String scheme;
     private final JaxRsClientFactory jaxRsClientFactory;
 
     // TODO: We might need to support per-host port specification.
     public JaxRsStrategyFactory(
             final Set<HostName> hostNames,
             final int port,
-            final JaxRsClientFactory jaxRsClientFactory) {
+            final JaxRsClientFactory jaxRsClientFactory,
+            String scheme) {
         if (hostNames.isEmpty()) {
             throw new IllegalArgumentException("hostNames argument must not be empty");
         }
@@ -48,19 +50,20 @@ public class JaxRsStrategyFactory {
         this.hostNames = hostNames;
         this.port = port;
         this.jaxRsClientFactory = jaxRsClientFactory;
+        this.scheme = scheme;
     }
 
     public <T> JaxRsStrategy<T> apiWithRetries(final Class<T> apiClass, final String pathPrefix) {
         Objects.requireNonNull(apiClass, "apiClass argument may not be null");
         Objects.requireNonNull(pathPrefix, "pathPrefix argument may not be null");
-        return new RetryingJaxRsStrategy<T>(hostNames, port, jaxRsClientFactory, apiClass, pathPrefix);
+        return new RetryingJaxRsStrategy<T>(hostNames, port, jaxRsClientFactory, apiClass, pathPrefix, scheme);
     }
 
     public <T> JaxRsStrategy<T> apiNoRetries(final Class<T> apiClass, final String pathPrefix) {
         Objects.requireNonNull(apiClass, "apiClass argument may not be null");
         Objects.requireNonNull(pathPrefix, "pathPrefix argument may not be null");
         final HostName hostName = getRandom(hostNames);
-        return new NoRetryJaxRsStrategy<T>(hostName, port, jaxRsClientFactory, apiClass, pathPrefix);
+        return new NoRetryJaxRsStrategy<T>(hostName, port, jaxRsClientFactory, apiClass, pathPrefix, scheme);
     }
 
     private static final Random random = new Random();