*/
package org.opendaylight.mdsal.binding.runtime.spi;
+import static java.util.Objects.requireNonNull;
+
import com.google.common.annotations.Beta;
-import java.util.NoSuchElementException;
+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.Set;
+import org.eclipse.jdt.annotation.NonNull;
import org.opendaylight.mdsal.binding.runtime.api.ModuleInfoSnapshot;
-import org.opendaylight.yangtools.concepts.CheckedBuilder;
-import org.opendaylight.yangtools.yang.model.parser.api.YangParserFactory;
-import org.opendaylight.yangtools.yang.parser.repo.YangTextSchemaContextResolver;
+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.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 extends AbstractModuleInfoTracker
- implements CheckedBuilder<ModuleInfoSnapshot, NoSuchElementException> {
+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;
+
+ 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 = BindingRuntimeHelpers.getYangModuleInfo(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) {
- super(YangTextSchemaContextResolver.create(name, parserFactory));
+ @SafeVarargs
+ public final @NonNull ModuleInfoSnapshotBuilder add(final Class<? extends BindingObject>... classes) {
+ for (var clazz : classes) {
+ add(clazz);
+ }
+ return this;
}
- @Override
- public synchronized ModuleInfoSnapshot build() {
- return updateSnapshot();
+ public @NonNull ModuleInfoSnapshotBuilder add(final YangModuleInfo info) {
+ ModuleInfoSnapshotResolver.flatDependencies(moduleInfos, info);
+ return this;
+ }
+
+ public @NonNull ModuleInfoSnapshotBuilder add(final YangModuleInfo... infos) {
+ for (var info : infos) {
+ add(info);
+ }
+ return this;
+ }
+
+ public @NonNull ModuleInfoSnapshotBuilder add(final Iterable<? extends YangModuleInfo> infos) {
+ for (YangModuleInfo info : infos) {
+ add(info);
+ }
+ return this;
+ }
+
+ 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 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();
+ final String infoRoot = Naming.getModelRootPackageName(infoClass.getPackage().getName());
+ classLoaders.put(infoRoot, infoClass.getClassLoader());
+ namespaces.put(infoRoot, info.getName().getModule());
+
+ try {
+ parser.addSource(source);
+ } catch (YangSyntaxErrorException | IOException e) {
+ throw new YangParserException("Failed to add source for " + info, e);
+ }
+ }
+
+ 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);
}
}