netconf/netconf-notifications-impl/src/main/java/org/opendaylight/netconf/notifications/impl/ops/CreateSubscription.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.netconf.notifications.impl.ops;
  10
  11 import com.google.common.base.Optional;
  12 import com.google.common.base.Preconditions;
  13 import com.google.common.collect.Lists;
  14 import java.util.List;
  15 import org.opendaylight.controller.config.util.xml.DocumentedException;
  16 import org.opendaylight.controller.config.util.xml.XmlElement;
  17 import org.opendaylight.controller.config.util.xml.XmlUtil;
  18 import org.opendaylight.netconf.api.NetconfSession;
  19 import org.opendaylight.netconf.api.xml.XmlNetconfConstants;
  20 import org.opendaylight.netconf.mapping.api.SessionAwareNetconfOperation;
  21 import org.opendaylight.netconf.notifications.NetconfNotification;
  22 import org.opendaylight.netconf.notifications.NetconfNotificationListener;
  23 import org.opendaylight.netconf.notifications.NetconfNotificationRegistry;
  24 import org.opendaylight.netconf.notifications.NotificationListenerRegistration;
  25 import org.opendaylight.netconf.notifications.impl.NetconfNotificationManager;
  26 import org.opendaylight.netconf.util.mapping.AbstractSingletonNetconfOperation;
  27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netconf.notification._1._0.rev080714.CreateSubscriptionInput;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netconf.notification._1._0.rev080714.StreamNameType;
  29 import org.slf4j.Logger;
  30 import org.slf4j.LoggerFactory;
  31 import org.w3c.dom.Document;
  32 import org.w3c.dom.Element;
  33
  34 /**
  35  * Create subscription listens for create subscription requests and registers notification listeners into notification registry.
  36  * Received notifications are sent to the client right away
  37  */
  38 public class CreateSubscription extends AbstractSingletonNetconfOperation implements SessionAwareNetconfOperation, AutoCloseable {
  39
  40     private static final Logger LOG = LoggerFactory.getLogger(CreateSubscription.class);
  41
  42     static final String CREATE_SUBSCRIPTION = "create-subscription";
  43
  44     private final NetconfNotificationRegistry notifications;
  45     private final List<NotificationListenerRegistration> subscriptions = Lists.newArrayList();
  46     private NetconfSession netconfSession;
  47
  48     public CreateSubscription(final String netconfSessionIdForReporting, final NetconfNotificationRegistry notifications) {
  49         super(netconfSessionIdForReporting);
  50         this.notifications = notifications;
  51     }
  52
  53     @Override
  54     protected Element handleWithNoSubsequentOperations(final Document document, final XmlElement operationElement) throws DocumentedException {
  55         operationElement.checkName(CREATE_SUBSCRIPTION);
  56         operationElement.checkNamespace(CreateSubscriptionInput.QNAME.getNamespace().toString());
  57         // FIXME reimplement using CODEC_REGISTRY and parse everything into generated class instance
  58         // Binding doesn't support anyxml nodes yet, so filter could not be retrieved
  59         // xml -> normalized node -> CreateSubscriptionInput conversion could be slower than current approach
  60
  61         // FIXME filter could be supported same way as netconf server filters get and get-config results
  62         final Optional<XmlElement> filter = operationElement.getOnlyChildElementWithSameNamespaceOptionally("filter");
  63         Preconditions.checkArgument(filter.isPresent() == false, "Filter element not yet supported");
  64
  65         // Replay not supported
  66         final Optional<XmlElement> startTime = operationElement.getOnlyChildElementWithSameNamespaceOptionally("startTime");
  67         Preconditions.checkArgument(startTime.isPresent() == false, "StartTime element not yet supported");
  68
  69         // Stop time not supported
  70         final Optional<XmlElement> stopTime = operationElement.getOnlyChildElementWithSameNamespaceOptionally("stopTime");
  71         Preconditions.checkArgument(stopTime.isPresent() == false, "StopTime element not yet supported");
  72
  73         final StreamNameType streamNameType = parseStreamIfPresent(operationElement);
  74
  75         Preconditions.checkNotNull(netconfSession);
  76         // Premature streams are allowed (meaning listener can register even if no provider is available yet)
  77         if(notifications.isStreamAvailable(streamNameType) == false) {
  78             LOG.warn("Registering premature stream {}. No publisher available yet for session {}", streamNameType, getNetconfSessionIdForReporting());
  79         }
  80
  81         final NotificationListenerRegistration notificationListenerRegistration =
  82                 notifications.registerNotificationListener(streamNameType, new NotificationSubscription(netconfSession));
  83         subscriptions.add(notificationListenerRegistration);
  84
  85         return XmlUtil.createElement(document, XmlNetconfConstants.OK, Optional.<String>absent());
  86     }
  87
  88     private static StreamNameType parseStreamIfPresent(final XmlElement operationElement) throws DocumentedException {
  89         final Optional<XmlElement> stream = operationElement.getOnlyChildElementWithSameNamespaceOptionally("stream");
  90         return stream.isPresent() ? new StreamNameType(stream.get().getTextContent()) : NetconfNotificationManager.BASE_STREAM_NAME;
  91     }
  92
  93     @Override
  94     protected String getOperationName() {
  95         return CREATE_SUBSCRIPTION;
  96     }
  97
  98     @Override
  99     protected String getOperationNamespace() {
 100         return CreateSubscriptionInput.QNAME.getNamespace().toString();
 101     }
 102
 103     @Override
 104     public void setSession(final NetconfSession session) {
 105         this.netconfSession = session;
 106     }
 107
 108     @Override
 109     public void close() {
 110         netconfSession = null;
 111         // Unregister from notification streams
 112         for (final NotificationListenerRegistration subscription : subscriptions) {
 113             subscription.close();
 114         }
 115     }
 116
 117     private static class NotificationSubscription implements NetconfNotificationListener {
 118         private final NetconfSession currentSession;
 119
 120         public NotificationSubscription(final NetconfSession currentSession) {
 121             this.currentSession = currentSession;
 122         }
 123
 124         @Override
 125         public void onNotification(final StreamNameType stream, final NetconfNotification notification) {
 126             currentSession.sendMessage(notification);
 127         }
 128     }
 129 }