1 files changed, 149 insertions, 0 deletions
diff --git a/v3po/v3po2vpp/src/main/java/io/fd/honeycomb/v3po/translate/v3po/notification/InterfaceChangeNotificationProducer.java b/v3po/v3po2vpp/src/main/java/io/fd/honeycomb/v3po/translate/v3po/notification/InterfaceChangeNotificationProducer.java
new file mode 100644
index 000000000..f3689d7e2
--- /dev/null
+++ b/v3po/v3po2vpp/src/main/java/io/fd/honeycomb/v3po/translate/v3po/notification/InterfaceChangeNotificationProducer.java
@@ -0,0 +1,149 @@
+/*
+ * Copyright (c) 2016 Cisco and/or its affiliates.
+ *
+ * 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.
+ */
+package io.fd.honeycomb.v3po.translate.v3po.notification;
+
+import com.google.common.collect.Lists;
+import io.fd.honeycomb.v3po.notification.ManagedNotificationProducer;
+import io.fd.honeycomb.v3po.notification.NotificationCollector;
+import io.fd.honeycomb.v3po.translate.MappingContext;
+import io.fd.honeycomb.v3po.translate.v3po.util.NamingContext;
+import io.fd.honeycomb.v3po.translate.v3po.util.TranslateUtils;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.concurrent.CompletionStage;
+import javax.annotation.Nonnull;
+import javax.annotation.Nullable;
+import javax.annotation.concurrent.NotThreadSafe;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev150105.InterfaceDeleted;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev150105.InterfaceDeletedBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev150105.InterfaceNameOrIndex;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev150105.InterfaceStateChange;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev150105.InterfaceStateChangeBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev150105.InterfaceStatus;
+import org.opendaylight.yangtools.yang.binding.Notification;
+import org.openvpp.jvpp.VppBaseCallException;
+import org.openvpp.jvpp.VppInvocationException;
+import org.openvpp.jvpp.dto.SwInterfaceSetFlagsNotification;
+import org.openvpp.jvpp.dto.WantInterfaceEvents;
+import org.openvpp.jvpp.dto.WantInterfaceEventsReply;
+import org.openvpp.jvpp.future.FutureJVpp;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * Notification producer for interface events. It starts interface notification stream and for every
+ * received notification, it transforms it into its BA equivalent and pushes into HC's notification collector.
+ */
+@NotThreadSafe
+public final class InterfaceChangeNotificationProducer implements ManagedNotificationProducer {
+
+    private static final Logger LOG = LoggerFactory.getLogger(InterfaceChangeNotificationProducer.class);
+
+    private final FutureJVpp jvpp;
+    private final NamingContext interfaceContext;
+    private final MappingContext mappingContext;
+    @Nullable
+    private AutoCloseable notificationListenerReg;
+
+    public InterfaceChangeNotificationProducer(@Nonnull final FutureJVpp jvpp,
+                                               @Nonnull final NamingContext interfaceContext,
+                                               @Nonnull final MappingContext mappingContext) {
+        this.jvpp = jvpp;
+        this.interfaceContext = interfaceContext;
+        this.mappingContext = mappingContext;
+    }
+
+    @Override
+    public void start(final NotificationCollector collector) {
+        LOG.trace("Starting interface notifications");
+        enableDisableIfcNotifications(1);
+        LOG.debug("Interface notifications started successfully");
+        notificationListenerReg = jvpp.getNotificationRegistry().registerSwInterfaceSetFlagsNotificationCallback(
+            swInterfaceSetFlagsNotification -> {
+                LOG.trace("Interface notification received: {}", swInterfaceSetFlagsNotification);
+                collector.onNotification(transformNotification(swInterfaceSetFlagsNotification));
+            }
+        );
+    }
+
+    private Notification transformNotification(final SwInterfaceSetFlagsNotification swInterfaceSetFlagsNotification) {
+        if(swInterfaceSetFlagsNotification.deleted == 1) {
+            return new InterfaceDeletedBuilder().setName(getIfcName(swInterfaceSetFlagsNotification)).build();
+        } else {
+            return new InterfaceStateChangeBuilder()
+                .setName(getIfcName(swInterfaceSetFlagsNotification))
+                .setAdminStatus(swInterfaceSetFlagsNotification.adminUpDown == 1 ? InterfaceStatus.Up : InterfaceStatus.Down)
+                .setOperStatus(swInterfaceSetFlagsNotification.linkUpDown == 1 ? InterfaceStatus.Up : InterfaceStatus.Down)
+                .build();
+        }
+    }
+
+    /**
+     * Get mapped name for the interface. Best effort only! The mapping might not yet be stored in context
+     * data tree (write transaction is still in progress and context changes have not been committed yet, or
+     * VPP sends the notification before it returns create request(that would store mapping)).
+     *
+     * In case mapping is not available, index is used as name. TODO inconsistent behavior, maybe just use indices ?
+     */
+    private InterfaceNameOrIndex getIfcName(final SwInterfaceSetFlagsNotification swInterfaceSetFlagsNotification) {
+        return interfaceContext.containsName(swInterfaceSetFlagsNotification.swIfIndex, mappingContext)
+            ? new InterfaceNameOrIndex(interfaceContext.getName(swInterfaceSetFlagsNotification.swIfIndex, mappingContext))
+            : new InterfaceNameOrIndex((long) swInterfaceSetFlagsNotification.swIfIndex);
+    }
+
+    @Override
+    public void stop() {
+        LOG.trace("Stopping interface notifications");
+        enableDisableIfcNotifications(0);
+        LOG.debug("Interface notifications stopped successfully");
+        try {
+            if (notificationListenerReg != null) {
+                notificationListenerReg.close();
+            }
+        } catch (Exception e) {
+            LOG.warn("Unable to properly close notification registration: {}", notificationListenerReg, e);
+        }
+    }
+
+    private void enableDisableIfcNotifications(int enableDisable) {
+        final WantInterfaceEvents wantInterfaceEvents = new WantInterfaceEvents();
+        wantInterfaceEvents.pid = 1;
+        wantInterfaceEvents.enableDisable = enableDisable;
+        final CompletionStage<WantInterfaceEventsReply> wantInterfaceEventsReplyCompletionStage;
+        try {
+            wantInterfaceEventsReplyCompletionStage = jvpp.wantInterfaceEvents(wantInterfaceEvents);
+            TranslateUtils.getReply(wantInterfaceEventsReplyCompletionStage.toCompletableFuture());
+        } catch (VppBaseCallException e) {
+            LOG.warn("Unable to {} interface notifications", enableDisable == 1 ? "enable" : "disable",  e);
+            throw new IllegalStateException("Unable to control interface notifications", e);
+        }
+
+    }
+
+    @Override
+    public Collection<Class<? extends Notification>> getNotificationTypes() {
+        final ArrayList<Class<? extends Notification>> classes = Lists.newArrayList();
+        classes.add(InterfaceStateChange.class);
+        classes.add(InterfaceDeleted.class);
+        return classes;
+    }
+
+    @Override
+    public void close() throws Exception {
+        LOG.trace("Closing interface notifications producer");
+        stop();
+    }
+}