*/
package org.opendaylight.yangtools.yang.parser.rfc7950.stmt.choice;
-import com.google.common.collect.ImmutableSet;
-import com.google.common.collect.ImmutableSortedMap;
-import java.util.LinkedHashSet;
-import java.util.Objects;
+import static java.util.Objects.requireNonNull;
+
+import com.google.common.collect.ImmutableList;
+import java.util.Collection;
import java.util.Optional;
-import java.util.Set;
-import java.util.SortedMap;
-import java.util.TreeMap;
+import org.eclipse.jdt.annotation.NonNull;
+import org.eclipse.jdt.annotation.Nullable;
import org.opendaylight.yangtools.yang.common.QName;
-import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
import org.opendaylight.yangtools.yang.model.api.CaseSchemaNode;
import org.opendaylight.yangtools.yang.model.api.ChoiceSchemaNode;
import org.opendaylight.yangtools.yang.model.api.DerivableSchemaNode;
+import org.opendaylight.yangtools.yang.model.api.SchemaPath;
import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
import org.opendaylight.yangtools.yang.model.api.stmt.ChoiceEffectiveStatement;
import org.opendaylight.yangtools.yang.model.api.stmt.ChoiceStatement;
-import org.opendaylight.yangtools.yang.model.api.stmt.DefaultEffectiveStatement;
-import org.opendaylight.yangtools.yang.model.api.stmt.MandatoryEffectiveStatement;
-import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.AbstractEffectiveDataSchemaNode;
-import org.opendaylight.yangtools.yang.parser.spi.meta.InferenceException;
+import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeEffectiveStatement;
+import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.AbstractDeclaredEffectiveStatement.DefaultWithSchemaTree.WithSubstatements;
+import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.AugmentationTargetMixin;
+import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.DataSchemaNodeMixin;
+import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.MandatoryMixin;
import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
-import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
-
-final class ChoiceEffectiveStatementImpl extends AbstractEffectiveDataSchemaNode<ChoiceStatement>
- implements ChoiceEffectiveStatement, ChoiceSchemaNode, DerivableSchemaNode {
- private final ImmutableSet<AugmentationSchemaNode> augmentations;
- private final ImmutableSortedMap<QName, CaseSchemaNode> cases;
+final class ChoiceEffectiveStatementImpl extends WithSubstatements<QName, ChoiceStatement, ChoiceEffectiveStatement>
+ implements ChoiceEffectiveStatement, ChoiceSchemaNode, DerivableSchemaNode,
+ DataSchemaNodeMixin<QName, ChoiceStatement>, AugmentationTargetMixin<QName, ChoiceStatement>,
+ MandatoryMixin<QName, ChoiceStatement> {
private final CaseSchemaNode defaultCase;
private final ChoiceSchemaNode original;
- private final boolean mandatory;
-
- ChoiceEffectiveStatementImpl(
- final StmtContext<QName, ChoiceStatement, EffectiveStatement<QName, ChoiceStatement>> ctx) {
- super(ctx);
- this.original = (ChoiceSchemaNode) ctx.getOriginalCtx().map(StmtContext::buildEffective).orElse(null);
-
- // initSubstatementCollectionsAndFields
- final Set<AugmentationSchemaNode> augmentationsInit = new LinkedHashSet<>();
- final SortedMap<QName, CaseSchemaNode> casesInit = new TreeMap<>();
-
- for (final EffectiveStatement<?, ?> effectiveStatement : effectiveSubstatements()) {
- if (effectiveStatement instanceof AugmentationSchemaNode) {
- final AugmentationSchemaNode augmentationSchema = (AugmentationSchemaNode) effectiveStatement;
- augmentationsInit.add(augmentationSchema);
- }
- if (effectiveStatement instanceof CaseSchemaNode) {
- final CaseSchemaNode choiceCaseNode = (CaseSchemaNode) effectiveStatement;
- // FIXME: we may be overwriting a previous entry, is that really okay?
- casesInit.put(choiceCaseNode.getQName(), choiceCaseNode);
- }
- }
-
- this.augmentations = ImmutableSet.copyOf(augmentationsInit);
- this.cases = ImmutableSortedMap.copyOfSorted(casesInit);
-
- final Optional<String> defaultArg = findFirstEffectiveSubstatementArgument(DefaultEffectiveStatement.class);
- if (defaultArg.isPresent()) {
- final String arg = defaultArg.get();
- final QName qname;
- try {
- qname = QName.create(getQName(), arg);
- } catch (IllegalArgumentException e) {
- throw new SourceException(ctx.getStatementSourceReference(), "Default statement has invalid name '%s'",
- arg, e);
- }
-
- // FIXME: this does not work with submodules, as they are
- defaultCase = InferenceException.throwIfNull(cases.get(qname), ctx.getStatementSourceReference(),
- "Default statement refers to missing case %s", qname);
- } else {
- defaultCase = null;
- }
+ private final @NonNull SchemaPath path;
+ private final int flags;
+
+ ChoiceEffectiveStatementImpl(final ChoiceStatement declared,
+ final StmtContext<QName, ChoiceStatement, ChoiceEffectiveStatement> ctx,
+ final ImmutableList<? extends EffectiveStatement<?, ?>> substatements, final int flags,
+ final @Nullable CaseSchemaNode defaultCase, final @Nullable ChoiceSchemaNode original) {
+ super(declared, ctx, substatements);
+ this.flags = flags;
+ this.path = ctx.getSchemaPath().get();
+ this.defaultCase = defaultCase;
+ this.original = original;
+ }
- mandatory = findFirstEffectiveSubstatementArgument(MandatoryEffectiveStatement.class).orElse(Boolean.FALSE)
- .booleanValue();
+ @Override
+ public @NonNull QName argument() {
+ return getQName();
}
@Override
- public Optional<ChoiceSchemaNode> getOriginal() {
- return Optional.ofNullable(original);
+ @Deprecated
+ public SchemaPath getPath() {
+ return path;
}
@Override
- public Set<AugmentationSchemaNode> getAvailableAugmentations() {
- return augmentations;
+ public int flags() {
+ return flags;
}
@Override
- public SortedMap<QName, CaseSchemaNode> getCases() {
- return cases;
+ public Optional<ChoiceSchemaNode> getOriginal() {
+ return Optional.ofNullable(original);
}
@Override
- public Optional<CaseSchemaNode> getDefaultCase() {
- return Optional.ofNullable(defaultCase);
+ public Optional<? extends CaseSchemaNode> findCase(final QName qname) {
+ final SchemaTreeEffectiveStatement<?> child = schemaTreeNamespace().get(requireNonNull(qname));
+ return child instanceof CaseSchemaNode ? Optional.of((CaseSchemaNode) child) : Optional.empty();
}
@Override
- public boolean isMandatory() {
- return mandatory;
+ public Collection<? extends @NonNull CaseSchemaNode> getCases() {
+ return filterEffectiveStatements(CaseSchemaNode.class);
}
@Override
- public int hashCode() {
- final int prime = 31;
- int result = 1;
- result = prime * result + Objects.hashCode(getQName());
- result = prime * result + Objects.hashCode(getPath());
- return result;
+ public Optional<CaseSchemaNode> getDefaultCase() {
+ return Optional.ofNullable(defaultCase);
}
@Override
- public boolean equals(final Object obj) {
- if (this == obj) {
- return true;
- }
- if (obj == null) {
- return false;
- }
- if (getClass() != obj.getClass()) {
- return false;
- }
- final ChoiceEffectiveStatementImpl other = (ChoiceEffectiveStatementImpl) obj;
- return Objects.equals(getQName(), other.getQName()) && Objects.equals(getPath(), other.getPath());
+ public ChoiceEffectiveStatement asEffectiveStatement() {
+ return this;
}
@Override
public String toString() {
- return ChoiceEffectiveStatementImpl.class.getSimpleName() + "["
- + "qname=" + getQName()
- + "]";
+ return ChoiceEffectiveStatementImpl.class.getSimpleName() + "[" + "qname=" + getQName() + "]";
}
}
Code Review / yangtools.git / blobdiff