Bump yangtools to 13.0.0

[mdsal.git] / binding / mdsal-binding-runtime-spi / src / main / java / org / opendaylight / mdsal / binding / runtime / spi / ModuleInfoSnapshotBuilder.java

diff --git a/binding/mdsal-binding-runtime-spi/src/main/java/org/opendaylight/mdsal/binding/runtime/spi/ModuleInfoSnapshotBuilder.java b/binding/mdsal-binding-runtime-spi/src/main/java/org/opendaylight/mdsal/binding/runtime/spi/ModuleInfoSnapshotBuilder.java
index 2c5ffd7cb44879d6a1b2879d388bcf7b701d45ae..363cdab4a37cfc42144e32c027c95ed7c6fa55d4 100644 (file)
--- a/binding/mdsal-binding-runtime-spi/src/main/java/org/opendaylight/mdsal/binding/runtime/spi/ModuleInfoSnapshotBuilder.java
+++ b/binding/mdsal-binding-runtime-spi/src/main/java/org/opendaylight/mdsal/binding/runtime/spi/ModuleInfoSnapshotBuilder.java
@@ -11,26 +11,34 @@ import static java.util.Objects.requireNonNull;
 
 import com.google.common.annotations.Beta;
 import com.google.common.base.Throwables;
+import com.google.common.collect.HashMultimap;
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableSet;
+import com.google.common.collect.Multimaps;
+import com.google.common.collect.SetMultimap;
 import java.io.IOException;
 import java.util.HashMap;
 import java.util.HashSet;
-import java.util.Map;
 import java.util.Set;
 import org.eclipse.jdt.annotation.NonNull;
 import org.opendaylight.mdsal.binding.runtime.api.ModuleInfoSnapshot;
-import org.opendaylight.mdsal.binding.spec.reflect.BindingReflections;
-import org.opendaylight.yangtools.concepts.CheckedBuilder;
 import org.opendaylight.yangtools.yang.binding.BindingObject;
+import org.opendaylight.yangtools.yang.binding.DataRoot;
+import org.opendaylight.yangtools.yang.binding.YangFeature;
 import org.opendaylight.yangtools.yang.binding.YangModuleInfo;
-import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
-import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
+import org.opendaylight.yangtools.yang.binding.contract.Naming;
+import org.opendaylight.yangtools.yang.common.QName;
+import org.opendaylight.yangtools.yang.common.QNameModule;
+import org.opendaylight.yangtools.yang.model.api.source.SourceIdentifier;
+import org.opendaylight.yangtools.yang.model.api.stmt.FeatureSet;
 import org.opendaylight.yangtools.yang.parser.api.YangParser;
 import org.opendaylight.yangtools.yang.parser.api.YangParserException;
 import org.opendaylight.yangtools.yang.parser.api.YangParserFactory;
 import org.opendaylight.yangtools.yang.parser.api.YangSyntaxErrorException;
 
 @Beta
