Deprecate BindingReflections.getModelRootPackageName()

[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 5fe2d89bc72248db5816f170315fd20a6e8be419..fbcc51fc1223037861aaaf1adba11bb92c19d995 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
@@ -7,39 +7,102 @@
  */
 package org.opendaylight.mdsal.binding.runtime.spi;
 
+import static java.util.Objects.requireNonNull;
+
 import com.google.common.annotations.Beta;
-import java.util.Arrays;
-import java.util.List;
-import java.util.NoSuchElementException;
+import com.google.common.base.Throwables;
+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.yangtools.concepts.CheckedBuilder;
+import org.opendaylight.mdsal.binding.spec.naming.BindingMapping;
+import org.opendaylight.mdsal.binding.spec.reflect.BindingReflections;
+import org.opendaylight.yangtools.yang.binding.BindingObject;
 import org.opendaylight.yangtools.yang.binding.YangModuleInfo;
-import org.opendaylight.yangtools.yang.model.parser.api.YangParserFactory;
+import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
+import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
+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, NoSuchElementException> {
-    private final ModuleInfoSnapshotResolver registry;
+public final class ModuleInfoSnapshotBuilder {
+    private final Set<YangModuleInfo> moduleInfos = new HashSet<>();
+    private final YangParserFactory parserFactory;
+
+    public ModuleInfoSnapshotBuilder(final YangParserFactory parserFactory) {
+        this.parserFactory = requireNonNull(parserFactory);
+    }
+
+    @SuppressWarnings("checkstyle:illegalCatch")
+    public @NonNull ModuleInfoSnapshotBuilder add(final Class<? extends BindingObject> clazz) {
+        final YangModuleInfo moduleInfo;
+        try {
+            moduleInfo = BindingReflections.getModuleInfo(clazz);
+        } catch (Exception e) {
+            Throwables.throwIfUnchecked(e);
+            throw new IllegalStateException("Failed to introspect " + clazz, e);
+        }
+
+        return add(moduleInfo);
+    }
 
-    public ModuleInfoSnapshotBuilder(final String name, final YangParserFactory parserFactory) {
-        registry = new ModuleInfoSnapshotResolver(name, parserFactory);
+    @SuppressWarnings("unchecked")
+    public @NonNull ModuleInfoSnapshotBuilder add(final Class<? extends BindingObject>... classes) {
+        for (Class<? extends BindingObject> clazz : classes) {
+            add(clazz);
+        }
+        return this;
     }
 
     public @NonNull ModuleInfoSnapshotBuilder add(final YangModuleInfo info) {
-        return add(List.of(info));
+        ModuleInfoSnapshotResolver.flatDependencies(moduleInfos, info);
+        return this;
     }
 
     public @NonNull ModuleInfoSnapshotBuilder add(final YangModuleInfo... infos) {
-        return add(Arrays.asList(infos));
+        for (YangModuleInfo info : infos) {
+            add(info);
+        }
+        return this;
     }
 
     public @NonNull ModuleInfoSnapshotBuilder add(final Iterable<? extends YangModuleInfo> infos) {
-        registry.registerModuleInfos(infos);
+        for (YangModuleInfo info : infos) {
+            add(info);
+        }
         return this;
     }
 
-    @Override
-    public ModuleInfoSnapshot build() {
-        return registry.takeSnapshot();
+    /**
+     * 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 Class<?> infoClass = info.getClass();
+            classLoaders.put(BindingMapping.getModelRootPackageName(infoClass.getPackage().getName()),
+                infoClass.getClassLoader());
+
+            try {
+                parser.addSource(source);
+            } catch (YangSyntaxErrorException | IOException e) {
+                throw new YangParserException("Failed to add source for " + info, e);
+            }
+        }
+
+        return new DefaultModuleInfoSnapshot(parser.buildEffectiveModel(), mappedInfos, classLoaders);
     }
 }