netconf/netconf-impl/src/main/java/org/opendaylight/netconf/impl/osgi/NetconfCapabilityMonitoringService.java

   1 /*
   2  * Copyright (c) 2016 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 package org.opendaylight.netconf.impl.osgi;
   9
  10 import static com.google.common.base.Preconditions.checkState;
  11 import static org.opendaylight.netconf.api.xml.XmlNetconfConstants.URN_IETF_PARAMS_NETCONF_CAPABILITY_CANDIDATE_1_0;
  12 import static org.opendaylight.netconf.api.xml.XmlNetconfConstants.URN_IETF_PARAMS_NETCONF_CAPABILITY_URL_1_0;
  13
  14 import com.google.common.base.Function;
  15 import com.google.common.collect.Collections2;
  16 import com.google.common.collect.ImmutableList;
  17 import com.google.common.collect.ImmutableList.Builder;
  18 import com.google.common.collect.Lists;
  19 import com.google.common.collect.Maps;
  20 import java.util.Collection;
  21 import java.util.Collections;
  22 import java.util.HashMap;
  23 import java.util.HashSet;
  24 import java.util.List;
  25 import java.util.Map;
  26 import java.util.Optional;
  27 import java.util.Set;
  28 import org.opendaylight.netconf.api.capability.BasicCapability;
  29 import org.opendaylight.netconf.api.capability.Capability;
  30 import org.opendaylight.netconf.api.monitoring.CapabilityListener;
  31 import org.opendaylight.netconf.api.monitoring.NetconfMonitoringService;
  32 import org.opendaylight.netconf.mapping.api.NetconfOperationServiceFactory;
  33 import org.opendaylight.netconf.notifications.BaseNotificationPublisherRegistration;
  34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
  35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.Yang;
  36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Capabilities;
  37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.CapabilitiesBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Schemas;
  39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.SchemasBuilder;
  40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema;
  41 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.SchemaBuilder;
  42 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.SchemaKey;
  43 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.notifications.rev120206.NetconfCapabilityChange;
  44 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.notifications.rev120206.NetconfCapabilityChangeBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.notifications.rev120206.changed.by.parms.ChangedByBuilder;
  46 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.notifications.rev120206.changed.by.parms.changed.by.server.or.user.ServerBuilder;
  47 import org.opendaylight.yangtools.yang.common.Empty;
  48
  49 class NetconfCapabilityMonitoringService implements CapabilityListener, AutoCloseable {
  50
  51     private static final Schema.Location NETCONF_LOCATION = new Schema.Location(Schema.Location.Enumeration.NETCONF);
  52     private static final List<Schema.Location> NETCONF_LOCATIONS = ImmutableList.of(NETCONF_LOCATION);
  53     private static final BasicCapability CANDIDATE_CAPABILITY =
  54             new BasicCapability(URN_IETF_PARAMS_NETCONF_CAPABILITY_CANDIDATE_1_0);
  55     private static final BasicCapability URL_CAPABILITY =
  56             new BasicCapability(URN_IETF_PARAMS_NETCONF_CAPABILITY_URL_1_0);
  57     private static final Function<Capability, Uri> CAPABILITY_TO_URI = input -> new Uri(input.getCapabilityUri());
  58
  59     private final NetconfOperationServiceFactory netconfOperationProvider;
  60     private final Map<Uri, Capability> capabilities = new HashMap<>();
  61     private final Map<String, Map<String, String>> mappedModulesToRevisionToSchema = new HashMap<>();
  62
  63
  64     private final Set<NetconfMonitoringService.CapabilitiesListener> listeners = new HashSet<>();
  65     private volatile BaseNotificationPublisherRegistration notificationPublisher;
  66
  67     NetconfCapabilityMonitoringService(final NetconfOperationServiceFactory netconfOperationProvider) {
  68         this.netconfOperationProvider = netconfOperationProvider;
  69         netconfOperationProvider.registerCapabilityListener(this);
  70     }
  71
  72     @SuppressWarnings("checkstyle:IllegalCatch")
  73     synchronized Schemas getSchemas() {
  74         try {
  75             return transformSchemas(netconfOperationProvider.getCapabilities());
  76         } catch (final RuntimeException e) {
  77             throw e;
  78         } catch (final Exception e) {
  79             throw new IllegalStateException("Exception while closing", e);
  80         }
  81     }
  82
  83     synchronized String getSchemaForModuleRevision(final String moduleName, final Optional<String> revision) {
  84         final Map<String, String> revisionMapRequest = mappedModulesToRevisionToSchema.get(moduleName);
  85         checkState(revisionMapRequest != null,
  86                 "Capability for module %s not present, available modules : %s",
  87                 moduleName, Collections2.transform(capabilities.values(), CAPABILITY_TO_URI));
  88
  89         final String revToLookup = revision.orElse("");
  90         final String schema = revisionMapRequest.get(revToLookup);
  91         if (schema != null) {
  92             // Exact match found
  93             return schema;
  94         }
  95
  96         // We can recover only if the revision was not specified
  97         checkState(revision.isEmpty(),
  98             "Capability for module %s:%s not present, available revisions for module: %s", moduleName,
  99             revToLookup, revisionMapRequest.keySet());
 100
 101         checkState(revisionMapRequest.size() == 1,
 102                 "Expected 1 capability for module %s, available revisions : %s", moduleName,
 103                 revisionMapRequest.keySet());
 104         // Only one revision is present, so return it
 105         return revisionMapRequest.values().iterator().next();
 106     }
 107
 108     private void updateCapabilityToSchemaMap(final Set<Capability> added, final Set<Capability> removed) {
 109         for (final Capability cap : added) {
 110             if (isValidModuleCapability(cap)) {
 111                 mappedModulesToRevisionToSchema.computeIfAbsent(cap.getModuleName().get(), k -> new HashMap<>())
 112                     .put(cap.getRevision().orElse(""), cap.getCapabilitySchema().get());
 113             }
 114         }
 115         for (final Capability cap : removed) {
 116             if (isValidModuleCapability(cap)) {
 117                 final Map<String, String> revisionMap = mappedModulesToRevisionToSchema.get(cap.getModuleName().get());
 118                 if (revisionMap != null) {
 119                     revisionMap.remove(cap.getRevision().get());
 120                     if (revisionMap.isEmpty()) {
 121                         mappedModulesToRevisionToSchema.remove(cap.getModuleName().get());
 122                     }
 123                 }
 124             }
 125         }
 126     }
 127
 128     private static boolean isValidModuleCapability(final Capability cap) {
 129         return cap.getModuleName().isPresent() && cap.getCapabilitySchema().isPresent();
 130     }
 131
 132     synchronized Capabilities getCapabilities() {
 133         return new CapabilitiesBuilder().setCapability(Lists.newArrayList(capabilities.keySet())).build();
 134     }
 135
 136     synchronized AutoCloseable registerListener(final NetconfMonitoringService.CapabilitiesListener listener) {
 137         listeners.add(listener);
 138         listener.onCapabilitiesChanged(getCapabilities());
 139         listener.onSchemasChanged(getSchemas());
 140         return () -> {
 141             synchronized (NetconfCapabilityMonitoringService.this) {
 142                 listeners.remove(listener);
 143             }
 144         };
 145     }
 146
 147     private static Schemas transformSchemas(final Set<Capability> caps) {
 148         final Map<SchemaKey, Schema> schemas = Maps.newHashMapWithExpectedSize(caps.size());
 149         for (final Capability cap : caps) {
 150             if (isValidModuleCapability(cap)) {
 151                 final SchemaKey key = new SchemaKey(Yang.class, cap.getModuleName().get(),
 152                     cap.getRevision().orElse(""));
 153                 schemas.put(key, new SchemaBuilder()
 154                     .withKey(key)
 155                     .setNamespace(new Uri(cap.getModuleNamespace().get()))
 156                     .setLocation(transformLocations(cap.getLocation()))
 157                     .build());
 158             }
 159         }
 160
 161         return new SchemasBuilder().setSchema(schemas).build();
 162     }
 163
 164     private static List<Schema.Location> transformLocations(final Collection<String> locations) {
 165         if (locations.isEmpty()) {
 166             return NETCONF_LOCATIONS;
 167         }
 168
 169         final Builder<Schema.Location> b = ImmutableList.builder();
 170         b.add(NETCONF_LOCATION);
 171
 172         for (final String location : locations) {
 173             b.add(new Schema.Location(new Uri(location)));
 174         }
 175
 176         return b.build();
 177     }
 178
 179     private static Set<Capability> setupCapabilities(final Set<Capability> caps) {
 180         Set<Capability> capabilities = new HashSet<>(caps);
 181         capabilities.add(CANDIDATE_CAPABILITY);
 182         capabilities.add(URL_CAPABILITY);
 183         // TODO rollback on error not supported EditConfigXmlParser:100
 184         // [RFC6241] 8.5.  Rollback-on-Error Capability
 185         // capabilities.add(new BasicCapability("urn:ietf:params:netconf:capability:rollback-on-error:1.0"));
 186         return capabilities;
 187     }
 188
 189     @Override
 190     public synchronized void close() {
 191         listeners.clear();
 192         capabilities.clear();
 193     }
 194
 195     @Override
 196     public synchronized void onCapabilitiesChanged(final Set<Capability> added, final Set<Capability> removed) {
 197         onCapabilitiesAdded(added);
 198         onCapabilitiesRemoved(removed);
 199         updateCapabilityToSchemaMap(added, removed);
 200         notifyCapabilityChanged(getCapabilities());
 201
 202         // publish notification to notification collector about changed capabilities
 203         if (notificationPublisher != null) {
 204             notificationPublisher.onCapabilityChanged(computeDiff(added, removed));
 205         }
 206     }
 207
 208     private void notifyCapabilityChanged(final Capabilities newCapabilities) {
 209         for (NetconfMonitoringService.CapabilitiesListener listener : listeners) {
 210             listener.onCapabilitiesChanged(newCapabilities);
 211             listener.onSchemasChanged(getSchemas());
 212         }
 213     }
 214
 215
 216     private static NetconfCapabilityChange computeDiff(final Set<Capability> added, final Set<Capability> removed) {
 217         final NetconfCapabilityChangeBuilder netconfCapabilityChangeBuilder = new NetconfCapabilityChangeBuilder();
 218         netconfCapabilityChangeBuilder
 219                 .setChangedBy(new ChangedByBuilder().setServerOrUser(
 220                     new ServerBuilder().setServer(Empty.value()).build()).build());
 221         netconfCapabilityChangeBuilder.setDeletedCapability(Lists.newArrayList(Collections2
 222                 .transform(removed, CAPABILITY_TO_URI)));
 223         netconfCapabilityChangeBuilder.setAddedCapability(Lists.newArrayList(Collections2
 224                 .transform(added, CAPABILITY_TO_URI)));
 225         // TODO modified should be computed ... but why ?
 226         netconfCapabilityChangeBuilder.setModifiedCapability(Collections.emptyList());
 227         return netconfCapabilityChangeBuilder.build();
 228     }
 229
 230
 231     private void onCapabilitiesAdded(final Set<Capability> addedCaps) {
 232         this.capabilities.putAll(Maps.uniqueIndex(setupCapabilities(addedCaps), CAPABILITY_TO_URI));
 233     }
 234
 235     private void onCapabilitiesRemoved(final Set<Capability> removedCaps) {
 236         for (final Capability addedCap : removedCaps) {
 237             capabilities.remove(new Uri(addedCap.getCapabilityUri()));
 238         }
 239     }
 240
 241     void setNotificationPublisher(final BaseNotificationPublisherRegistration notificationPublisher) {
 242         this.notificationPublisher = notificationPublisher;
 243     }
 244 }