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

   1 /*******************************************************************************
   2  * ============LICENSE_START========================================================================
   3  * ONAP : ccsdk feature sdnr wt
   4  * =================================================================================================
   5  * Copyright (C) 2019 highstreet technologies GmbH Intellectual Property. All rights reserved.
   6  * =================================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
   8  * in compliance with the License. You may obtain a copy of the License at
   9  *
  10  * http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software distributed under the License
  13  * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
  14  * or implied. See the License for the specific language governing permissions and limitations under
  15  * the License.
  16  * ============LICENSE_END==========================================================================
  17  ******************************************************************************/
  18 package org.onap.ccsdk.features.sdnr.wt.common.database.responses;
  19
  20 import org.elasticsearch.client.Response;
  21 import org.json.JSONArray;
  22 import org.json.JSONObject;
  23 import org.onap.ccsdk.features.sdnr.wt.common.database.SearchHit;
  24
  25 public class SearchResponse extends BaseResponse {
  26
  27         private long total;
  28         private SearchHit[] searchHits;
  29         private JSONObject aggregations;
  30
  31         public SearchResponse(Response response) {
  32                 super(response);
  33                 this.handleResult(this.getJson(response));
  34         }
  35
  36         public SearchResponse(String json) {
  37                 super(null);
  38                 this.handleResult(this.getJson(json));
  39         }
  40
  41         private void handleResult(JSONObject result) {
  42                 if(result!=null && this.isResponseSucceeded()) {
  43                         JSONObject hitsouter=result.getJSONObject("hits");
  44                         this.total = hitsouter.getLong("total");
  45                         JSONArray a=hitsouter.getJSONArray("hits");
  46                         SearchHit[] hits=new SearchHit[a.length()];
  47                         for(int i=0;i<a.length();i++) {
  48                                 hits[i]=new SearchHit(a.getJSONObject(i));
  49                         }
  50                         this.searchHits=hits;
  51                         if(result.has("aggregations")) {
  52                                 this.aggregations = result.getJSONObject("aggregations");
  53                         }
  54                         else {
  55                                 this.aggregations=null;
  56                         }
  57                 }
  58                 else {
  59                         this.searchHits=new SearchHit[0];
  60                 }
  61         }
  62         public SearchHit[] getHits() {
  63                 return this.searchHits;
  64         }
  65         public long getTotal() {
  66                 return this.total;
  67         }
  68         public boolean hasAggregations() {
  69                 return this.aggregations!=null;
  70         }
  71         public AggregationEntries getAggregations(String property) {
  72                 AggregationEntries entries=new AggregationEntries();
  73                 if(this.aggregations!=null && this.aggregations.has(property)) {
  74                         JSONArray a=this.aggregations.getJSONObject(property).getJSONArray("buckets");
  75                         for(int i=0;i<a.length();i++) {
  76                                 entries.put(a.getJSONObject(i).getString("key"),a.getJSONObject(i).getLong("doc_count") );
  77                         }
  78                 }
  79                 return entries;
  80         }
  81 }