gerrit.onap Code Review - ccsdk/features.git/blob

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * ONAP : ccsdk features
   4  * ================================================================================
   5  * Copyright (C) 2019 highstreet technologies GmbH Intellectual Property.
   6  * All rights reserved.
   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  */
  22 package org.onap.ccsdk.features.sdnr.wt.common.database.queries;
  23
  24 import org.json.JSONArray;
  25 import org.json.JSONObject;
  26
  27 public class BoolQueryBuilder extends QueryBuilder {
  28
  29     private JSONObject inner;
  30
  31     public BoolQueryBuilder() {
  32         super();
  33         this.inner = new JSONObject();
  34         this.setQuery("bool", this.inner);
  35     }
  36
  37     @Override
  38     public String toString() {
  39         return "BoolQueryBuilder [inner=" + inner + "]";
  40     }
  41
  42     public static boolean isEmpty(JSONObject o) {
  43         return o.keySet().isEmpty();
  44     }
  45
  46     public BoolQueryBuilder must(QueryBuilder query) {
  47
  48         if (!isEmpty(this.inner)) {
  49             Object x = this.inner.has("must") ? this.inner.get("must") : this.inner;
  50             if (x instanceof JSONArray) {
  51                 ((JSONArray) x).put(query.getInner());
  52             } else {
  53                 this.inner = new JSONObject();
  54                 this.inner.put("must", new JSONObject());
  55                 JSONArray a = new JSONArray();
  56                 a.put(x);
  57                 a.put(query.getInner());
  58                 this.inner.put("must", a);
  59             }
  60         } else {
  61             this.inner.put("must", query.getInner());
  62         }
  63         this.setQuery("bool", this.inner);
  64         return this;
  65     }
  66
  67     /**
  68      * @param query
  69      */
  70     public BoolQueryBuilder should(QueryBuilder query) {
  71         if (!isEmpty(this.inner)) {
  72             Object x = this.inner.has("should") ? this.inner.get("should") : this.inner;
  73             if (x instanceof JSONArray) {
  74                 ((JSONArray) x).put(query.getInner());
  75             } else {
  76                 this.inner = new JSONObject();
  77                 //                              this.inner.put("should", new JSONObject());
  78                 JSONArray a = new JSONArray();
  79                 if (!x.toString().equals("{}")) {
  80                     a.put(x);
  81                 }
  82                 a.put(query.getInner());
  83                 this.inner.put("should", a);
  84             }
  85         } else {
  86             this.inner.put("should", query.getInner());
  87         }
  88         this.setQuery("bool", this.inner);
  89         return this;
  90     }
  91
  92 }