From 6e4d5d8ba9da5133d3ac6db99f357d7357a516c5 Mon Sep 17 00:00:00 2001
From: Bjørn Christian Seime <bjorncs@oath.com>
Date: Wed, 17 Jan 2018 11:01:46 +0100
Subject: Handle empty responses

---
 .../controller/proxy/ConfigServerRestExecutorImpl.java | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

(limited to 'controller-server/src')

diff --git a/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/proxy/ConfigServerRestExecutorImpl.java b/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/proxy/ConfigServerRestExecutorImpl.java
index 4d4cc2f3146..1757e7eec02 100644
--- a/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/proxy/ConfigServerRestExecutorImpl.java
+++ b/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/proxy/ConfigServerRestExecutorImpl.java
@@ -34,6 +34,7 @@ import javax.net.ssl.SSLSession;
 import javax.net.ssl.SSLSocket;
 import java.io.IOException;
 import java.io.InputStream;
+import java.io.UncheckedIOException;
 import java.net.URI;
 import java.security.cert.X509Certificate;
 import java.time.Duration;
@@ -139,7 +140,7 @@ public class ConfigServerRestExecutorImpl implements ConfigServerRestExecutor {
                 CloseableHttpClient client = createHttpClient(config, sslContextProvider, zoneRegistry, proxyRequest);
                 CloseableHttpResponse response = client.execute(requestBase);
         ) {
-            String content = EntityUtils.toString(response.getEntity());
+            String content = getContent(response);
             int status = response.getStatusLine().getStatusCode();
             if (status / 100 == 5) {
                 errorBuilder.append("Talking to server ").append(uri.getHost());
@@ -158,7 +159,7 @@ public class ConfigServerRestExecutorImpl implements ConfigServerRestExecutor {
             }
             // Send response back
             return Optional.of(new ProxyResponse(proxyRequest, content, status, Optional.of(uri), contentType));
-        } catch (IOException|RuntimeException e) {
+        } catch (Exception e) {
             errorBuilder.append("Talking to server ").append(uri.getHost());
             errorBuilder.append(" got exception ").append(e.getMessage());
             log.log(LogLevel.DEBUG, e, () -> "Got exception while sending request to " + uri.getHost());
@@ -166,6 +167,19 @@ public class ConfigServerRestExecutorImpl implements ConfigServerRestExecutor {
         }
     }
 
+    private static String getContent(CloseableHttpResponse response) {
+        return Optional.ofNullable(response.getEntity())
+                .map(entity ->
+                     {
+                         try {
+                             return EntityUtils.toString(entity);
+                         } catch (IOException e) {
+                             throw new UncheckedIOException(e);
+                         }
+                     }
+                ).orElse("");
+    }
+
     private HttpRequestBase createHttpBaseRequest(String method, String uri, InputStream data) throws ProxyException {
         Method enumMethod =  Method.valueOf(method);
         switch (enumMethod) {
-- 
cgit v1.2.3