Optic enhancements #1282

Author: ehennum

marklogic-client-api/src/main/java/com/marklogic/client/expression/PlanBuilder.java

@@ -16,10 +16,12 @@
 package com.marklogic.client.impl;
 
 import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.databind.JsonNode;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.fasterxml.jackson.databind.node.ArrayNode;
 import com.fasterxml.jackson.databind.node.ObjectNode;
+import com.marklogic.client.MarkLogicIOException;
 import com.marklogic.client.io.*;
 import com.marklogic.client.io.marker.*;
 import org.w3c.dom.Document;
@@ -595,6 +597,43 @@ static public String ReaderToString(Reader reader) {
       }
    }
 
+   static public <T extends JSONReadHandle> T jsonNodeToHandle(JsonNode node, T handle) {
+      if (node == null) {
+         return null;
+      } else if (handle == null) {
+         throw new IllegalArgumentException("cannot convert JSON node to null handle");
+      }
+
+      try {
+         HandleImplementation handleImpl = HandleAccessor.checkHandle(handle, "json");
+
+         Class<?> as = handleImpl.receiveAs();
+         if (as == null) {
+            throw new IllegalArgumentException("handle does not specify class to receive content");
+         } else if (byte[].class.isAssignableFrom(as)) {
+            handleImpl.receiveContent(getMapper().writeValueAsBytes(node));
+         } else if (File.class.isAssignableFrom(as)) {
+            Path tempFile = Files.createTempFile("tmp", ".json");
+            Files.copy(new ByteArrayInputStream(getMapper().writeValueAsBytes(node)), tempFile, StandardCopyOption.REPLACE_EXISTING);
+            handleImpl.receiveContent(tempFile.toFile());
+// TODO: JsonParser.class.isAssignableFrom(as)
+         } else if (InputStream.class.isAssignableFrom(as)) {
+            handleImpl.receiveContent(new ByteArrayInputStream(getMapper().writeValueAsBytes(node)));
+         } else if (Reader.class.isAssignableFrom(as)) {
+            handleImpl.receiveContent(new StringReader(getMapper().writeValueAsString(node)));
+         } else if (String.class.isAssignableFrom(as)) {
+            handleImpl.receiveContent(getMapper().writeValueAsString(node));
+         } else {
+            throw new IllegalArgumentException("handle receives content with unsupported class: "+as.getSimpleName());
+         }
+
+         return handle;
+      } catch (JsonProcessingException e) {
+         throw new MarkLogicIOException("could not set handle to JsonNode", e);
+      } catch (IOException e) {
+         throw new MarkLogicIOException("could not create file for JsonNode", e);
+      }
+   }
    static public JsonNode handleToJsonNode(JSONWriteHandle jsonHandle) {
       if (jsonHandle == null) {
          return null;

marklogic-client-api/src/main/java/com/marklogic/client/expression/PlanBuilderBase.java

@@ -15,6 +15,7 @@
  */
 package com.marklogic.client.impl;
 
+import java.util.HashMap;
 import java.util.Map;
 
 import com.marklogic.client.DatabaseClientFactory.HandleFactoryRegistry;
@@ -24,6 +25,8 @@
 import com.marklogic.client.io.marker.AbstractWriteHandle;
 import com.marklogic.client.type.PlanColumn;
 import com.marklogic.client.type.PlanParamExpr;
+import com.marklogic.client.type.PlanSearchOptions;
+import com.marklogic.client.type.XsDoubleVal;
 
 abstract class PlanBuilderBaseImpl extends PlanBuilder {
   final static PlanBuilderSubImpl pb = new PlanBuilderSubImpl();
@@ -53,6 +56,57 @@ static BaseTypeImpl.Literal literal(Object value) {
     return new BaseTypeImpl.Literal(value);
   }
 
+  public PlanSearchOptions searchOptions() {
+    return new PlanSearchOptionsImpl(this);
+  }
+
+  static class PlanSearchOptionsImpl implements PlanSearchOptions {
+    private PlanBuilderBaseImpl pb;
+    private XsDoubleVal qualityWeight;
+    private ScoreMethod scoreMethod;
+    PlanSearchOptionsImpl(PlanBuilderBaseImpl pb) {
+      this.pb = pb;
+    }
+    PlanSearchOptionsImpl(PlanBuilderBaseImpl pb, XsDoubleVal qualityWeight, ScoreMethod scoreMethod) {
+      this(pb);
+      this.qualityWeight = qualityWeight;
+      this.scoreMethod   = scoreMethod;
+    }
+
+    @Override
+    public XsDoubleVal getQualityWeight() {
+      return qualityWeight;
+    }
+    @Override
+    public ScoreMethod getScoreMethod() {
+      return scoreMethod;
+    }
+    @Override
+    public PlanSearchOptions withQualityWeight(double qualityWeight) {
+      return withQualityWeight(pb.xs.doubleVal(qualityWeight));
+    }
+    @Override
+    public PlanSearchOptions withQualityWeight(XsDoubleVal qualityWeight) {
+      return new PlanSearchOptionsImpl(pb, qualityWeight, getScoreMethod());
+    }
+    @Override
+    public PlanSearchOptions withScoreMethod(ScoreMethod scoreMethod) {
+      return new PlanSearchOptionsImpl(pb, getQualityWeight(), scoreMethod);
+    }
+    Map<String,String> makeMap() {
+      if (qualityWeight == null && scoreMethod == null) return null;
+
+      Map<String, String> map = new HashMap<String, String>();
+      if (qualityWeight != null) {
+        map.put("qualityWeight", String.valueOf(qualityWeight));
+      }
+      if (scoreMethod != null) {
+        map.put("scoreMethod", scoreMethod.name().toLowerCase());
+      }
+      return map;
+    }
+  }
+
   static class PlanParamBase extends BaseTypeImpl.BaseCallImpl<XsValueImpl.StringValImpl> implements PlanParamExpr {
     String name = null;
     PlanParamBase(String name) {
@@ -121,6 +175,9 @@ String getAst() {
     public PlanColumn col(String column) {
       return pb.col(column);
     }
+    public PlanColumn exprCol(String column) {
+      return pb.exprCol(column);
+    }
   }
 
   static class PlanCallImpl extends BaseTypeImpl.ServerExpressionCallImpl {