Fix hang on non-200 responses on hadoop query

1 files changed, 4 insertions, 4 deletions

diff --git a/hadoop/src/main/java/com/yahoo/vespa/hadoop/mapreduce/util/VespaHttpClient.java b/hadoop/src/main/java/com/yahoo/vespa/hadoop/mapreduce/util/VespaHttpClient.java
index d62ec99d1bd..e63a4044c95 100644
--- a/hadoop/src/main/java/com/yahoo/vespa/hadoop/mapreduce/util/VespaHttpClient.java
+++ b/hadoop/src/main/java/com/yahoo/vespa/hadoop/mapreduce/util/VespaHttpClient.java
@@ -30,10 +30,6 @@ public class VespaHttpClient {
     public String get(String url) throws IOException {
         HttpGet httpGet = new HttpGet(url);
         HttpResponse httpResponse = httpClient.execute(httpGet);
-        if (httpResponse.getStatusLine().getStatusCode() != 200) {
-            return null;
-        }
-
         HttpEntity entity = httpResponse.getEntity();
         InputStream is = entity.getContent();
 
@@ -44,6 +40,10 @@ public class VespaHttpClient {
         }
         EntityUtils.consume(entity);
 
+        if (httpResponse.getStatusLine().getStatusCode() != 200) {
+            return null;
+        }
+
         return result;
     }