Bug 1312 - msg-spy config validation fix

[openflowplugin.git] / openflowplugin / src / main / java / org / opendaylight / openflowplugin / openflow / md / core / sal / OpenflowPluginProvider.java

diff --git a/openflowplugin/src/main/java/org/opendaylight/openflowplugin/openflow/md/core/sal/OpenflowPluginProvider.java b/openflowplugin/src/main/java/org/opendaylight/openflowplugin/openflow/md/core/sal/OpenflowPluginProvider.java
index e3e67b94c9e6406c75f404702b12805a2b8731cd..4e34451db08752a75f5376da5b0e66ac2953538c 100644 (file)
--- a/openflowplugin/src/main/java/org/opendaylight/openflowplugin/openflow/md/core/sal/OpenflowPluginProvider.java
+++ b/openflowplugin/src/main/java/org/opendaylight/openflowplugin/openflow/md/core/sal/OpenflowPluginProvider.java
@@ -1,34 +1,108 @@
+/**
+ * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
 package org.opendaylight.openflowplugin.openflow.md.core.sal;
 
 import java.util.Collection;
 import java.util.Collections;
 
 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
-import org.opendaylight.controller.sal.binding.api.BindingAwareProvider;
 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.ConsumerContext;
 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.ProviderContext;
+import org.opendaylight.controller.sal.binding.api.BindingAwareProvider;
+import org.opendaylight.openflowjava.protocol.spi.connection.SwitchConnectionProvider;
+import org.opendaylight.openflowplugin.openflow.md.core.MDController;
+import org.opendaylight.openflowplugin.statistics.MessageCountCommandProvider;
+import org.opendaylight.openflowplugin.statistics.MessageCountDumper;
+import org.opendaylight.openflowplugin.statistics.MessageObservatory;
+import org.opendaylight.openflowplugin.statistics.MessageSpyCounterImpl;
+import org.opendaylight.yangtools.yang.binding.DataContainer;
 import org.opendaylight.yangtools.yang.binding.RpcService;
 import org.osgi.framework.BundleContext;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
-public class OpenflowPluginProvider implements BindingAwareProvider {
+/**
+ * OFPlugin provider implementation
+ */
+public class OpenflowPluginProvider implements BindingAwareProvider, AutoCloseable {
+
+    private static Logger LOG = LoggerFactory.getLogger(OpenflowPluginProvider.class);
 
     private BindingAwareBroker broker;
 
     private BundleContext context;
 
+    private Collection<SwitchConnectionProvider> switchConnectionProviders;
+
+    private MDController mdController;
+
+    private MessageCountCommandProvider messageCountCommandProvider;
+
+    private MessageObservatory<DataContainer> messageCountProvider;
+
+    private SalRegistrationManager registrationManager;
+
+    /**
+     * Initialization of services and msgSpy counter
+     */
+    public void initialization() {
+        messageCountProvider = new MessageSpyCounterImpl();
+        this.registerProvider();
+    }
+
+    /**
+     * @param switchConnectionProvider
+     */
+    public void setSwitchConnectionProviders(Collection<SwitchConnectionProvider> switchConnectionProvider) {
+        this.switchConnectionProviders = switchConnectionProvider;
+    }
+
+    /**
+     * @return osgi context
+     */
     public BundleContext getContext() {
         return context;
     }
 
+    /**
+     * dependencymanager requirement
+     * @param context
+     *
+     * @deprecated we should stop relying on osgi to provide cli interface for messageCounter
+     */
+    @Deprecated
     public void setContext(BundleContext context) {
         this.context = context;
     }
 
-    SalRegistrationManager registrationManager = new SalRegistrationManager();
-
     @Override
     public void onSessionInitiated(ProviderContext session) {
+        LOG.debug("onSessionInitiated");
+        registrationManager = new SalRegistrationManager();
         registrationManager.onSessionInitiated(session);
+        mdController = new MDController();
+        mdController.setSwitchConnectionProviders(switchConnectionProviders);
+        mdController.setMessageSpyCounter(messageCountProvider);
+        mdController.init();
+        mdController.start();
+        messageCountCommandProvider = new MessageCountCommandProvider(context, messageCountProvider);
+        messageCountCommandProvider.onSessionInitiated(session);
+    }
+
+    @Override
+    public void close() {
+        LOG.debug("close");
+        mdController.stop();
+        mdController = null;
+        registrationManager.close();
+        registrationManager = null;
+        messageCountCommandProvider.close();
+        messageCountCommandProvider = null;
     }
 
     @Override
@@ -41,20 +115,51 @@ public class OpenflowPluginProvider implements BindingAwareProvider {
         return Collections.emptySet();
     }
 
+    @Override
     public java.util.Collection<? extends RpcService> getImplementations() {
         return Collections.emptySet();
     }
 
+    /**
+     * @return BA default broker
+     */
     public BindingAwareBroker getBroker() {
         return broker;
     }
 
+    /**
+     * dependencymanager requirement
+     * @param broker
+     */
     public void setBroker(BindingAwareBroker broker) {
         this.broker = broker;
-        broker.registerProvider(this, context);
-    };
+    }
 
-    public void unsetBroker(BindingAwareBroker broker) {
+    /**
+     * dependencymanager requirement
+     * @param brokerArg
+     */
+    public void unsetBroker(BindingAwareBroker brokerArg) {
         this.broker = null;
-    };
-}
+    }
+
+    private boolean hasAllDependencies(){
+        if(this.broker != null && this.switchConnectionProviders != null) {
+            return true;
+        }
+        return false;
+    }
+
+    /**
+     * register providers for md-sal
+     */
+    private void registerProvider() {
+        if(hasAllDependencies()) {
+            this.broker.registerProvider(this,context);
+        }
+    }
+
+    public MessageCountDumper getMessageCountDumper() {
+        return messageCountProvider;
+    }
+ }