Initial search service commit
[aai/search-data-service.git] / src / main / java / org / openecomp / sa / searchdbabstraction / searchapi / QueryStatement.java
diff --git a/src/main/java/org/openecomp/sa/searchdbabstraction/searchapi/QueryStatement.java b/src/main/java/org/openecomp/sa/searchdbabstraction/searchapi/QueryStatement.java
new file mode 100644 (file)
index 0000000..f5fc367
--- /dev/null
@@ -0,0 +1,142 @@
+/**
+ * ============LICENSE_START=======================================================
+ * Search Data Service
+ * ================================================================================
+ * Copyright © 2017 AT&T Intellectual Property.
+ * Copyright © 2017 Amdocs
+ * All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License ati
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ *
+ * ECOMP and OpenECOMP are trademarks
+ * and service marks of AT&T Intellectual Property.
+ */
+package org.openecomp.sa.searchdbabstraction.searchapi;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+public class QueryStatement {
+
+  private TermQuery match;
+
+  @JsonProperty("not-match")
+  private TermQuery notMatch;
+
+  @JsonProperty("parsed-query")
+  private ParsedQuery parsedQuery;
+
+  private RangeQuery range;
+
+  public TermQuery getMatch() {
+    return match;
+  }
+
+  public void setMatch(TermQuery match) {
+    this.match = match;
+  }
+
+  public TermQuery getNotMatch() {
+    return notMatch;
+  }
+
+  public void setNotMatch(TermQuery notMatch) {
+    this.notMatch = notMatch;
+  }
+
+  public ParsedQuery getParsedQuery() {
+    return parsedQuery;
+  }
+
+  public void setParsedQuery(ParsedQuery parsedQuery) {
+    this.parsedQuery = parsedQuery;
+  }
+
+  public RangeQuery getRange() {
+    return range;
+  }
+
+  public void setRange(RangeQuery range) {
+    this.range = range;
+  }
+
+  public boolean isNotMatch() {
+    return (notMatch != null);
+  }
+
+  public String toElasticSearch() {
+
+    if (match != null) {
+      return match.toElasticSearch();
+
+    } else if (notMatch != null) {
+      return notMatch.toElasticSearch();
+
+    } else if (parsedQuery != null) {
+
+      // We need some special wrapping if this query is against a nested field.
+      if (fieldIsNested(parsedQuery.getField())) {
+        return "{\"nested\": { \"path\": \"" + pathForNestedField(parsedQuery.getField())
+            + "\", \"query\": " + parsedQuery.toElasticSearch() + "}}";
+      } else {
+        return parsedQuery.toElasticSearch();
+      }
+
+    } else if (range != null) {
+
+      // We need some special wrapping if this query is against a nested field.
+      if (fieldIsNested(range.getField())) {
+        return "{\"nested\": { \"path\": \"" + pathForNestedField(range.getField())
+            + "\", \"query\": " + range.toElasticSearch() + "}}";
+      } else {
+        return range.toElasticSearch();
+      }
+
+    } else {
+      // throw an exception?
+      return null;
+    }
+  }
+
+  private boolean fieldIsNested(String field) {
+    return field.contains(".");
+  }
+
+  private String pathForNestedField(String field) {
+    int index = field.lastIndexOf('.');
+    return field.substring(0, index);
+  }
+
+  @Override
+  public String toString() {
+
+    StringBuilder sb = new StringBuilder();
+
+    sb.append("{");
+
+    if (match != null) {
+      sb.append("TERM QUERY: { match: {").append(match.toString()).append("}}");
+    } else if (notMatch != null) {
+      sb.append("TERM QUERY: { not-match: {").append(match.toString()).append("}}");
+    } else if (parsedQuery != null) {
+      sb.append("PARSED QUERY: { ").append(parsedQuery.toString()).append("}");
+    } else if (range != null) {
+      sb.append("RANGE QUERY: { ").append(range.toString()).append("}");
+    } else {
+      sb.append("UNDEFINED");
+    }
+
+    sb.append("}");
+    return sb.toString();
+  }
+}