fix Honeynode issues with fluorine
[transportpce.git] / tests / honeynode / honeynode-plugin-impl / src / main / java / io / fd / honeycomb / transportpce / device / read / DeviceReaderFactory.java
index c18042dd8bbd7521cfb2f5ef858e5e7010226886..c16a1a8b339331c8760f378b850fa782b51e0234 100644 (file)
  */
 package io.fd.honeycomb.transportpce.device.read;
 
+import com.google.common.collect.Sets;
 import com.google.common.util.concurrent.Futures;
 import com.google.inject.Inject;
 import com.google.inject.name.Named;
-
 import io.fd.honeycomb.translate.read.ReaderFactory;
 import io.fd.honeycomb.translate.read.registry.ModifiableReaderRegistryBuilder;
 import io.fd.honeycomb.translate.util.read.BindingBrokerReader;
 import io.fd.honeycomb.transportpce.device.DeviceConfiguration;
-
+import java.io.File;
+import java.io.IOException;
+import java.text.DateFormat;
+import java.text.SimpleDateFormat;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Set;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.Future;
-
+import org.opendaylight.controller.config.util.capability.Capability;
+import org.opendaylight.controller.config.util.capability.YangModuleCapability;
 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
 import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev170206.org.openroadm.device.container.OrgOpenroadmDevice;
 import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev170206.org.openroadm.device.container.OrgOpenroadmDeviceBuilder;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.NetconfState;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.NetconfStateBuilder;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.Yang;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.Schemas;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.SchemasBuilder;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema.Location;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.Schema.Location.Enumeration;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.netconf.monitoring.rev101004.netconf.state.schemas.SchemaBuilder;
 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import org.opendaylight.yangtools.yang.model.api.Module;
+import org.opendaylight.yangtools.yang.model.api.SchemaContext;
+import org.opendaylight.yangtools.yang.model.repo.api.RevisionSourceIdentifier;
+import org.opendaylight.yangtools.yang.model.repo.api.SchemaSourceFilter;
+import org.opendaylight.yangtools.yang.model.repo.api.SchemaSourceRepresentation;
+import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
+import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
+import org.opendaylight.yangtools.yang.model.repo.spi.PotentialSchemaSource;
+import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceListener;
+import org.opendaylight.yangtools.yang.model.repo.util.FilesystemSchemaSourceCache;
+import org.opendaylight.yangtools.yang.parser.repo.SharedSchemaRepository;
+import org.opendaylight.yangtools.yang.parser.util.TextToASTTransformer;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -44,6 +72,7 @@ public final class DeviceReaderFactory implements ReaderFactory {
     private static final Logger LOG = LoggerFactory.getLogger(DeviceReaderFactory.class);
     public static final InstanceIdentifier<OrgOpenroadmDevice> DEVICE_CONTAINER_ID =
             InstanceIdentifier.create(OrgOpenroadmDevice.class);
+    private static final String YANG_MODELS = "yang";
 
     @Inject
     @Named("device-databroker")
@@ -58,6 +87,7 @@ public final class DeviceReaderFactory implements ReaderFactory {
         registry.add(new BindingBrokerReader<>(DEVICE_CONTAINER_ID, dataBroker,LogicalDatastoreType.OPERATIONAL,
                 OrgOpenroadmDeviceBuilder.class));
         if(writeXMLDataToOper()) {
+            writeNetconfState();
             loadConfigData();
         }
     }
@@ -133,4 +163,128 @@ public final class DeviceReaderFactory implements ReaderFactory {
         }
         return res;
     }
