add rest serve and distribution framework
[multicloud/framework.git] / artifactbroker / main / src / main / java / org / onap / policy / distribution / main / startstop / DistributionActivator.java
diff --git a/artifactbroker/main/src/main/java/org/onap/policy/distribution/main/startstop/DistributionActivator.java b/artifactbroker/main/src/main/java/org/onap/policy/distribution/main/startstop/DistributionActivator.java
new file mode 100644 (file)
index 0000000..2a5dd88
--- /dev/null
@@ -0,0 +1,153 @@
+/*-
+ * ============LICENSE_START=======================================================
+ *  Copyright (C) 2018 Ericsson. All rights reserved.
+ * ================================================================================
+ * 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.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.distribution.main.startstop;
+
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Map.Entry;
+
+import org.onap.policy.common.logging.flexlogger.FlexLogger;
+import org.onap.policy.common.logging.flexlogger.Logger;
+import org.onap.policy.common.parameters.ParameterService;
+import org.onap.policy.distribution.main.PolicyDistributionException;
+import org.onap.policy.distribution.main.parameters.DistributionParameterGroup;
+import org.onap.policy.distribution.main.rest.DistributionRestServer;
+
+/**
+ * This class wraps a distributor so that it can be activated as a complete service together with all its distribution
+ * and forwarding handlers.
+ */
+public class DistributionActivator {
+    // The logger for this class
+    private static final Logger LOGGER = FlexLogger.getLogger(DistributionActivator.class);
+
+    // The parameters of this policy distribution activator
+    private final DistributionParameterGroup distributionParameterGroup;
+
+    // The map of reception handlers initialized by this distribution activator
+
+    private static boolean alive = false;
+
+    private DistributionRestServer restServer;
+
+    /**
+     * Instantiate the activator for policy distribution as a complete service.
+     *
+     * @param distributionParameterGroup the parameters for the distribution service
+     */
+    public DistributionActivator(final DistributionParameterGroup distributionParameterGroup) {
+        this.distributionParameterGroup = distributionParameterGroup;
+    }
+
+    /**
+     * Initialize distribution as a complete service.
+     *
+     * @throws PolicyDistributionException on errors in initializing the service
+     */
+    @SuppressWarnings("unchecked")
+    public void initialize() throws PolicyDistributionException {
+        LOGGER.debug("Policy distribution starting as a service . . .");
+        startDistributionRestServer();
+        registerToParameterService(distributionParameterGroup);
+        DistributionActivator.setAlive(true);
+        LOGGER.debug("Policy distribution started as a service");
+    }
+
+    /**
+     * Starts the distribution rest server using configuration parameters.
+     *
+     * @throws PolicyDistributionException if server start fails
+     */
+    private void startDistributionRestServer() throws PolicyDistributionException {
+        distributionParameterGroup.getRestServerParameters().setName(distributionParameterGroup.getName());
+        restServer = new DistributionRestServer(distributionParameterGroup.getRestServerParameters());
+        if (!restServer.start()) {
+            throw new PolicyDistributionException(
+                    "Failed to start distribution rest server. Check log for more details...");
+        }
+    }
+
+    /**
+     * Terminate policy distribution.
+     *
+     * @throws PolicyDistributionException on termination errors
+     */
+    public void terminate() throws PolicyDistributionException {
+        try {
+            deregisterToParameterService(distributionParameterGroup);
+            DistributionActivator.setAlive(false);
+
+            // Stop the distribution rest server
+            restServer.stop();
+        } catch (final Exception exp) {
+            LOGGER.error("Policy distribution service termination failed", exp);
+            throw new PolicyDistributionException(exp.getMessage(), exp);
+        }
+    }
+
+    /**
+     * Get the parameters used by the activator.
+     *
+     * @return the parameters of the activator
+     */
+    public DistributionParameterGroup getParameterGroup() {
+        return distributionParameterGroup;
+    }
+
+    /**
+     * Method to register the parameters to Common Parameter Service.
+     *
+     * @param distributionParameterGroup the distribution parameter group
+     */
+    public void registerToParameterService(final DistributionParameterGroup distributionParameterGroup) {
+        ParameterService.register(distributionParameterGroup);
+        //@formatter:off
+        //@formatter:on
+    }
+
+    /**
+     * Method to deregister the parameters from Common Parameter Service.
+     *
+     * @param distributionParameterGroup the distribution parameter group
+     */
+    public void deregisterToParameterService(final DistributionParameterGroup distributionParameterGroup) {
+        ParameterService.deregister(distributionParameterGroup.getName());
+        //@formatter:on
+    }
+
+    /**
+     * Returns the alive status of distribution service.
+     *
+     * @return the alive
+     */
+    public static boolean isAlive() {
+        return alive;
+    }
+
+    /**
+     * Change the alive status of distribution service.
+     *
+     * @param status the status
+     */
+    public static void setAlive(final boolean status) {
+        alive = status;
+    }
+}