summaryrefslogtreecommitdiffstats
path: root/sparkybe-onap-service/src/main/java/org/onap/aai/sparky/aggregatevnf/search/AggregateVnfSearchProvider.java
blob: 6e7b456a215242f3ac859df708c702b0bc72f3d8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/**
 * ============LICENSE_START===================================================
 * SPARKY (AAI UI 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 at
 *
 *      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.onap.aai.sparky.aggregatevnf.search;

import java.util.ArrayList;
import java.util.List;

import javax.json.JsonObject;
import javax.ws.rs.core.MediaType;

import org.json.JSONArray;
import org.json.JSONObject;
import org.onap.aai.cl.api.Logger;
import org.onap.aai.cl.eelf.LoggerFactory;
import org.onap.aai.restclient.client.OperationResult;
import org.onap.aai.sparky.common.search.CommonSearchSuggestion;
import org.onap.aai.sparky.dal.ElasticSearchAdapter;
import org.onap.aai.sparky.logging.AaiUiMsgs;
import org.onap.aai.sparky.search.api.SearchProvider;
import org.onap.aai.sparky.search.entity.QuerySearchEntity;
import org.onap.aai.sparky.search.entity.SearchSuggestion;
import org.onap.aai.sparky.search.filters.entity.UiFilterValueEntity;
import org.onap.aai.sparky.util.NodeUtils;
import org.onap.aai.sparky.viewandinspect.config.SparkyConstants;

import com.fasterxml.jackson.databind.ObjectMapper;

public class AggregateVnfSearchProvider implements SearchProvider {
  
  private static final Logger LOG = LoggerFactory.getInstance().getLogger(AggregateVnfSearchProvider.class);

  private ObjectMapper mapper;
  private ElasticSearchAdapter elasticSearchAdapter = null;
  private String autoSuggestIndexName;
  private String vnfSearchSuggestionRoute;

  public AggregateVnfSearchProvider(ElasticSearchAdapter elasticSearchAdapter,
      String autoSuggestIndexName, String vnfSearchSuggestionRoute) {
    mapper = new ObjectMapper();
    this.elasticSearchAdapter = elasticSearchAdapter;
    this.autoSuggestIndexName = autoSuggestIndexName;
    this.vnfSearchSuggestionRoute = vnfSearchSuggestionRoute;
  }
  
  public void setAutoSuggestIndexName(String autoSuggestIndexName) {
    this.autoSuggestIndexName = autoSuggestIndexName;
  }

  @Override
  public List<SearchSuggestion> search(QuerySearchEntity queryRequest) {

    List<SearchSuggestion> returnList = new ArrayList<SearchSuggestion>();

    try {

      /* Create suggestions query */
      JsonObject vnfSearch = VnfSearchQueryBuilder.createSuggestionsQuery(String.valueOf(queryRequest.getMaxResults()), queryRequest.getQueryStr());

      /* Parse suggestions response */
      OperationResult opResult = elasticSearchAdapter.doPost(
          elasticSearchAdapter.buildElasticSearchUrlForApi(autoSuggestIndexName,
              SparkyConstants.ES_SUGGEST_API),
          vnfSearch.toString(), MediaType.APPLICATION_JSON_TYPE);

      String result = opResult.getResult();

      if (!opResult.wasSuccessful()) {
        LOG.error(AaiUiMsgs.ERROR_PARSING_JSON_PAYLOAD_VERBOSE, result);
        return returnList;
      }

      JSONObject responseJson = new JSONObject(result);
      String suggestionsKey = "vnfs";
      JSONArray suggestionsArray = new JSONArray();
      JSONArray suggestions = responseJson.getJSONArray(suggestionsKey);
      if (suggestions.length() > 0) {
        suggestionsArray = suggestions.getJSONObject(0).getJSONArray("options");
        for (int i = 0; i < suggestionsArray.length(); i++) {
          JSONObject querySuggestion = suggestionsArray.getJSONObject(i);
          if (querySuggestion != null) {
            CommonSearchSuggestion responseSuggestion = new CommonSearchSuggestion();
            responseSuggestion.setText(querySuggestion.getString("text"));
            responseSuggestion.setRoute(vnfSearchSuggestionRoute);
            responseSuggestion.setHashId(NodeUtils.generateUniqueShaDigest(querySuggestion.getString("text")));

            // Extract filter list from JSON and add to response suggestion
            JSONObject payload = querySuggestion.getJSONObject("payload");
            if (payload.length() > 0) {
              JSONArray filterList = payload.getJSONArray("filterList");
              for (int filter = 0; filter < filterList.length(); filter++) {
                String filterValueString = filterList.getJSONObject(filter).toString();
                UiFilterValueEntity filterValue = mapper.readValue(filterValueString, UiFilterValueEntity.class);
                responseSuggestion.getFilterValues().add(filterValue);
              }
            }
            returnList.add(responseSuggestion);
          }
        }
      }
    } catch (Exception exc) {
      LOG.error(AaiUiMsgs.ERROR_GENERIC, "Search failed due to error = " + exc.getMessage());
    }

    return returnList;
  }
  
}