DDP-7443 | implement ScriptBuilder and two child classes which will c…

…reate scripts for single and nested types

src/main/java/org/broadinstitute/dsm/db/KitRequestShipping.java

@@ -18,6 +18,9 @@
 import org.broadinstitute.dsm.model.*;
 import org.broadinstitute.dsm.model.ddp.DDPParticipant;
 import org.broadinstitute.dsm.model.ddp.KitDetail;
+import org.broadinstitute.dsm.model.elastic.export.generate.Generator;
+import org.broadinstitute.dsm.model.elastic.export.painless.ParamsGenerator;
+import org.broadinstitute.dsm.model.elastic.export.painless.UpsertPainless;
 import org.broadinstitute.dsm.statics.ApplicationConfigConstants;
 import org.broadinstitute.dsm.statics.DBConstants;
 import org.broadinstitute.dsm.statics.QueryExtension;
@@ -868,6 +871,9 @@ public static String writeRequest(@NonNull String instanceId, @NonNull String dd
             throw new RuntimeException("Error adding kit request  w/ ddpKitRequestId " + ddpKitRequestId, results.resultException);
         }
 
+        new ParamsGenerator("", )
+        new UpsertPainless()
+
         logger.info("Added kitRequest w/ ddpKitRequestId " + ddpKitRequestId);
         return (String) results.resultValue;
     }

src/main/java/org/broadinstitute/dsm/model/elastic/Util.java

@@ -13,6 +13,7 @@
 import org.broadinstitute.dsm.db.*;
 import org.broadinstitute.dsm.db.structure.ColumnName;
 import org.broadinstitute.dsm.db.structure.DBElement;
+import org.broadinstitute.dsm.db.structure.TableName;
 import org.broadinstitute.dsm.model.elastic.export.generate.BaseGenerator;
 import org.broadinstitute.dsm.model.elastic.export.parse.DynamicFieldsParser;
 import org.broadinstitute.dsm.model.elastic.export.parse.ValueParser;
@@ -87,6 +88,11 @@ private static String handleAllUppercase(String word) {
         return CAMEL_CASE_REGEX.matcher(word).matches() ? word : word.toLowerCase();
     }
 
+    public static String getPrimaryKeyFromClass(Class<?> clazz) {
+        TableName tableName = Objects.requireNonNull(clazz.getAnnotation(TableName.class));
+        return underscoresToCamelCase(tableName.primaryKey());
+    }
+
     private static boolean hasNoUnderscores(String[] splittedWords) {
         return splittedWords.length < 2;
     }

src/main/java/org/broadinstitute/dsm/model/elastic/export/generate/BaseGenerator.java

