Merge "Add unit test cases."
[aai/sparky-be.git] / src / main / java / org / onap / aai / sparky / dal / elasticsearch / SearchAdapter.java
1 /**
2  * ============LICENSE_START=======================================================
3  * org.onap.aai
4  * ================================================================================
5  * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
6  * Copyright © 2017 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  * ECOMP is a trademark and service mark of AT&T Intellectual Property.
22  */
23 package org.onap.aai.sparky.dal.elasticsearch;
24
25 import java.util.Arrays;
26 import java.util.HashMap;
27 import java.util.List;
28 import java.util.Map;
29
30 import javax.ws.rs.core.MediaType;
31
32 import org.onap.aai.sparky.dal.rest.OperationResult;
33 import org.onap.aai.sparky.dal.sas.config.SearchServiceConfig;
34 import org.onap.aai.sparky.util.Encryptor;
35 import org.onap.aai.sparky.viewandinspect.config.TierSupportUiConstants;
36 import org.onap.aai.cl.api.Logger;
37 import org.onap.aai.cl.eelf.LoggerFactory;
38 import org.slf4j.MDC;
39
40 import org.onap.aai.restclient.client.RestClient;
41 import org.onap.aai.restclient.enums.RestAuthenticationMode;
42 import org.onap.aai.restclient.client.Headers;
43 import org.onap.aai.cl.mdc.MdcContext;
44
45 import org.onap.aai.cl.mdc.MdcContext;
46
47 /**
48  * The Class SearchAdapter.
49  */
50 public class SearchAdapter {
51
52   private static final Logger LOG = LoggerFactory.getInstance().getLogger(SearchAdapter.class);
53   
54   private RestClient client;
55
56   /**
57    * @return the client
58    */
59   public RestClient getClient() {
60     return client;
61   }
62
63   /**
64    * @param client the client to set
65    */
66   public void setClient(RestClient client) {
67     this.client = client;
68   }
69
70   /**
71    * @return the commonHeaders
72    */
73   public Map<String, List<String>> getCommonHeaders() {
74     return commonHeaders;
75   }
76
77   /**
78    * @param commonHeaders the commonHeaders to set
79    */
80   public void setCommonHeaders(Map<String, List<String>> commonHeaders) {
81     this.commonHeaders = commonHeaders;
82   }
83
84   /**
85    * @return the log
86    */
87   public static Logger getLog() {
88     return LOG;
89   }
90
91   private Map<String, List<String>> commonHeaders;
92   private SearchServiceConfig sasConfig;
93
94   /**
95    * Instantiates a new search adapter.
96    * @throws Exception 
97    */
98   public SearchAdapter() throws Exception {
99     sasConfig = SearchServiceConfig.getConfig();
100     Encryptor encryptor = new Encryptor();
101     client = new RestClient().authenticationMode(RestAuthenticationMode.SSL_CERT)
102         .validateServerHostname(false).validateServerCertChain(false)
103         .clientCertFile(TierSupportUiConstants.CONFIG_AUTH_LOCATION + sasConfig.getCertName())
104         .clientCertPassword(encryptor.decryptValue(sasConfig.getKeystorePassword()))
105         .trustStore(TierSupportUiConstants.CONFIG_AUTH_LOCATION + sasConfig.getKeystore());
106
107     commonHeaders = new HashMap<String, List<String>>();
108     commonHeaders.put("Accept", Arrays.asList("application/json"));
109     commonHeaders.put(Headers.FROM_APP_ID, Arrays.asList("AAI-UI"));
110   }
111
112   public SearchServiceConfig getSasConfig() {
113     return sasConfig;
114   }
115
116   public void setSasConfig(SearchServiceConfig sasConfig) {
117     this.sasConfig = sasConfig;
118   }
119
120   public OperationResult doPost(String url, String jsonPayload, String acceptContentType) {
121     org.onap.aai.restclient.client.OperationResult or = client.post(url, jsonPayload, getTxnHeader(),
122         MediaType.APPLICATION_JSON_TYPE, MediaType.APPLICATION_JSON_TYPE);
123     return new OperationResult(or.getResultCode(), or.getResult());
124   }
125
126   public OperationResult doGet(String url, String acceptContentType) {
127     org.onap.aai.restclient.client.OperationResult or =
128         client.get(url, getTxnHeader(), MediaType.APPLICATION_JSON_TYPE);
129     return new OperationResult(or.getResultCode(), or.getResult());
130   }
131
132   public OperationResult doPut(String url, String payload, String acceptContentType) {
133     org.onap.aai.restclient.client.OperationResult or = client.put(url, payload, getTxnHeader(),
134         MediaType.APPLICATION_JSON_TYPE, MediaType.APPLICATION_JSON_TYPE);
135     return new OperationResult(or.getResultCode(), or.getResult());
136   }
137
138   public OperationResult doDelete(String url, String acceptContentType) {
139
140     org.onap.aai.restclient.client.OperationResult or =
141         client.delete(url, getTxnHeader(), MediaType.APPLICATION_JSON_TYPE);
142     return new OperationResult(or.getResultCode(), or.getResult());
143   }
144
145   public Map<String, List<String>> getTxnHeader() {
146     Map headers = new HashMap<String, List<String>>();
147     headers.putAll(this.commonHeaders);
148     headers.put("X-TransactionId", Arrays.asList(MDC.get(MdcContext.MDC_REQUEST_ID)));
149     headers.put("X-FromAppId", Arrays.asList(MDC.get(MdcContext.MDC_PARTNER_NAME)));
150     return headers;
151   }
152
153
154 }