-public final class ModuleInfoSnapshotBuilder implements CheckedBuilder<ModuleInfoSnapshot, YangParserException> {
+public final class ModuleInfoSnapshotBuilder {
+    private final SetMultimap<Class<? extends DataRoot>, YangFeature<?, ?>> moduleFeatures = HashMultimap.create();
     private final Set<YangModuleInfo> moduleInfos = new HashSet<>();
     private final YangParserFactory parserFactory;
 
@@ -42,7 +50,7 @@ public final class ModuleInfoSnapshotBuilder implements CheckedBuilder<ModuleInf
     public @NonNull ModuleInfoSnapshotBuilder add(final Class<? extends BindingObject> clazz) {
         final YangModuleInfo moduleInfo;
         try {
-            moduleInfo = BindingReflections.getModuleInfo(clazz);
+            moduleInfo = BindingRuntimeHelpers.getYangModuleInfo(clazz);
         } catch (Exception e) {
             Throwables.throwIfUnchecked(e);
             throw new IllegalStateException("Failed to introspect " + clazz, e);
@@ -51,9 +59,9 @@ public final class ModuleInfoSnapshotBuilder implements CheckedBuilder<ModuleInf
         return add(moduleInfo);
     }
 
-    @SuppressWarnings("unchecked")
-    public @NonNull ModuleInfoSnapshotBuilder add(final Class<? extends BindingObject>... classes) {
-        for (Class<? extends BindingObject> clazz : classes) {
+    @SafeVarargs
+    public final @NonNull ModuleInfoSnapshotBuilder add(final Class<? extends BindingObject>... classes) {
+        for (var clazz : classes) {
             add(clazz);
         }
         return this;
@@ -65,7 +73,7 @@ public final class ModuleInfoSnapshotBuilder implements CheckedBuilder<ModuleInf
     }
 
     public @NonNull ModuleInfoSnapshotBuilder add(final YangModuleInfo... infos) {
-        for (YangModuleInfo info : infos) {
+        for (var info : infos) {
             add(info);
         }
         return this;
@@ -78,18 +86,33 @@ public final class ModuleInfoSnapshotBuilder implements CheckedBuilder<ModuleInf
         return this;
     }
 
-    @Override
-    public ModuleInfoSnapshot build() throws YangParserException {
+    public <R extends @NonNull DataRoot> @NonNull ModuleInfoSnapshotBuilder addModuleFeatures(final Class<R> module,
+            final Set<? extends YangFeature<?, R>> supportedFeatures) {
+        moduleFeatures.putAll(requireNonNull(module), ImmutableList.copyOf(supportedFeatures));
+        return this;
+    }
+
+    /**
+     * Build {@link ModuleInfoSnapshot} from all {@code moduleInfos} in this builder.
+     *
+     * @return Resulting {@link ModuleInfoSnapshot}
+     * @throws YangParserException if parsing any of the {@link YangModuleInfo} instances fails
+     */
+    public @NonNull ModuleInfoSnapshot build() throws YangParserException {
         final YangParser parser = parserFactory.createParser();
-        final Map<SourceIdentifier, YangModuleInfo> mappedInfos = new HashMap<>();
-        final Map<String, ClassLoader> classLoaders = new HashMap<>();
-        for (YangModuleInfo info : moduleInfos) {
-            final YangTextSchemaSource source = ModuleInfoSnapshotResolver.toYangTextSource(info);
-            mappedInfos.put(source.getIdentifier(), info);
+
+        final var mappedInfos = new HashMap<SourceIdentifier, YangModuleInfo>();
+        final var classLoaders = new HashMap<String, ClassLoader>();
+        final var namespaces = new HashMap<String, QNameModule>();
+
+        for (var info : moduleInfos) {
+            final var source = ModuleInfoSnapshotResolver.toYangTextSource(info);
+            mappedInfos.put(source.sourceId(), info);
 
             final Class<?> infoClass = info.getClass();
-            classLoaders.put(BindingReflections.getModelRootPackageName(infoClass.getPackage()),
-                infoClass.getClassLoader());
+            final String infoRoot = Naming.getModelRootPackageName(infoClass.getPackage().getName());
+            classLoaders.put(infoRoot, infoClass.getClassLoader());
+            namespaces.put(infoRoot, info.getName().getModule());
 
             try {
                 parser.addSource(source);
@@ -98,6 +121,25 @@ public final class ModuleInfoSnapshotBuilder implements CheckedBuilder<ModuleInf
             }
         }
 
+        if (!moduleFeatures.isEmpty()) {
+            final var featuresByModule = FeatureSet.builder();
+            for (var entry : Multimaps.asMap(moduleFeatures).entrySet()) {
+                final var moduleData = entry.getKey();
+                final var moduleRoot = Naming.getModelRootPackageName(moduleData.getPackage().getName());
+                final var moduleNamespace = namespaces.get(moduleRoot);
+                if (moduleNamespace == null) {
+                    throw new YangParserException("Failed to resolve namespace of " + moduleData);
+                }
+
+                featuresByModule.addModuleFeatures(moduleNamespace, entry.getValue().stream()
+                    .map(YangFeature::qname)
+                    .map(QName::getLocalName)
+                    .sorted()
+                    .collect(ImmutableSet.toImmutableSet()));
+            }
+            parser.setSupportedFeatures(featuresByModule.build());
+        }
+
         return new DefaultModuleInfoSnapshot(parser.buildEffectiveModel(), mappedInfos, classLoaders);
     }
 }