src/main/java/org/onap/music/rest/RestMusicHealthCheckAPI.java

   1 /*
   2  * ============LICENSE_START==========================================
   3  * org.onap.music
   4  * ===================================================================
   5  *  Copyright (c) 2017 AT&T Intellectual Property
   6  *
   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
  12  *
  13  *     http://www.apache.org/licenses/LICENSE-2.0
  14  *
  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.
  20  *
  21  * ============LICENSE_END=============================================
  22  * ====================================================================
  23  */
  24
  25 package org.onap.music.rest;
  26
  27 import java.util.HashMap;
  28 /**
  29  * @author inam
  30  *
  31  */
  32 import java.util.Map;
  33
  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;
  43
  44
  45 import org.onap.music.eelf.healthcheck.MusicHealthCheck;
  46 import org.onap.music.eelf.logging.EELFLoggerDelegate;
  47 import org.onap.music.main.MusicUtil;
  48
  49 import com.datastax.driver.core.ConsistencyLevel;
  50
  51 import io.swagger.annotations.Api;
  52 import io.swagger.annotations.ApiOperation;
  53 import io.swagger.annotations.ApiParam;
  54
  55
  56
  57
  58 @Path("/v2/service")
  59 @Api(value="Healthcheck Api")
  60 public class RestMusicHealthCheckAPI {
  61
  62
  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";
  70
  71     @GET
  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");
  80
  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();
  87         }
  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();
  95         } else {
  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();
 100         }
 101     }
 102
 103     @GET
 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();
 111
 112         String status = healthCheck.getCassandraStatus(ConsistencyLevel.ANY.toString());
 113         if(status.equals(ACTIVE_STATUS)) {
 114             resultMap.put("Cassandra", "Active");
 115         } else {
 116             resultMap.put("Cassandra", "Inactive");
 117         }
 118         resultMap.put("MUSIC", "Active");
 119         return Response.status(Status.OK).entity(resultMap).build();
 120     }
 121
 122 }