src/main/java/org/onap/aai/sa/searchdbabstraction/searchapi/Query.java

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * org.onap.aai
   4  * ================================================================================
   5  * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
   6  * Copyright © 2017-2018 Amdocs
   7  * ================================================================================
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  *       http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  * ============LICENSE_END=========================================================
  20  */
  21 package org.onap.aai.sa.searchdbabstraction.searchapi;
  22
  23 public class Query {
  24
  25   private QueryStatement may;
  26   private QueryStatement must;
  27
  28   public QueryStatement getMay() {
  29     return may;
  30   }
  31
  32   public void setMay(QueryStatement may) {
  33     this.may = may;
  34   }
  35
  36   public QueryStatement getMust() {
  37     return must;
  38   }
  39
  40   public void setMust(QueryStatement must) {
  41     this.must = must;
  42   }
  43
  44   public QueryStatement getQueryStatement() {
  45     if (isMust()) {
  46       return must;
  47     } else if (isMay()) {
  48       return may;
  49     } else {
  50       return null;
  51     }
  52   }
  53
  54   public boolean isMust() {
  55     return must != null;
  56   }
  57
  58   public boolean isMay() {
  59     return may != null;
  60   }
  61
  62   public String toElasticSearch() {
  63
  64     if (isMust()) {
  65       return must.toElasticSearch();
  66     } else if (isMay()) {
  67       return may.toElasticSearch();
  68     } else {
  69       return ""; // throw an exception?
  70     }
  71   }
  72
  73   @Override
  74   public String toString() {
  75
  76     StringBuilder sb = new StringBuilder();
  77
  78     sb.append("Query:[");
  79     if (isMust()) {
  80       sb.append("must: ").append(must.toString());
  81     } else if (isMay()) {
  82       sb.append("may: ").append(may.toString());
  83     } else {
  84       sb.append("INVALID");
  85     }
  86     sb.append("]");
  87
  88     return sb.toString();
  89   }
  90 }