Switch to MD-SAL APIs

[openflowplugin.git] / applications / notification-supplier / src / main / java / org / opendaylight / openflowplugin / applications / notification / supplier / impl / AbstractNotificationSupplierForItemRoot.java

diff --git a/applications/notification-supplier/src/main/java/org/opendaylight/openflowplugin/applications/notification/supplier/impl/AbstractNotificationSupplierForItemRoot.java b/applications/notification-supplier/src/main/java/org/opendaylight/openflowplugin/applications/notification/supplier/impl/AbstractNotificationSupplierForItemRoot.java
index a8e8d21ac0553128ef32b29978aabc3b1dffee1b..bd58e15e0dc585af64bfd8cdfa9644c1490ed430 100644 (file)
--- a/applications/notification-supplier/src/main/java/org/opendaylight/openflowplugin/applications/notification/supplier/impl/AbstractNotificationSupplierForItemRoot.java
+++ b/applications/notification-supplier/src/main/java/org/opendaylight/openflowplugin/applications/notification/supplier/impl/AbstractNotificationSupplierForItemRoot.java
@@ -5,14 +5,14 @@
  * 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.applications.notification.supplier.impl;
 
 import com.google.common.base.Preconditions;
-import java.util.Map.Entry;
-import org.opendaylight.controller.md.sal.binding.api.DataBroker;
-import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
-import org.opendaylight.controller.sal.binding.api.NotificationProviderService;
+import java.util.Collection;
+import org.opendaylight.mdsal.binding.api.DataBroker;
+import org.opendaylight.mdsal.binding.api.DataObjectModification;
+import org.opendaylight.mdsal.binding.api.DataTreeModification;
+import org.opendaylight.mdsal.binding.api.NotificationPublishService;
 import org.opendaylight.openflowplugin.applications.notification.supplier.NotificationSupplierForItemRoot;
 import org.opendaylight.yangtools.yang.binding.DataObject;
 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
@@ -20,56 +20,77 @@ import org.opendaylight.yangtools.yang.binding.Notification;
 
 /**
  * Class is package protected abstract implementation for all Root Items
- * Notification Suppliers
+ * Notification Suppliers.
  *
  * @param <O> - data tree item Object
  * @param <C> - Create notification
  * @param <D> - Delete notification
  */
-abstract class AbstractNotificationSupplierForItemRoot<O extends DataObject,
-                                               C extends Notification,
-                                               D extends Notification>
-                    extends AbstractNotificationSupplierBase<O>
-                    implements NotificationSupplierForItemRoot<O, C, D> {
+public abstract class AbstractNotificationSupplierForItemRoot<O extends DataObject, C extends Notification, D extends
+        Notification> extends AbstractNotificationSupplierBase<O> implements NotificationSupplierForItemRoot<O, C, D> {
 
-    private final NotificationProviderService notificationProviderService;
+    private final NotificationPublishService notificationProviderService;
 
     /**
-     * Default constructor for all Root Item Notification Supplier implementation
+     * Default constructor for all Root Item Notification Supplier implementation.
      *
      * @param notificationProviderService - notification publisher
-     * @param db - DataBroker for DataChangeEvent registration
-     * @param clazz - Statistics Notification Class
+     * @param db                          - DataBroker for DataTreeChangeListener registration
+     * @param clazz                       - Statistics Notification Class
      */
-    public AbstractNotificationSupplierForItemRoot(final NotificationProviderService notificationProviderService, final DataBroker db,
-            final Class<O> clazz) {
+    public AbstractNotificationSupplierForItemRoot(final NotificationPublishService notificationProviderService,
+                                                   final DataBroker db, final Class<O> clazz) {
         super(db, clazz);
         this.notificationProviderService = Preconditions.checkNotNull(notificationProviderService);
     }
 
     @Override
-    public void onDataChanged(final AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
-        Preconditions.checkArgument(change != null, "ChangeEvent can not be null!");
-        if (change.getCreatedData() != null && ! (change.getCreatedData().isEmpty())) {
-            for (final Entry<InstanceIdentifier<?>, DataObject> createDataObj : change.getCreatedData().entrySet()) {
-                if (clazz.isAssignableFrom(createDataObj.getKey().getTargetType())) {
-                    final InstanceIdentifier<O> ii = createDataObj.getKey().firstIdentifierOf(clazz);
-                    final C notif = createNotification((O) createDataObj.getValue(), ii);
-                    if (notif != null) {
-                        notificationProviderService.publish(notif);
+    public void onDataTreeChanged(Collection<DataTreeModification<O>> changes) {
+
+        Preconditions.checkNotNull(changes, "Changes may not be null!");
+
+        for (DataTreeModification<O> change : changes) {
+            final InstanceIdentifier<O> key = change.getRootPath().getRootIdentifier();
+            final DataObjectModification<O> mod = change.getRootNode();
+            switch (mod.getModificationType()) {
+                case DELETE:
+                    remove(key, mod.getDataBefore());
+                    break;
+                case SUBTREE_MODIFIED:
+                    update(key, mod.getDataBefore(), mod.getDataAfter());
+                    break;
+                case WRITE:
+                    if (mod.getDataBefore() == null) {
+                        add(key, mod.getDataAfter());
+                    } else {
+                        update(key, mod.getDataBefore(), mod.getDataAfter());
                     }
-                }
+                    break;
+                default:
+                    throw new IllegalArgumentException("Unhandled modification type " + mod.getModificationType());
             }
         }
+    }
 
-        if (change.getRemovedPaths() != null && !(change.getRemovedPaths().isEmpty())) {
-            for (final InstanceIdentifier<?> deleteDataPath : change.getRemovedPaths()) {
-                if (clazz.isAssignableFrom(deleteDataPath.getTargetType())) {
-                    final D notif = deleteNotification(deleteDataPath.firstIdentifierOf(clazz));
-                    if (notif != null) {
-                        notificationProviderService.publish(notif);
-                    }
-                }
+
+    public void add(InstanceIdentifier<O> identifier, O add) {
+        putNotification(createNotification(add, identifier));
+    }
+
+    public void remove(InstanceIdentifier<O> identifier, O del) {
+        putNotification(deleteNotification(identifier.firstIdentifierOf(clazz)));
+    }
+
+    public void update(InstanceIdentifier<O> identifier, O before, O after) {
+        //EMPTY NO-OP
+    }
+
+    private void putNotification(Notification notif) {
+        if (notif != null) {
+            try {
+                notificationProviderService.putNotification(notif);
+            } catch (InterruptedException e) {
+                throw new IllegalStateException("Interrupted while publishing " + notif, e);
             }
         }
     }