+
+    /**
+     * Write xml data from {@link DeviceConfiguration}
+     * to operational data.
+     *
+     */
+    public boolean writeNetconfState() {
+        Boolean res = false;
+        LOG.info("writting netconf state to oper datastore");
+        final SharedSchemaRepository schemaRepo = new SharedSchemaRepository("honeynode-simulator");
+        final Set<Capability> capabilities = parseSchemasToModuleCapabilities(schemaRepo);
+        final Set<Capability> transformedCapabilities = Sets.newHashSet(capabilities);
+        DummyMonitoringService monitor = new DummyMonitoringService(transformedCapabilities);
+        List<Schema> schemaList = new ArrayList<Schema>();
+        List<Location> locationList = new ArrayList<Location>();
+        Location location = new Location(Enumeration.NETCONF);
+        locationList.add(location );
+        Schema schematobuild = null;
+        for (final Schema schema : monitor.getSchemas().getSchema()) {
+            schematobuild = new SchemaBuilder()
+                    .setIdentifier(schema.getIdentifier())
+                    .setNamespace(schema.getNamespace())
+                    .setVersion(schema.getVersion())
+                    .setFormat(Yang.class)
+                    .setLocation(locationList)
+                    .build();
+            schemaList.add(schematobuild);
+        }
+        Schemas schemas = new SchemasBuilder()
+                .setSchema(schemaList)
+                .build();
+        NetconfState netconfState = new NetconfStateBuilder()
+                .setSchemas(schemas)
+                .build();
+        if (netconfState !=null) {
+            InstanceIdentifier<NetconfState> iid = InstanceIdentifier.create(NetconfState.class);
+            WriteTransaction writeTx = this.dataBroker.newWriteOnlyTransaction();
+            if (writeTx != null ) {
+                LOG.info("WriteTransaction is ok, copy device info to oper datastore");
+                writeTx.put(LogicalDatastoreType.OPERATIONAL, iid, netconfState);
+                Future<Void> future = writeTx.submit();
+                try {
+                    Futures.getChecked(future, ExecutionException.class);
+                    LOG.info("netconf state writed to oper datastore");
+                    res = true;
+                } catch (ExecutionException e) {
+                    LOG.error("Failed to write netconf state to oper datastore");
+                }
+            } else {
+                LOG.error("WriteTransaction object is null");
+            }
+        } else {
+            LOG.error("device data operation gets from xml file is null !");
+        }
+        return res;
+    }
+
+    private Set<Capability> parseSchemasToModuleCapabilities(final SharedSchemaRepository consumer) {
+        final Set<SourceIdentifier> loadedSources = Sets.newHashSet();
+        consumer.registerSchemaSourceListener(TextToASTTransformer.create(consumer, consumer));
+        consumer.registerSchemaSourceListener(new SchemaSourceListener() {
+            @Override
+            public void schemaSourceEncountered(final SchemaSourceRepresentation schemaSourceRepresentation) {}
+
+            @Override
+            public void schemaSourceRegistered(final Iterable<PotentialSchemaSource<?>> potentialSchemaSources) {
+                for (final PotentialSchemaSource<?> potentialSchemaSource : potentialSchemaSources) {
+                    loadedSources.add(potentialSchemaSource.getSourceIdentifier());
+                }
+            }
+
+            @Override
+            public void schemaSourceUnregistered(final PotentialSchemaSource<?> potentialSchemaSource) {}
+        });
+        LOG.info("Loading models from directory.");
+        ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
+        File models = new File(classLoader.getResource(YANG_MODELS).getFile());
+        if (models.exists() && models.isDirectory()) {
+            LOG.info("folder '{}' exists !", models.getAbsolutePath());
+            final FilesystemSchemaSourceCache<YangTextSchemaSource> cache = new FilesystemSchemaSourceCache<>(
+                    consumer, YangTextSchemaSource.class, models);
+                consumer.registerSchemaSourceListener(cache);
+        } else {
+            LOG.warn("folder '{}' not exists !", models.getAbsolutePath());
+            LOG.info("Custom module loading skipped.");
+        }
+        SchemaContext schemaContext;
+        try {
+            //necessary for creating mdsal data stores and operations
+            schemaContext = consumer.createSchemaContextFactory(SchemaSourceFilter.ALWAYS_ACCEPT)
+                .createSchemaContext(loadedSources).get();
+        } catch (final InterruptedException | ExecutionException e) {
+            throw new RuntimeException("Cannot parse schema context", e);
+        }
+
+        final Set<Capability> capabilities = Sets.newHashSet();
+
+        for (final Module module : schemaContext.getModules()) {
+            for (final Module subModule : module.getSubmodules()) {
+                addModuleCapability(consumer, capabilities, subModule);
+            }
+            addModuleCapability(consumer, capabilities, module);
+        }
+        return capabilities;
+    }
+
+    private static void addModuleCapability(final SharedSchemaRepository consumer, final Set<Capability> capabilities,
+                                     final Module module) {
+        DateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd");
+
+        //to convert Date to String, use format method of SimpleDateFormat class.
+        String revision = dateFormat.format(module.getRevision());
+        final SourceIdentifier moduleSourceIdentifier = RevisionSourceIdentifier.create(module.getName(),
+            revision);
+        try {
+            final String moduleContent = new String(
+                consumer.getSchemaSource(moduleSourceIdentifier, YangTextSchemaSource.class).get().read());
+            capabilities.add(new YangModuleCapability(module, moduleContent));
+            //IOException would be thrown in creating SchemaContext already
+        } catch (ExecutionException | InterruptedException | IOException e) {
+            LOG.warn("Cannot retrieve schema source for module {} from schema repository",
+                    moduleSourceIdentifier.toString(), e);
+        }
+}
 }