ee974c5792782f8ab923078efdbc146fc529166d
[policy/models.git] / models-sim / policy-models-sim-pdp / src / main / java / org / onap / policy / models / sim / pdp / PdpSimulatorMain.java
1 /*-
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2019 Nordix Foundation.
4  *  Modifications Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
5  * ================================================================================
6  * Licensed under the Apache License, Version 2.0 (the "License");
7  * you may not use this file except in compliance with the License.
8  * You may obtain a copy of the License at
9  *
10  *      http://www.apache.org/licenses/LICENSE-2.0
11  *
12  * Unless required by applicable law or agreed to in writing, software
13  * distributed under the License is distributed on an "AS IS" BASIS,
14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15  * See the License for the specific language governing permissions and
16  * limitations under the License.
17  *
18  * SPDX-License-Identifier: Apache-2.0
19  * ============LICENSE_END=========================================================
20  */
21
22 package org.onap.policy.models.sim.pdp;
23
24 import java.util.Arrays;
25 import java.util.Properties;
26 import org.onap.policy.common.endpoints.utils.ParameterUtils;
27 import org.onap.policy.common.utils.services.Registry;
28 import org.onap.policy.models.sim.pdp.exception.PdpSimulatorException;
29 import org.onap.policy.models.sim.pdp.parameters.PdpSimulatorParameterGroup;
30 import org.onap.policy.models.sim.pdp.parameters.PdpSimulatorParameterHandler;
31 import org.slf4j.Logger;
32 import org.slf4j.LoggerFactory;
33
34 /**
35  * This class initiates PdpSimulator.
36  *
37  * @author Ajith Sreekumar (ajith.sreekumar@est.tech)
38  */
39 public class PdpSimulatorMain {
40
41     private static final String PDP_SIMULATOR_FAIL_MSG = "start of pdp simulator failed";
42
43     private static final Logger LOGGER = LoggerFactory.getLogger(PdpSimulatorMain.class);
44
45     private PdpSimulatorActivator activator;
46     private PdpSimulatorParameterGroup parameterGroup;
47
48     /**
49      * Instantiates the PdpSimulator.
50      *
51      * @param args the command line arguments
52      */
53     public PdpSimulatorMain(final String[] args) {
54         if (LOGGER.isInfoEnabled()) {
55             LOGGER.info("In PdpSimulator with parameters {}", Arrays.toString(args));
56         }
57
58         // Check the arguments
59         final PdpSimulatorCommandLineArguments arguments = new PdpSimulatorCommandLineArguments();
60         try {
61             // The arguments return a string if there is a message to print and we should exit
62             final String argumentMessage = arguments.parse(args);
63             if (argumentMessage != null) {
64                 LOGGER.debug(argumentMessage);
65                 return;
66             }
67             // Validate that the arguments are sane
68             arguments.validate();
69         } catch (final PdpSimulatorException e) {
70             LOGGER.error(PDP_SIMULATOR_FAIL_MSG, e);
71             return;
72         }
73
74         // Read the parameters
75         try {
76             parameterGroup = new PdpSimulatorParameterHandler().getParameters(arguments);
77         } catch (final Exception e) {
78             LOGGER.error(PDP_SIMULATOR_FAIL_MSG, e);
79             return;
80         }
81
82         // Read the properties
83         Properties topicProperties = ParameterUtils.getTopicProperties(parameterGroup.getTopicParameterGroup());
84
85         // create the activator
86         activator = new PdpSimulatorActivator(parameterGroup, topicProperties);
87         Registry.register(PdpSimulatorConstants.REG_PDP_SIMULATOR_ACTIVATOR, activator);
88         // Start the activator
89         try {
90             activator.initialize();
91         } catch (final PdpSimulatorException e) {
92             LOGGER.error("start of PdpSimulator failed, used parameters are {}", Arrays.toString(args), e);
93             Registry.unregister(PdpSimulatorConstants.REG_PDP_SIMULATOR_ACTIVATOR);
94             return;
95         }
96
97         // Add a shutdown hook to shut everything down in an orderly manner
98         Runtime.getRuntime().addShutdownHook(new PdpSimulatorShutdownHookClass());
99
100         LOGGER.info("Started PdpSimulator service");
101     }
102
103     /**
104      * Get the parameters specified in JSON.
105      *
106      * @return parameterGroup the parameters
107      */
108     public PdpSimulatorParameterGroup getParameters() {
109         return parameterGroup;
110     }
111
112
113     /**
114      * Shut down Execution.
115      *
116      * @throws PdpSimulatorException on shutdown errors
117      */
118     public void shutdown() throws PdpSimulatorException {
119         // clear the parameterGroup variable
120         parameterGroup = null;
121
122         // clear the pdp simulator activator
123         if (activator != null && activator.isAlive()) {
124             activator.terminate();
125         }
126     }
127
128     /**
129      * The Class PdpSimulatorShutdownHookClass terminates the pdp simulator when its run method is called.
130      */
131     private class PdpSimulatorShutdownHookClass extends Thread {
132         /*
133          * (non-Javadoc)
134          *
135          * @see java.lang.Runnable#run()
136          */
137         @Override
138         public void run() {
139             try {
140                 // Shutdown the pdp simulator service and wait for everything to stop
141                 if (activator != null && activator.isAlive()) {
142                     activator.terminate();
143                 }
144             } catch (final PdpSimulatorException e) {
145                 LOGGER.warn("error occured during shut down of the pdp simulator service", e);
146             }
147         }
148     }
149
150     /**
151      * The main method.
152      *
153      * @param args the arguments
154      *
155      */
156     public static void main(final String[] args) {
157         new PdpSimulatorMain(args);
158     }
159 }