809e43d0bf1e76fde65a3632ebb69da7cdcaaab7
[policy/xacml-pdp.git] / main / src / main / java / org / onap / policy / pdpx / main / comm / XacmlPdpMessage.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * ONAP
4  * ================================================================================
5  * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  *
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  *
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  *
19  * SPDX-License-Identifier: Apache-2.0
20  * ============LICENSE_END=========================================================
21  */
22
23 package org.onap.policy.pdpx.main.comm;
24
25 import java.net.UnknownHostException;
26 import org.onap.policy.common.utils.network.NetworkUtil;
27 import org.onap.policy.models.pdp.concepts.PdpStateChange;
28 import org.onap.policy.models.pdp.concepts.PdpStatus;
29 import org.onap.policy.models.pdp.concepts.PdpUpdate;
30 import org.onap.policy.models.pdp.enums.PdpHealthStatus;
31 import org.onap.policy.models.pdp.enums.PdpState;
32 import org.onap.policy.pdpx.main.rest.XacmlPdpApplicationManager;
33 import org.onap.policy.pdpx.main.startstop.XacmlPdpActivator;
34 import org.slf4j.Logger;
35 import org.slf4j.LoggerFactory;
36
37 public class XacmlPdpMessage {
38
39     // The logger for this class
40     private static final Logger LOGGER = LoggerFactory.getLogger(XacmlPdpMessage.class);
41
42     /**
43      * Method used to format the status message.
44      *
45      * @param state of the PDP
46      * @return status message of the PDP
47      */
48     public PdpStatus formatStatusMessage(PdpState state) {
49         PdpStatus status = new PdpStatus();
50         status.setName(NetworkUtil.getHostname());
51
52         if (XacmlPdpActivator.getCurrent().isAlive()) {
53             status.setHealthy(PdpHealthStatus.HEALTHY);
54         } else {
55             status.setHealthy(PdpHealthStatus.NOT_HEALTHY);
56         }
57
58         status.setPdpType("xacml");
59         status.setState(state);
60         status.setSupportedPolicyTypes(XacmlPdpApplicationManager.getToscaPolicyTypeIdents());
61
62         LOGGER.debug("formatStatusMessage state {} status{}", state, status);
63
64         return status;
65
66     }
67
68     /**
69      * Method used to format the heartbeat status message.
70      *
71      * @param message PdpStateChange message received from the PAP
72      * @return status message of the PDP
73      */
74     public PdpStatus formatHeartbeatMessage(PdpStateChange message) {
75         PdpStatus status = new PdpStatus();
76         status.setName(NetworkUtil.getHostname());
77
78         if (XacmlPdpActivator.getCurrent().isAlive()) {
79             status.setHealthy(PdpHealthStatus.HEALTHY);
80         } else {
81             status.setHealthy(PdpHealthStatus.NOT_HEALTHY);
82         }
83
84         status.setPdpType("xacml");
85         status.setState(message.getState());
86         status.setPdpGroup(message.getPdpGroup());
87         status.setPdpSubgroup(message.getPdpSubgroup());
88         status.setSupportedPolicyTypes(XacmlPdpApplicationManager.getToscaPolicyTypeIdents());
89
90         return status;
91     }
92
93     /**
94      * Method used to format the PdpUpdate message.
95      *
96      * @param message PdpUpdate message that was received from the PAP
97      * @return status message of the PDP
98      */
99     public PdpStatus formatPdpUpdateMessage(PdpUpdate message, PdpState state) {
100         PdpStatus status = new PdpStatus();
101         status.setName(NetworkUtil.getHostname());
102
103         if (XacmlPdpActivator.getCurrent().isAlive()) {
104             status.setHealthy(PdpHealthStatus.HEALTHY);
105         } else {
106             status.setHealthy(PdpHealthStatus.NOT_HEALTHY);
107         }
108
109         status.setPdpType("xacml");
110         status.setState(state);
111         status.setPdpGroup(message.getPdpGroup());
112         status.setPdpSubgroup(message.getPdpSubgroup());
113         status.setSupportedPolicyTypes(XacmlPdpApplicationManager.getToscaPolicyTypeIdents());
114         status.setPolicies(XacmlPdpApplicationManager.getToscaPolicies());
115
116         return status;
117     }
118 }