Reindex scripts now accept both Map and plain JSON String as inputs.

New test cases are added for reindexing with scripts.

Author: musabgelisgen

jest-common/src/main/java/io/searchbox/indices/reindex/Reindex.java

@@ -14,6 +14,7 @@ public class Reindex extends GenericResultAbstractAction {
 
     private Object source;
     private Object destination;
+    private Object script;
     Map<String, Object> body;
 
     Reindex(Builder builder) {
@@ -22,16 +23,14 @@ public class Reindex extends GenericResultAbstractAction {
 
         source = builder.source;
         destination = builder.dest;
+        script = builder.script;
 
         if (builder.conflicts != null) {
             body.put("conflicts", builder.conflicts);
         }
         if (builder.size != null) {
             body.put("size", builder.size);
         }
-        if (builder.script != null) {
-            body.put("script", builder.script);
-        }
     }
 
     @Override
@@ -61,6 +60,23 @@ public String getData(Gson gson) {
             }
         }
 
+
+        if (this.script != null) {
+            if (this.script instanceof String) {
+                this.body.put("script", gson.fromJson((String) this.script, Map.class));
+            } else if (this.script instanceof Map) {
+                Map<String, Object> script = new HashMap<String, Object>((Map) this.script);
+                Object params = script.get("params");
+                if (params instanceof String) {
+                    Map paramMap = gson.fromJson((String) params, Map.class);
+                    script.put("params", paramMap);
+                    body.put("script", script);
+                } else {
+                    body.put("script", this.script);
+                }
+            }
+        }
+
         return gson.toJson(this.body);
     }
 

jest/src/test/java/io/searchbox/indices/reindex/ReindexIntegrationTest.java

@@ -103,4 +103,40 @@ public void documentShouldNotBeTransferredWithQuery() throws IOException {
         assertTrue(result.getErrorMessage(), result.isSucceeded());
         assertFalse(indexExists(destIndex));
     }
+
+    @Test
+    public void testWithScriptString() throws IOException {
+        String sourceIndex = "my_source_index";
+        String destIndex = "my_dest_index";
+        String documentType = "my_type";
+        String documentId = "my_id";
+
+        createIndex(sourceIndex);
+        index(sourceIndex, documentType, documentId, "{\"user\": \"kimchy\"}");
+        flushAndRefresh(sourceIndex);
+
+        String scriptString = "{\n" +
+                "    \"lang\": \"painless\",\n" +
+                "    \"source\": \"ctx._source.last = params.last;\\nctx._source.nick = params.nick\",\n" +
+                "    \"params\": {\n" +
+                "      \"last\": \"gaudreau\",\n" +
+                "      \"nick\": \"hockey\"\n" +
+                "    }\n" +
+                "  }";
+
+
+        Map<String, Object> script = new HashMap<>();
+        script.put("source", scriptString);
+        script.put("lang", "painless");
+
+        ImmutableMap<String, Object> source = ImmutableMap.of("index", sourceIndex);
+        ImmutableMap<String, Object> dest = ImmutableMap.of("index", destIndex);
+
+        Reindex reindex = new Reindex.Builder(source, dest).script(scriptString).refresh(true).build();
+        JestResult result = client.execute(reindex);
+
+        assertTrue(result.getErrorMessage(), result.isSucceeded());
+        assertTrue(indexExists(destIndex));
+        assertTrue(get(destIndex, documentType, documentId).isExists());
+    }
 }