diff options
Diffstat (limited to 'policy-healthcheck/src/main/java')
3 files changed, 297 insertions, 0 deletions
diff --git a/policy-healthcheck/src/main/java/org/openecomp/policy/drools/healthcheck/HealthCheck.java b/policy-healthcheck/src/main/java/org/openecomp/policy/drools/healthcheck/HealthCheck.java new file mode 100644 index 00000000..2ed2e075 --- /dev/null +++ b/policy-healthcheck/src/main/java/org/openecomp/policy/drools/healthcheck/HealthCheck.java @@ -0,0 +1,178 @@ +package org.openecomp.policy.drools.healthcheck; + +import java.util.ArrayList; +import java.util.Properties; + +import javax.ws.rs.core.Response; + +import org.openecomp.policy.drools.http.client.HttpClient; +import org.openecomp.policy.drools.http.server.HttpServletServer; +import org.openecomp.policy.drools.persistence.SystemPersistence; +import org.openecomp.policy.drools.properties.Startable; +import org.openecomp.policy.drools.system.PolicyEngine; + +public interface HealthCheck extends Startable { + + public static class Report { + public String name; + public String url; + public boolean healthy; + public int code; + public String message; + + @Override + public String toString() { + StringBuilder builder = new StringBuilder(); + builder.append("Report [name="); + builder.append(name); + builder.append(", url="); + builder.append(url); + builder.append(", healthy="); + builder.append(healthy); + builder.append(", code="); + builder.append(code); + builder.append(", message="); + builder.append(message); + builder.append("]"); + return builder.toString(); + } + } + + public static class Reports { + public boolean healthy; + public ArrayList<Report> details = new ArrayList<>(); + + @Override + public String toString() { + StringBuilder builder = new StringBuilder(); + builder.append("Reports [healthy="); + builder.append(healthy); + builder.append(", details="); + builder.append(details); + builder.append("]"); + return builder.toString(); + } + } + + public Reports healthCheck(); + + public static final HealthCheck monitor = new HealthCheckMonitor(); +} + +class HealthCheckMonitor implements HealthCheck { + + protected volatile ArrayList<HttpServletServer> servers = new ArrayList<>(); + protected volatile ArrayList<HttpClient> clients = new ArrayList<>(); + protected volatile Properties healthCheckProperties = null; + + public Reports healthCheck() { + Reports reports = new Reports(); + reports.healthy = PolicyEngine.manager.isAlive(); + + HealthCheck.Report engineReport = new Report(); + engineReport.healthy = PolicyEngine.manager.isAlive(); + engineReport.name = "PDP-D"; + engineReport.url = "self"; + engineReport.code = (PolicyEngine.manager.isAlive()) ? 200 : 500; + engineReport.message = (PolicyEngine.manager.isAlive()) ? "alive" : "not alive"; + reports.details.add(engineReport); + + for (HttpClient client : clients) { + HealthCheck.Report report = new Report(); + report.name = client.getName(); + report.url = client.getBaseUrl(); + report.healthy = true; + try { + Response response = client.get(); + report.code = response.getStatus(); + if (report.code != 200) { + report.healthy = false; + reports.healthy = false; + } + + try { + report.message = HttpClient.getBody(response, String.class); + } catch (Exception e) { + e.printStackTrace(); + } + } catch (Exception e) { + report.healthy = false; + reports.healthy = false; + } + reports.details.add(report); + } + return reports; + } + + @Override + public boolean start() throws IllegalStateException { + try { + this.healthCheckProperties = SystemPersistence.manager.getProperties(HealthCheckFeature.CONFIGURATION_PROPERTIES_NAME); + this.servers = HttpServletServer.factory.build(healthCheckProperties); + this.clients = HttpClient.factory.build(healthCheckProperties); + + for (HttpServletServer server : servers) { + try { + server.start(); + } catch (Exception e) { + e.printStackTrace(); + } + } + } catch (Exception e) { + return false; + } + + return true; + } + + @Override + public boolean stop() throws IllegalStateException { + for (HttpServletServer server : servers) { + try { + server.stop(); + } catch (Exception e) { + e.printStackTrace(); + } + } + + for (HttpClient client : clients) { + try { + client.stop(); + } catch (Exception e) { + e.printStackTrace(); + } + } + + return true; + } + + @Override + public void shutdown() throws IllegalStateException { + this.stop(); + } + + @Override + public synchronized boolean isAlive() { + return this.healthCheckProperties != null; + } + + public ArrayList<HttpServletServer> getServers() { + return this.servers; + } + + public ArrayList<HttpClient> getClients() { + return this.clients; + } + + @Override + public String toString() { + StringBuilder builder = new StringBuilder(); + builder.append("HealthCheckMonitor [servers="); + builder.append(servers); + builder.append(", clients="); + builder.append(clients); + builder.append("]"); + return builder.toString(); + } + +} diff --git a/policy-healthcheck/src/main/java/org/openecomp/policy/drools/healthcheck/HealthCheckFeature.java b/policy-healthcheck/src/main/java/org/openecomp/policy/drools/healthcheck/HealthCheckFeature.java new file mode 100644 index 00000000..ce1dd90b --- /dev/null +++ b/policy-healthcheck/src/main/java/org/openecomp/policy/drools/healthcheck/HealthCheckFeature.java @@ -0,0 +1,92 @@ +package org.openecomp.policy.drools.healthcheck; + +import java.util.Properties; + +import org.kie.api.runtime.KieSession; +import org.openecomp.policy.drools.core.FeatureAPI; +import org.openecomp.policy.drools.core.PolicyContainer; +import org.openecomp.policy.drools.core.PolicySession; + +public class HealthCheckFeature implements FeatureAPI { + + public static final String CONFIGURATION_PROPERTIES_NAME = "policy-healthcheck"; + + @Override + public int getSequenceNumber() { + return 2; + } + + @Override + public void globalInit(String[] args, String configDir) { + return; + } + + @Override + public KieSession activatePolicySession(PolicyContainer policyContainer, String name, String kieBaseName) { + return null; + } + + @Override + public void disposeKieSession(PolicySession policySession) { + return; + } + + @Override + public void destroyKieSession(PolicySession policySession) { + return; + } + + @Override + public void beforeStartEngine() throws IllegalStateException { + return; + } + + @Override + public void afterStartEngine() { + try { + HealthCheck.monitor.start(); + } catch (IllegalStateException e) { + e.printStackTrace(); + } + } + + @Override + public void beforeShutdownEngine() { + return; + } + + @Override + public void afterShutdownEngine() { + try { + HealthCheck.monitor.stop(); + } catch (IllegalStateException e) { + e.printStackTrace(); + } + } + + @Override + public void beforeCreateController(String name, Properties properties) { + return; + } + + @Override + public void afterCreateController(String name) { + return; + } + + @Override + public void beforeStartController(String name) { + return; + } + + @Override + public void afterStartController(String name) { + return; + } + + @Override + public boolean isPersistenceEnabled() { + return false; + } + +} diff --git a/policy-healthcheck/src/main/java/org/openecomp/policy/drools/healthcheck/RestHealthCheck.java b/policy-healthcheck/src/main/java/org/openecomp/policy/drools/healthcheck/RestHealthCheck.java new file mode 100644 index 00000000..e37e758a --- /dev/null +++ b/policy-healthcheck/src/main/java/org/openecomp/policy/drools/healthcheck/RestHealthCheck.java @@ -0,0 +1,27 @@ +package org.openecomp.policy.drools.healthcheck; + +import javax.ws.rs.GET; +import javax.ws.rs.Path; +import javax.ws.rs.Produces; +import javax.ws.rs.core.MediaType; + +import org.openecomp.policy.drools.healthcheck.HealthCheck.Reports; + +@Path("/") +public class RestHealthCheck { + + @GET + @Path("{a:healthcheck|test}") + @Produces(MediaType.APPLICATION_JSON) + public Reports healthcheck() { + Reports reports = HealthCheck.monitor.healthCheck(); + return reports; + } + + @GET + @Path("healthcheck/configuration") + @Produces(MediaType.APPLICATION_JSON) + public HealthCheck configuration() { + return HealthCheck.monitor; + } +} |