2 * ============LICENSE_START==========================================
4 * ===================================================================
5 * Copyright (c) 2017 AT&T Intellectual Property
7 * Modifications Copyright (C) 2018 IBM.
8 * ===================================================================
9 * Licensed under the Apache License, Version 2.0 (the "License");
10 * you may not use this file except in compliance with the License.
11 * You may obtain a copy of the License at
13 * http://www.apache.org/licenses/LICENSE-2.0
15 * Unless required by applicable law or agreed to in writing, software
16 * distributed under the License is distributed on an "AS IS" BASIS,
17 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
18 * See the License for the specific language governing permissions and
19 * limitations under the License.
21 * ============LICENSE_END=============================================
22 * ====================================================================
25 package org.onap.music.rest;
27 import java.util.HashMap;
34 import javax.servlet.http.HttpServletResponse;
35 import javax.ws.rs.GET;
36 import javax.ws.rs.Path;
37 import javax.ws.rs.PathParam;
38 import javax.ws.rs.Produces;
39 import javax.ws.rs.core.Context;
40 import javax.ws.rs.core.MediaType;
41 import javax.ws.rs.core.Response;
42 import javax.ws.rs.core.Response.Status;
45 import org.onap.music.eelf.healthcheck.MusicHealthCheck;
46 import org.onap.music.eelf.logging.EELFLoggerDelegate;
47 import org.onap.music.main.MusicUtil;
49 import com.datastax.driver.core.ConsistencyLevel;
51 import io.swagger.annotations.Api;
52 import io.swagger.annotations.ApiOperation;
53 import io.swagger.annotations.ApiParam;
59 @Api(value="Healthcheck Api")
60 public class RestMusicHealthCheckAPI {
63 private static EELFLoggerDelegate logger = EELFLoggerDelegate.getLogger(MusicUtil.class);
64 private static final String ACTIVE_STATUS = "ACTIVE";
65 private static final String INVALID_STATUS = "INVALID";
66 private static final String INACTIVE_STATUS = "INACTIVE";
67 private static final String INVALID_MESSAGE = "Consistency level is invalid...";
68 private static final String INACTIVE_MESSAGE = "One or more nodes in the Cluster is/are down or not responding.";
69 private static final String ACTIVE_MESSAGE = "Cassandra Running and Listening to requests";
72 @Path("/pingCassandra/{consistency}")
73 @ApiOperation(value = "Get Health Status", response = Map.class)
74 @Produces(MediaType.APPLICATION_JSON)
75 public Response cassandraStatus(
76 @Context HttpServletResponse response,
77 @ApiParam(value = "Consistency level",required = true)
78 @PathParam("consistency") String consistency) {
79 logger.info(EELFLoggerDelegate.applicationLogger,"Replying to request for MUSIC Health Check status for Cassandra");
81 Map<String, Object> resultMap = new HashMap<>();
82 if(ConsistencyLevel.valueOf(consistency) == null) {
83 resultMap.put("status",INVALID_STATUS);
84 resultMap.put("message", INVALID_MESSAGE);
85 resultMap.put(INVALID_STATUS, INVALID_STATUS);
86 return Response.status(Status.BAD_REQUEST).entity(resultMap).build();
88 MusicHealthCheck cassHealthCheck = new MusicHealthCheck();
89 String status = cassHealthCheck.getCassandraStatus(consistency);
90 if(status.equals(ACTIVE_STATUS)) {
91 resultMap.put("status",ACTIVE_STATUS);
92 resultMap.put("message", ACTIVE_MESSAGE);
93 resultMap.put(ACTIVE_STATUS, ACTIVE_MESSAGE);
94 return Response.status(Status.OK).entity(resultMap).build();
96 resultMap.put("status",INACTIVE_STATUS);
97 resultMap.put("message", INACTIVE_MESSAGE);
98 resultMap.put(INACTIVE_STATUS, INACTIVE_MESSAGE);
99 return Response.status(Status.BAD_REQUEST).entity(resultMap).build();
104 @Path("/musicHealthCheck")
105 @ApiOperation(value = "Get Health Status", response = Map.class)
106 @Produces(MediaType.APPLICATION_JSON)
107 public Response musicHealthCheck() {
108 logger.info(EELFLoggerDelegate.applicationLogger,"Replying to request for Health Check status for MUSIC");
109 Map<String, Object> resultMap = new HashMap<>();
110 MusicHealthCheck healthCheck = new MusicHealthCheck();
112 String status = healthCheck.getCassandraStatus(ConsistencyLevel.ANY.toString());
113 if(status.equals(ACTIVE_STATUS)) {
114 resultMap.put("Cassandra", "Active");
116 resultMap.put("Cassandra", "Inactive");
118 resultMap.put("MUSIC", "Active");
119 return Response.status(Status.OK).entity(resultMap).build();