Dmaap micro service jar

[appc.git] / services / appc-dmaap-service / appc-event-listener-bundle / src / main / java / org / onap / appc / listener / impl / ControllerImpl.java

diff --git a/services/appc-dmaap-service/appc-event-listener-bundle/src/main/java/org/onap/appc/listener/impl/ControllerImpl.java b/services/appc-dmaap-service/appc-event-listener-bundle/src/main/java/org/onap/appc/listener/impl/ControllerImpl.java
new file mode 100644 (file)
index 0000000..d1c5d61
--- /dev/null
+++ b/services/appc-dmaap-service/appc-event-listener-bundle/src/main/java/org/onap/appc/listener/impl/ControllerImpl.java
@@ -0,0 +1,146 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * ONAP : APPC
+ * ================================================================================
+ * Copyright (C) 2017-2018 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Copyright (C) 2017 Amdocs
+ * ================================================================================
+ * Modifications Copyright (C) 2019 Ericsson
+ * =============================================================================
+ * 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.appc.listener.impl;
+
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.Map;
+import java.util.Set;
+import java.util.concurrent.ArrayBlockingQueue;
+import java.util.concurrent.ThreadPoolExecutor;
+import java.util.concurrent.TimeUnit;
+
+import org.apache.commons.lang3.concurrent.BasicThreadFactory;
+import org.onap.appc.listener.Controller;
+import org.onap.appc.listener.Listener;
+import org.onap.appc.listener.ListenerProperties;
+
+import com.att.eelf.configuration.EELFLogger;
+import com.att.eelf.configuration.EELFManager;
+
+
+/**
+ * A common implementation of a controller. This controller should not need to be modified to implement new listeners
+ *
+ */
+public class ControllerImpl implements Controller {
+
+    private final EELFLogger LOG = EELFManager.getInstance().getLogger(ControllerImpl.class);
+
+    private int LISTENER_COUNT = 1;
+
+    private Map<ListenerProperties, Listener> listeners = null;
+
+    private ThreadPoolExecutor executor;
+
+    /**
+     * Creates a Controller with the set of listener properties which will be used to start listener threads.
+     *
+     * @param properties
+     *            A non null Set of ListenerProperties
+     */
+    public ControllerImpl(Set<ListenerProperties> properties) {
+        listeners = new HashMap<ListenerProperties, Listener>();
+        for (ListenerProperties props : properties) {
+            if (props.getListenerClass() != null) {
+                listeners.put(props, null);
+            } else {
+                LOG.error(String.format(
+                    "The ListenerProperties %s has no Listener class associated with it and will not run.", props));
+                properties.remove(props);
+            }
+        }
+
+        LISTENER_COUNT = properties.size();
+
+        // Only create executor if listeners are configured
+        if (LISTENER_COUNT > 0) {
+        executor = new ThreadPoolExecutor(LISTENER_COUNT, LISTENER_COUNT, 1, TimeUnit.SECONDS,
+            new ArrayBlockingQueue<Runnable>(LISTENER_COUNT));
+
+        // Custom Named thread factory
+        BasicThreadFactory threadFactory = new BasicThreadFactory.Builder().namingPattern("Appc-Listener-%d").build();
+        executor.setThreadFactory(threadFactory);
+        }
+    }
+
+    @Override
+    public void start() {
+        LOG.info("Starting DMaaP Controller.");
+        for (ListenerProperties props : listeners.keySet()) {
+            try {
+                if (props.isDisabled()) {
+                    LOG.warn(String.format("The listener %s is disabled and will not be run", props.getPrefix()));
+                } else {
+                    Listener l = props.getListenerClass().getConstructor(ListenerProperties.class).newInstance(props);
+                    l.setListenerId(props.getPrefix());
+                    listeners.put(props, l);
+                    executor.execute(l);
+                }
+            } catch (Exception e) {
+                LOG.error(String.format("Exception while starting listener %s.", props), e);
+            }
+        }
+    }
+
+    @Override
+    public void stop(boolean stopNow) {
+        LOG.info("Stopping DMaaP Controller.");
+        Iterator<Listener> itr = listeners.values().iterator();
+        while (itr.hasNext()) {
+            Listener l = itr.next();
+            if (stopNow && l != null) {
+                l.stopNow();
+            } else if(l != null){
+                l.stop();
+            }
+            itr.remove();
+        }
+        // disable new tasks from being submitted
+        if(executor != null) {
+            executor.shutdown();
+            int timeout = 300;
+            try {
+                if (!executor.awaitTermination(timeout, TimeUnit.SECONDS)) {
+                    LOG.error("Not all tasks completed execution after " + timeout + " seconds. " +
+                            "Attempting to stop all actively executing tasks.");
+                    executor.shutdownNow();
+                }
+                if (!executor.awaitTermination(timeout, TimeUnit.SECONDS)) {
+                    LOG.error("Could not terminate all tasks after " + (timeout * 2) + " seconds.");
+                }
+            } catch (InterruptedException e) {
+                executor.shutdownNow();
+                Thread.currentThread().interrupt();
+            }
+        }
+    }
+
+    @Override
+    public Map<ListenerProperties, Listener> getListeners() {
+        return listeners;
+    }
+}