2 * ============LICENSE_START=======================================================
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
12 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
21 package org.onap.aai.sparky.search;
23 import java.util.Arrays;
24 import java.util.HashMap;
25 import java.util.List;
28 import javax.ws.rs.core.MediaType;
30 import org.onap.aai.cl.mdc.MdcContext;
31 import org.onap.aai.restclient.client.Headers;
32 import org.onap.aai.restclient.client.OperationResult;
33 import org.onap.aai.restclient.client.RestClient;
34 import org.onap.aai.sparky.dal.rest.RestClientFactory;
35 import org.onap.aai.sparky.dal.rest.config.RestEndpointConfig;
40 * The Class SearchServiceAdapter.
42 public class SearchServiceAdapter {
44 private static final String VALUE_QUERY = "query";
45 private static final String SUGGEST_QUERY = "suggest";
46 private static final String BULK_API = "bulk";
47 private static final String DOCUMENT_EDNPOINT = "documents";
48 private static final String SEARH_SERVICE_BULK_TEMPLATE =
49 "{\"create\":{\"metaData\":{\"url\":\"%s\"},\"document\":\"%s\"}}\n";
51 private static final String SEARH_SERVICE_SINGLE_ENTITY_TEMPLATE =
52 "{\"queries\":[{\"must\":{\"match\":{\"field\":\"_id\",\"value\":\"%s\"}}}]}\n";
54 private RestClient client;
55 private RestEndpointConfig endpointConfig;
56 private String serviceApiVersion;
58 private Map<String, List<String>> commonHeaders;
61 * Instantiates a new search adapter.
65 public SearchServiceAdapter(RestEndpointConfig endpointConfig, String serviceApiVersion)
68 client = RestClientFactory.buildClient(endpointConfig);
70 commonHeaders = new HashMap<String, List<String>>();
71 commonHeaders.put("Accept", Arrays.asList("application/json"));
72 commonHeaders.put(Headers.FROM_APP_ID, Arrays.asList("AAI-UI"));
74 this.serviceApiVersion = serviceApiVersion;
75 this.endpointConfig = endpointConfig;
78 public String getServiceApiVersion() {
79 return serviceApiVersion;
82 public void setServiceApiVersion(String serviceApiVersion) {
83 this.serviceApiVersion = serviceApiVersion;
86 public RestEndpointConfig getEndpointConfig() {
87 return endpointConfig;
90 public void setEndpointConfig(RestEndpointConfig endpointConfig) {
91 this.endpointConfig = endpointConfig;
94 public OperationResult doPost(String url, String jsonPayload) {
95 OperationResult or = client.post(url, jsonPayload, getTxnHeader(),
96 MediaType.APPLICATION_JSON_TYPE, MediaType.APPLICATION_JSON_TYPE);
97 return new OperationResult(or.getResultCode(), or.getResult());
101 public OperationResult doPost(String url, String jsonPayload, String acceptContentType) {
102 OperationResult or = client.post(url, jsonPayload, getTxnHeader(),
103 MediaType.APPLICATION_JSON_TYPE, MediaType.APPLICATION_JSON_TYPE);
104 return new OperationResult(or.getResultCode(), or.getResult());
107 public OperationResult doGet(String url, String acceptContentType) {
108 OperationResult or = client.get(url, getTxnHeader(), MediaType.APPLICATION_JSON_TYPE);
109 return new OperationResult(or.getResultCode(), or.getResult());
112 public OperationResult doPut(String url, String payload, String acceptContentType) {
113 OperationResult or = client.put(url, payload, getTxnHeader(), MediaType.APPLICATION_JSON_TYPE,
114 MediaType.APPLICATION_JSON_TYPE);
115 return new OperationResult(or.getResultCode(), or.getResult());
118 public OperationResult doDelete(String url, String acceptContentType) {
120 OperationResult or = client.delete(url, getTxnHeader(), MediaType.APPLICATION_JSON_TYPE);
121 return new OperationResult(or.getResultCode(), or.getResult());
124 public OperationResult doBulkOperation(String url, String jsonPayload) {
126 OperationResult or = client.post(url, jsonPayload, getTxnHeader(),
127 MediaType.APPLICATION_JSON_TYPE, MediaType.APPLICATION_JSON_TYPE);
128 return new OperationResult(or.getResultCode(), or.getResult());
131 public Map<String, List<String>> getTxnHeader() {
132 HashMap<String, List<String>> headers = new HashMap<String, List<String>>();
133 headers.putAll(this.commonHeaders);
134 headers.put("X-TransactionId", Arrays.asList(MDC.get(MdcContext.MDC_REQUEST_ID)));
135 headers.put("X-FromAppId", Arrays.asList(MDC.get(MdcContext.MDC_PARTNER_NAME)));
139 public String buildBulkImportOperationRequest(String indexName, String id, String payload){
141 StringBuilder requestPayload = new StringBuilder(128);
142 String SearchTarget = buildSearchServiceDocUrl(indexName,id);
144 requestPayload.append(String.format(SEARH_SERVICE_BULK_TEMPLATE,SearchTarget,payload));
145 return requestPayload.toString();
149 * Get Full URL for search
153 * @return the full url
155 public String buildSearchServiceQueryUrl(String indexName) {
156 return buildSearchServiceUrlForApi(indexName, VALUE_QUERY);
160 * Get Full URL for search
164 * @return the full url
166 public String buildSuggestServiceQueryUrl(String indexName) {
167 return buildSearchServiceUrlForApi(indexName, SUGGEST_QUERY);
170 public String buildSearchServiceDocUrl(String indexName,String api) {
172 return String.format("https://%s:%s/services/search-data-service/%s/search/indexes/%s/%s/%s",
173 endpointConfig.getEndpointIpAddress(), endpointConfig.getEndpointServerPort(),
174 serviceApiVersion, indexName,DOCUMENT_EDNPOINT, api);
178 public String buildSearchServiceCreateDocApi(String indexName){
180 return String.format("https://%s:%s/services/search-data-service/%s/search/indexes/%s/%s",
181 endpointConfig.getEndpointIpAddress(), endpointConfig.getEndpointServerPort(),
182 serviceApiVersion, indexName,DOCUMENT_EDNPOINT );
185 public String buildSearchServiceUrlForApi(String indexName, String api) {
187 return String.format("https://%s:%s/services/search-data-service/%s/search/indexes/%s/%s",
188 endpointConfig.getEndpointIpAddress(), endpointConfig.getEndpointServerPort(),
189 serviceApiVersion, indexName, api);
192 public String buildSearchServiceBulkUrl() {
194 return String.format("https://%s:%s/services/search-data-service/%s/search/%s", endpointConfig.getEndpointIpAddress(),
195 endpointConfig.getEndpointServerPort(),serviceApiVersion,BULK_API);
198 public OperationResult retrieveEntityById(String entityId,String indexName) {
200 StringBuilder requestPayload = new StringBuilder(128);
201 requestPayload.append(String.format(SEARH_SERVICE_SINGLE_ENTITY_TEMPLATE,entityId));
202 String payload = requestPayload.toString();
203 String searchServiceUrl = buildSearchServiceQueryUrl(indexName);
205 return this.doPost(searchServiceUrl,payload);
208 public String buildSearchServiceCreateIndexUrl(String indexName) {
210 return String.format("https://%s:%s/services/search-data-service/%s/search/indexes/dynamic/%s", endpointConfig.getEndpointIpAddress(),
211 endpointConfig.getEndpointServerPort(),serviceApiVersion,indexName);