opendaylight/md-sal/sal-netconf-connector/src/main/java/org/opendaylight/controller/sal/connect/netconf/NetconfRemoteSchemaSourceProvider.java

   1 package org.opendaylight.controller.sal.connect.netconf;
   2
   3 import java.util.Set;
   4
   5 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.NetconfState;
   6 import org.opendaylight.yangtools.yang.common.QName;
   7 import org.opendaylight.yangtools.yang.common.RpcResult;
   8 import org.opendaylight.yangtools.yang.data.api.CompositeNode;
   9 import org.opendaylight.yangtools.yang.data.api.SimpleNode;
  10 import org.opendaylight.yangtools.yang.data.impl.ImmutableCompositeNode;
  11 import org.opendaylight.yangtools.yang.data.impl.util.CompositeNodeBuilder;
  12 import org.opendaylight.yangtools.yang.model.util.repo.SchemaSourceProvider;
  13
  14 import com.google.common.base.Optional;
  15 import com.google.common.base.Preconditions;
  16
  17 class NetconfRemoteSchemaSourceProvider implements SchemaSourceProvider<String> {
  18
  19     public static final QName IETF_NETCONF_MONITORING = QName.create(
  20             "urn:ietf:params:xml:ns:yang:ietf-netconf-monitoring", "2010-10-04", "ietf-netconf-monitoring");
  21     public static final QName GET_SCHEMA_QNAME = QName.create(IETF_NETCONF_MONITORING, "get-schema");
  22     public static final QName GET_DATA_QNAME = QName.create(IETF_NETCONF_MONITORING, "data");
  23
  24     NetconfDevice device;
  25
  26     public NetconfRemoteSchemaSourceProvider(NetconfDevice device) {
  27         super();
  28         this.device = device;
  29     }
  30
  31     @Override
  32     public Optional<String> getSchemaSource(String moduleName, Optional<String> revision) {
  33         CompositeNodeBuilder<ImmutableCompositeNode> request = ImmutableCompositeNode.builder(); //
  34         request.setQName(GET_SCHEMA_QNAME) //
  35                 .addLeaf("format", "yang") //
  36                 .addLeaf("identifier", moduleName); //
  37         if (revision.isPresent()) {
  38             request.addLeaf("version", revision.get());
  39         }
  40
  41         device.logger.info("Loading YANG schema source for {}:{}", moduleName, revision);
  42         RpcResult<CompositeNode> schemaReply = device.invokeRpc(GET_SCHEMA_QNAME, request.toInstance());
  43         if (schemaReply.isSuccessful()) {
  44             String schemaBody = getSchemaFromRpc(schemaReply.getResult());
  45             if (schemaBody != null) {
  46                 device.logger.info("YANG Schema successfully retrieved from remote for {}:{}", moduleName, revision);
  47                 return Optional.of(schemaBody);
  48             }
  49         }
  50         device.logger.info("YANG shcema was not successfully retrieved.");
  51         return Optional.absent();
  52     }
  53
  54     private String getSchemaFromRpc(CompositeNode result) {
  55         if (result == null) {
  56             return null;
  57         }
  58         SimpleNode<?> simpleNode = result.getFirstSimpleByName(GET_DATA_QNAME.withoutRevision());
  59         Object potential = simpleNode.getValue();
  60         if (potential instanceof String) {
  61             return (String) potential;
  62         }
  63         return null;
  64     }
  65
  66     public static final boolean isSupportedFor(Set<QName> capabilities) {
  67         return capabilities.contains(IETF_NETCONF_MONITORING);
  68     }
  69 }