/*-
* ============LICENSE_START=======================================================
* OPENECOMP - MSO
* ================================================================================
* Copyright (C) 2017 AT&T Intellectual Property. 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=========================================================
*/
package org.openecomp.mso.bpmn.core;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Base64;
import java.util.Map;
import java.util.UUID;
import javax.ws.rs.GET;
import javax.ws.rs.HEAD;
import javax.ws.rs.Path;
import javax.ws.rs.Produces;
import javax.ws.rs.QueryParam;
import javax.ws.rs.core.Response;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.HttpStatus;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClientBuilder;
import org.camunda.bpm.engine.ProcessEngines;
import org.openecomp.mso.HealthCheckUtils;
import org.openecomp.mso.logger.MessageEnum;
import org.openecomp.mso.logger.MsoLogger;
import org.openecomp.mso.utils.CryptoUtils;
import org.openecomp.mso.utils.UUIDChecker;
@Path("/")
public class HealthCheckHandler {
private static MsoLogger msoLogger = MsoLogger.getMsoLogger(MsoLogger.Catalog.BPEL);
private static final String SITENAME = "mso.sitename";
private static final String ADPTER_ENDPOINT = "mso.openecomp.adapters.db.endpoint";
private static final String OPENECOMP_ADAPTER_NAMESPACE = "mso.openecomp.adapter.namespace";
private static final String CONFIG = "mso.bpmn.urn.properties";
private static final String CREDENTIAL = "mso.adapters.db.auth";
private static final String MSOKEY = "mso.msoKey";
private String healthcheckDebugEnabled = "mso.healthcheck.log.debug";
private static final String CHECK_HTML = "Health CheckApplication ready";
private static final String NOT_FOUND = "Application Not StartedApplication not started. Properties file missing or invalid or database Connection failed";
private static final String NOT_HEALTHY = "Application Not StartedApplication not available or at least one of the sub-modules is not available.";
public static final Response HEALTH_CHECK_RESPONSE = Response.status (HttpStatus.SC_OK)
.entity (CHECK_HTML)
.build ();
public static final Response HEALTH_CHECK_NOK_RESPONSE = Response.status (HttpStatus.SC_SERVICE_UNAVAILABLE)
.entity (NOT_HEALTHY)
. build ();
public static final Response NOT_STARTED_RESPONSE = Response.status (HttpStatus.SC_SERVICE_UNAVAILABLE)
.entity (NOT_FOUND)
.build ();
@HEAD
@GET
@Path("/nodehealthcheck")
@Produces("text/html")
public Response nodeHealthcheck () {
MsoLogger.setServiceName ("NodeHealthcheck");
// Generate a Request Id
String requestId = UUIDChecker.generateUUID(msoLogger);
PropertyConfiguration propertyConfiguration = PropertyConfiguration.getInstance();
Map props = propertyConfiguration.getProperties(CONFIG);
if (props == null) {
msoLogger.error(MessageEnum.BPMN_GENERAL_EXCEPTION, "BPMN", MsoLogger.getServiceName(), MsoLogger.ErrorCode.AvailabilityError, "Unable to load " + CONFIG);
return NOT_STARTED_RESPONSE;
}
String siteName = props.get(SITENAME);
String endpoint = props.get(ADPTER_ENDPOINT);
if (null == siteName || siteName.length () == 0 || null == endpoint || endpoint.length () == 0) {
msoLogger.error(MessageEnum.BPMN_GENERAL_EXCEPTION, "BPMN", MsoLogger.getServiceName(), MsoLogger.ErrorCode.DataError, "Unable to load key attributes (" + SITENAME + " or " + ADPTER_ENDPOINT + ") from the config file:" + CONFIG);
return NOT_STARTED_RESPONSE;
}
try {
if (!this.getSiteStatus (endpoint, siteName, props.get(CREDENTIAL), props.get(MSOKEY), props.get(OPENECOMP_ADAPTER_NAMESPACE))) {
msoLogger.debug("This site is currently disabled for maintenance.");
return HEALTH_CHECK_NOK_RESPONSE;
}
} catch (Exception e) {
msoLogger.error(MessageEnum.GENERAL_EXCEPTION_ARG, "BPMN", MsoLogger.getServiceName(), MsoLogger.ErrorCode.UnknownError, "Exception while getting SiteStatus", e);
msoLogger.debug("Exception while getting SiteStatus");
return NOT_STARTED_RESPONSE;
}
HealthCheckUtils healthCheck = new HealthCheckUtils ();
if (healthCheck.verifyNodeHealthCheck(HealthCheckUtils.NodeType.BPMN, requestId)) {
msoLogger.debug("nodeHealthcheck - Successful");
return HealthCheckUtils.HEALTH_CHECK_RESPONSE;
} else {
msoLogger.debug("nodeHealthcheck - At leaset one of the sub-modules is not available.");
return HealthCheckUtils.HEALTH_CHECK_NOK_RESPONSE;
}
}
@HEAD
@GET
@Path("/healthcheck")
@Produces("text/html")
public Response healthcheck (@QueryParam("requestId") String requestId) {
MsoLogger.setServiceName ("Healthcheck");
verifyOldUUID(requestId);
PropertyConfiguration propertyConfiguration = PropertyConfiguration.getInstance();
Map props = propertyConfiguration.getProperties(CONFIG);
if (props == null) {
msoLogger.error(MessageEnum.BPMN_GENERAL_EXCEPTION, "BPMN", MsoLogger.getServiceName(), MsoLogger.ErrorCode.AvailabilityError, "Unable to load " + CONFIG);
return NOT_STARTED_RESPONSE;
}
String siteName = props.get(SITENAME);
String endpoint = props.get(ADPTER_ENDPOINT);
if (null == siteName || siteName.length () == 0 || null == endpoint || endpoint.length () == 0) {
msoLogger.error(MessageEnum.BPMN_GENERAL_EXCEPTION, "BPMN", MsoLogger.getServiceName(), MsoLogger.ErrorCode.DataError, "Unable to load key attributes (" + SITENAME + " or " + ADPTER_ENDPOINT + ") from the config file:" + CONFIG);
return NOT_STARTED_RESPONSE;
}
try {
if (!this.getSiteStatus (endpoint, siteName, props.get(CREDENTIAL), props.get(MSOKEY), props.get(OPENECOMP_ADAPTER_NAMESPACE))) {
msoLogger.debug("This site is currently disabled for maintenance.");
return HEALTH_CHECK_NOK_RESPONSE;
}
} catch (Exception e) {
msoLogger.error(MessageEnum.GENERAL_EXCEPTION_ARG, "BPMN", MsoLogger.getServiceName(), MsoLogger.ErrorCode.UnknownError, "Exception while getting SiteStatus", e);
msoLogger.debug("Exception while getting SiteStatus");
return NOT_STARTED_RESPONSE;
}
try {
ProcessEngines.getDefaultProcessEngine().getIdentityService().createGroupQuery().list();
} catch (final Exception e) {
msoLogger.error(MessageEnum.GENERAL_EXCEPTION_ARG, "BPMN", MsoLogger.getServiceName(), MsoLogger.ErrorCode.UnknownError, "Exception while verifying Camunda engine", e);
msoLogger.debug("Exception while verifying Camunda engine");
return NOT_STARTED_RESPONSE;
}
return HEALTH_CHECK_RESPONSE;
}
private String verifyOldUUID (String oldId) {
if (!isValidUUID(oldId)) {
String newId = UUID.randomUUID().toString();
MsoLogger.setLogContext(newId, null);
return newId;
}
MsoLogger.setLogContext(oldId, null);
return oldId;
}
private boolean isValidUUID (String id) {
try {
if (null == id) {
return false;
}
UUID uuid = UUID.fromString(id);
return uuid.toString().equalsIgnoreCase(id);
} catch (IllegalArgumentException iae) {
return false;
}
}
private String decrypt(String encryptedString, String key){
try {
if (encryptedString != null && !encryptedString.isEmpty() && key != null && !key.isEmpty()) {
return CryptoUtils.decrypt(encryptedString, key);
}
} catch (Exception e) {
msoLogger.error(MessageEnum.GENERAL_EXCEPTION_ARG, "BPMN", MsoLogger.getServiceName(), MsoLogger.ErrorCode.UnknownError, "Failed to decrypt credentials", e);
}
return null;
}
private boolean getSiteStatus (String url, String site, String credential, String key, String adapterNamespace) throws Exception {
// set the connection timeout value to 30 seconds (30000 milliseconds)
RequestConfig.Builder requestBuilder = RequestConfig.custom();
requestBuilder = requestBuilder.setConnectTimeout(30000);
requestBuilder = requestBuilder.setConnectionRequestTimeout(30000);
HttpClientBuilder builder = HttpClientBuilder.create ();
builder.setDefaultRequestConfig (requestBuilder.build ());
HttpPost post = new HttpPost(url);
String cred = decrypt(credential, key);
if (cred != null && !cred.isEmpty()) {
post.setHeader("Authorization", "Basic " + Base64.getEncoder().encodeToString(cred.getBytes()));
}
if(healthcheckDebugEnabled == null){
healthcheckDebugEnabled = "false";
}
BPMNLogger.debug(healthcheckDebugEnabled, "Post url is: " + url);
//now create a soap request message as follows:
final StringBuffer payload = new StringBuffer();
payload.append("\n");
payload.append("\n");
payload.append("\n");
payload.append("\n");
payload.append("\n");
payload.append("" + site + "\n");
payload.append("\n");
payload.append("\n");
payload.append("\n");
BPMNLogger.debug(healthcheckDebugEnabled, "Initialize SOAP request to url:" + url);
BPMNLogger.debug(healthcheckDebugEnabled, "The payload of the request is:" + payload);
HttpEntity entity = new StringEntity(payload.toString(),"UTF-8");
post.setEntity(entity);
CloseableHttpClient client = builder.build ();
HttpResponse response = client.execute(post);
BPMNLogger.debug(healthcheckDebugEnabled, "Response received is:" + response);
int statusCode = response.getStatusLine().getStatusCode();
if (statusCode != 200) {
msoLogger.error(MessageEnum.GENERAL_EXCEPTION_ARG, "BPMN", MsoLogger.getServiceName(), MsoLogger.ErrorCode.DataError,
"Communication with DB Adapter failed, The response received from DB Adapter is with failed status code:" + statusCode);
Exception e = new Exception("Communication with DB Adapter failed");
throw e;
}
BufferedReader rd = new BufferedReader(
new InputStreamReader(response.getEntity().getContent()));
StringBuffer result = new StringBuffer();
String line = "";
while ((line = rd.readLine()) != null) {
result.append(line);
}
BPMNLogger.debug(healthcheckDebugEnabled, "Content of the response is:" + result);
String status = result.substring(result.indexOf("") + 8, result.indexOf(""));
client.close (); //shut down the connection
return Boolean.valueOf(status);
}
}