*/
package org.opendaylight.mdsal.binding.generator.impl.reactor;
-import com.google.common.collect.ImmutableList;
+import java.util.function.Function;
import org.eclipse.jdt.annotation.NonNull;
-import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
import org.opendaylight.yangtools.yang.model.api.stmt.AugmentEffectiveStatement;
import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeAwareEffectiveStatement;
-import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeAwareEffectiveStatement.SchemaTreeNamespace;
-import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeEffectiveStatement;
/**
* Generator corresponding to a {@code augment} statement used as a child of a {@code module} statement.
@Override
TargetAugmentEffectiveStatement effectiveIn(final SchemaTreeAwareEffectiveStatement<?, ?> target) {
- final var augment = statement();
- final var stmts = augment.effectiveSubstatements();
- final var builder = ImmutableList.<EffectiveStatement<?, ?>>builderWithExpectedSize(stmts.size());
- for (var child : stmts) {
- if (child instanceof SchemaTreeEffectiveStatement) {
- final var qname = ((SchemaTreeEffectiveStatement<?>) child).getIdentifier();
- // FIXME: orElseThrow()?
- target.get(SchemaTreeNamespace.class, qname).ifPresent(builder::add);
- } else {
- builder.add(child);
- }
- }
- return new TargetAugmentEffectiveStatement(augment, target, builder.build());
+ return effectiveIn(target, Function.identity());
}
@NonNull AugmentRequirement startLinkage(final GeneratorContext context) {