aboutsummaryrefslogtreecommitdiffstats
path: root/music-rest/src/main/java/org/onap/music/rest/RestMusicHealthCheckAPI.java
blob: eef3aa3ae556e8477d23a5b9e5f214bfe66891ad (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
/*
 * ============LICENSE_START==========================================
 * org.onap.music
 * ===================================================================
 *  Copyright (c) 2017 AT&T Intellectual Property
 *
 *  Modifications Copyright (C) 2018 IBM.
 * ===================================================================
 *  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.onap.music.rest;

import java.util.HashMap;
/**
 * @author inam
 *
 */
import java.util.Map;

import javax.servlet.http.HttpServletResponse;
import javax.ws.rs.GET;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.Response.Status;


import org.onap.music.eelf.healthcheck.MusicHealthCheck;
import org.onap.music.eelf.logging.EELFLoggerDelegate;
import org.onap.music.main.MusicUtil;

import com.datastax.driver.core.ConsistencyLevel;

import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;




@Path("/v2/service")
@Api(value="Healthcheck Api")
public class RestMusicHealthCheckAPI {
    
    
    private static EELFLoggerDelegate logger = EELFLoggerDelegate.getLogger(MusicUtil.class);
    private static final String ACTIVE_STATUS = "ACTIVE";
    private static final String INVALID_STATUS = "INVALID";
    private static final String INACTIVE_STATUS = "INACTIVE";
    private static final String INVALID_MESSAGE = "Consistency level is invalid...";
    private static final String INACTIVE_MESSAGE = "One or more nodes in the Cluster is/are down or not responding.";
    private static final String ACTIVE_MESSAGE = "Cassandra Running and Listening to requests";
    private static final String STATUS_KEY = "status";
    private static final String MESSAGE_KEY = "message";
    
    @GET
    @Path("/pingCassandra/{consistency}")
    @ApiOperation(value = "Get Health Status", response = Map.class)
    @Produces(MediaType.APPLICATION_JSON)
    public Response cassandraStatus(
        @Context HttpServletResponse response, 
        @ApiParam(value = "Consistency level",required = true) 
        @PathParam("consistency") String consistency) {
        logger.info(EELFLoggerDelegate.applicationLogger,"Replying to request for MUSIC Health Check status for Cassandra");
        
        Map<String, Object> resultMap = new HashMap<>();
        if(ConsistencyLevel.valueOf(consistency) == null) {
            resultMap.put(STATUS_KEY,INVALID_STATUS);
            resultMap.put(MESSAGE_KEY, INVALID_MESSAGE);
            resultMap.put(INVALID_STATUS, INVALID_STATUS);
            return Response.status(Status.BAD_REQUEST).entity(resultMap).build();
        }
        MusicHealthCheck cassHealthCheck = new MusicHealthCheck();
        String status = cassHealthCheck.getCassandraStatus(consistency);
        if(status.equals(ACTIVE_STATUS)) {
            resultMap.put(STATUS_KEY,ACTIVE_STATUS);
            resultMap.put(MESSAGE_KEY, ACTIVE_MESSAGE);
            resultMap.put(ACTIVE_STATUS, ACTIVE_MESSAGE);
            return Response.status(Status.OK).entity(resultMap).build();
        } else {
            resultMap.put(STATUS_KEY,INACTIVE_STATUS);
            resultMap.put(MESSAGE_KEY, INACTIVE_MESSAGE);
            resultMap.put(INACTIVE_STATUS, INACTIVE_MESSAGE);
            return Response.status(Status.BAD_REQUEST).entity(resultMap).build();
        }
    }
    
    @GET
    @Path("/musicHealthCheck")
    @ApiOperation(value = "Get Health Status", response = Map.class)
    @Produces(MediaType.APPLICATION_JSON)
    public Response musicHealthCheck() {
        logger.info(EELFLoggerDelegate.applicationLogger,"Replying to request for Health Check status for MUSIC");
        Map<String, Object> resultMap = new HashMap<>();
        MusicHealthCheck healthCheck = new MusicHealthCheck();

        String status = healthCheck.getCassandraStatus(ConsistencyLevel.ANY.toString());
        if(status.equals(ACTIVE_STATUS)) {
            resultMap.put("Cassandra", "Active");
        } else {
            resultMap.put("Cassandra", "Inactive");
        }
        resultMap.put("MUSIC", "Active");
        return Response.status(Status.OK).entity(resultMap).build();
    }

}