@@ -132,8 +132,7 @@ public String getPropertyName() {
         }
 
         public String getPrimaryKey() {
-            TableName tableName = Objects.requireNonNull(propertyClass.getAnnotation(TableName.class));
-            return Util.underscoresToCamelCase(tableName.primaryKey());
+            return Util.getPrimaryKeyFromClass(propertyClass);
         }
 
         public boolean isCollection() {

src/main/java/org/broadinstitute/dsm/model/elastic/export/generate/Generator.java

@@ -8,4 +8,8 @@ public interface Generator {
     default String getPropertyName() {
         throw new UnsupportedOperationException();
     }
+
+    default String getUniqueIdentifier() {
+        throw new UnsupportedOperationException();
+    }
 }

src/main/java/org/broadinstitute/dsm/model/elastic/export/painless/NestedScriptBuilder.java

@@ -0,0 +1,22 @@
+package org.broadinstitute.dsm.model.elastic.export.painless;
+
+public class NestedScriptBuilder implements ScriptBuilder {
+
+    private final static String SCRIPT = "if (ctx._source.dsm.#propertyName == null) {ctx._source.dsm.#propertyName = [params.dsm.#propertyName]} " +
+            "else {def targets = ctx._source.dsm.#propertyName.findAll(obj -> obj.#uniqueIdentifier == params.dsm.#propertyName.#uniqueIdentifier); " +
+            "for(target in targets) { for (entry in params.dsm.#propertyName.entrySet()) { target.put(entry.getKey(), entry.getValue()) } }}";
+
+    private final String propertyName;
+    private final String uniqueIdentifier;
+
+    public NestedScriptBuilder(String propertyName, String uniqueIdentifier) {
+        this.propertyName = propertyName;
+        this.uniqueIdentifier = uniqueIdentifier;
+    }
+
+    @Override
+    public String build() {
+        return SCRIPT.replace("#propertyName", propertyName)
+                .replace("#uniqueIdentifier", uniqueIdentifier);
+    }
+}

src/main/java/org/broadinstitute/dsm/model/elastic/export/painless/ParamsGenerator.java

@@ -30,4 +30,9 @@ public Map<String, Object> generate() {
     public String getPropertyName() {
         return Util.capitalCamelCaseToLowerCamelCase(source.getClass().getSimpleName());
     }
+
+    @Override
+    public String getUniqueIdentifier() {
+        return Util.getPrimaryKeyFromClass(source.getClass());
+    }
 }

src/main/java/org/broadinstitute/dsm/model/elastic/export/painless/ScriptBuilder.java

@@ -0,0 +1,5 @@
+package org.broadinstitute.dsm.model.elastic.export.painless;
+
+public interface ScriptBuilder {
+    String build();
+}

src/main/java/org/broadinstitute/dsm/model/elastic/export/painless/SingleScriptBuilder.java

@@ -0,0 +1,19 @@
+package org.broadinstitute.dsm.model.elastic.export.painless;
+
+public class SingleScriptBuilder implements ScriptBuilder {
+
+    private final static String SCRIPT = "if (ctx._source.dsm.#propertyName == null) " +
+            "{ctx._source.dsm.#propertyName = params.dsm.#propertyName} " +
+            "else {for (entry in params.dsm.#propertyName.entrySet()) { ctx._source.dsm.#propertyName.put(entry.getKey(), entry.getValue()) }}";
+
+    private final String propertyName;
+
+    public SingleScriptBuilder(String propertyName) {
+        this.propertyName = propertyName;
+    }
+
+    @Override
+    public String build() {
+        return SCRIPT.replace("#propertyName", propertyName);
+    }
+}

src/main/java/org/broadinstitute/dsm/model/elastic/export/painless/BaseUpsertPainless.java → src/main/java/org/broadinstitute/dsm/model/elastic/export/painless/UpsertPainless.java

@@ -13,23 +13,23 @@
 import org.elasticsearch.script.Script;
 import org.elasticsearch.script.ScriptType;
 
-public abstract class BaseUpsertPainless implements Exportable {
+public class UpsertPainless implements Exportable {
 
-    protected String script;
     private Generator generator;
     private RequestPayload requestPayload; // index, docId
+    private ScriptBuilder scriptBuilder;
 
-    public BaseUpsertPainless(String script, Generator generator, RequestPayload requestPayload) {
-        this.script = script;
+    public UpsertPainless(Generator generator, RequestPayload requestPayload, ScriptBuilder scriptBuilder) {
         this.generator = generator;
         this.requestPayload = requestPayload;
+        this.scriptBuilder = scriptBuilder;
     }
 
     @Override
     public void export() {
         RestHighLevelClient clientInstance = ElasticSearchUtil.getClientInstance();
         UpdateRequest updateRequest = new UpdateRequest(requestPayload.getIndex(), Util.DOC, requestPayload.getDocId());
-        Script painless = new Script(ScriptType.INLINE, "painless", script, generator.generate());
+        Script painless = new Script(ScriptType.INLINE, "painless", scriptBuilder.build(), generator.generate());
         updateRequest.script(painless);
         try {
             clientInstance.update(updateRequest, RequestOptions.DEFAULT);

src/test/java/org/broadinstitute/dsm/model/elastic/export/painless/ScriptBuilderTest.java

@@ -0,0 +1,43 @@
+package org.broadinstitute.dsm.model.elastic.export.painless;
+
+import org.junit.Assert;
+import org.junit.Test;
+
+public class ScriptBuilderTest {
+
+    @Test
+    public void buildNested() {
+
+        String propertyName = "kitRequestShipping";
+        String uniqueIdentifier = "kitRequestId";
+
+        ScriptBuilder builder = new NestedScriptBuilder(propertyName, uniqueIdentifier);
+        String script = builder.build();
+
+        String expectedScript = "if (ctx._source.dsm.kitRequestShipping == null) {ctx._source.dsm.kitRequestShipping = [params.dsm.kitRequestShipping]} " +
+                "else {def targets = ctx._source.dsm.kitRequestShipping.findAll(obj -> obj.kitRequestId == params.dsm.kitRequestShipping.kitRequestId); " +
+                "for(target in targets) { for (entry in params.dsm.kitRequestShipping.entrySet()) { target.put(entry.getKey(), entry.getValue()) } " +
+                "}}";
+
+        Assert.assertEquals(expectedScript, script);
+    }
+
+    @Test
+    public void buildSingle() {
+
+        String propertyName = "kitRequestShipping";
+
+        ScriptBuilder builder = new SingleScriptBuilder(propertyName);
+        String script = builder.build();
+
+        String expectedScript = "" +
+                "if (ctx._source.dsm.kitRequestShipping == null) " +
+                "{ctx._source.dsm.kitRequestShipping = params.dsm.kitRequestShipping} " +
+                "else {" +
+                "for (entry in params.dsm.kitRequestShipping.entrySet()) " +
+                "{ ctx._source.dsm.kitRequestShipping.put(entry.getKey(), entry.getValue()) }" +
+                "}";
+
+        Assert.assertEquals(expectedScript, script);
+    }
+}