added http response code to JestResult - #208 & #220

searchbox-io · Jul 8, 2015 · 4f04175 · 4f04175
1 parent 372017a
commit 4f04175
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 0 deletions.
diff --git a/jest-common/src/main/java/io/searchbox/action/AbstractAction.java b/jest-common/src/main/java/io/searchbox/action/AbstractAction.java
@@ -65,6 +65,7 @@ public AbstractAction(Builder builder) {
 
     protected T createNewElasticSearchResult(T result, String responseBody, int statusCode, String reasonPhrase, Gson gson) {
         JsonObject jsonMap = parseResponseBody(responseBody);
+        result.setResponseCode(statusCode);
         result.setJsonString(responseBody);
         result.setJsonObject(jsonMap);
         result.setPathToResult(getPathToResult());

diff --git a/jest-common/src/main/java/io/searchbox/client/JestResult.java b/jest-common/src/main/java/io/searchbox/client/JestResult.java
@@ -26,6 +26,7 @@ public class JestResult {
     protected JsonObject jsonObject;
     protected String jsonString;
     protected String pathToResult;
+    protected int responseCode;
     protected boolean isSucceeded;
     protected String errorMessage;
     protected Gson gson;
@@ -37,6 +38,7 @@ public JestResult(JestResult source) {
         this.jsonObject = source.jsonObject;
         this.jsonString = source.jsonString;
         this.pathToResult = source.pathToResult;
+        this.responseCode = source.responseCode;
         this.isSucceeded = source.isSucceeded;
         this.errorMessage = source.errorMessage;
         this.gson = source.gson;
@@ -78,6 +80,14 @@ public String getErrorMessage() {
         return errorMessage;
     }
 
+    public int getResponseCode() {
+        return responseCode;
+    }
+
+    public void setResponseCode(int responseCode) {
+        this.responseCode = responseCode;
+    }
+
     /**
      * manually set an error message, eg. for the cases where non-200 response code is received
      */

diff --git a/jest-common/src/main/java/io/searchbox/core/Bulk.java b/jest-common/src/main/java/io/searchbox/core/Bulk.java
@@ -130,6 +130,7 @@ public BulkResult createNewElasticSearchResult(String responseBody, int statusCo
     @Override
     protected BulkResult createNewElasticSearchResult(BulkResult result, String responseBody, int statusCode, String reasonPhrase, Gson gson) {
         JsonObject jsonMap = parseResponseBody(responseBody);
+        result.setResponseCode(statusCode);
         result.setJsonString(responseBody);
         result.setJsonObject(jsonMap);
         result.setPathToResult(getPathToResult());

diff --git a/jest-common/src/test/java/io/searchbox/action/AbstractActionTest.java b/jest-common/src/test/java/io/searchbox/action/AbstractActionTest.java
@@ -197,6 +197,7 @@ public void getSuccessIndexResult() {
         Index index = new Index.Builder("{\"abc\":\"dce\"}").index("test").build();
         JestResult result = index.createNewElasticSearchResult(jsonString, 200, null, new Gson());
         assertTrue(result.getErrorMessage(), result.isSucceeded());
+        assertEquals(200, result.getResponseCode());
     }
 
     @Test