*/
package org.opendaylight.yangtools.yang.parser.rfc7950.stmt.key;
+import static com.google.common.base.Verify.verify;
+
import com.google.common.base.Splitter;
+import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.ImmutableSet.Builder;
-import java.util.Collection;
+import java.util.Set;
+import org.eclipse.jdt.annotation.NonNull;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.common.QNameModule;
import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
+import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
+import org.opendaylight.yangtools.yang.model.api.stmt.KeyEffectiveStatement;
import org.opendaylight.yangtools.yang.model.api.stmt.KeyStatement;
-import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier;
-import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
-import org.opendaylight.yangtools.yang.parser.spi.meta.QNameCacheNamespace;
+import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.BaseStatementSupport;
+import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils;
import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
public final class KeyStatementSupport
- extends AbstractStatementSupport<Collection<SchemaNodeIdentifier>, KeyStatement,
- EffectiveStatement<Collection<SchemaNodeIdentifier>, KeyStatement>> {
+ extends BaseStatementSupport<Set<QName>, KeyStatement, KeyEffectiveStatement> {
private static final Splitter LIST_KEY_SPLITTER = Splitter.on(' ').omitEmptyStrings().trimResults();
private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(
YangStmtMapping.KEY)
}
@Override
- public Collection<SchemaNodeIdentifier> parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
- final Builder<SchemaNodeIdentifier> builder = ImmutableSet.builder();
+ public ImmutableSet<QName> parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
+ final Builder<QName> builder = ImmutableSet.builder();
int tokens = 0;
for (String keyToken : LIST_KEY_SPLITTER.split(value)) {
- builder.add(SchemaNodeIdentifier.SAME.createChild(StmtContextUtils.qnameFromArgument(ctx, keyToken)));
+ builder.add(StmtContextUtils.parseNodeIdentifier(ctx, keyToken));
tokens++;
}
// Throws NPE on nulls, retains first inserted value, cannot be modified
- final Collection<SchemaNodeIdentifier> ret = builder.build();
- SourceException.throwIf(ret.size() != tokens, ctx.getStatementSourceReference(),
- "Key argument '%s' contains duplicates", value);
+ final ImmutableSet<QName> ret = builder.build();
+ SourceException.throwIf(ret.size() != tokens, ctx.sourceReference(), "Key argument '%s' contains duplicates",
+ value);
return ret;
}
@Override
- public Collection<SchemaNodeIdentifier> adaptArgumentValue(
- final StmtContext<Collection<SchemaNodeIdentifier>, KeyStatement,
- EffectiveStatement<Collection<SchemaNodeIdentifier>, KeyStatement>> ctx,
+ public Set<QName> adaptArgumentValue(final StmtContext<Set<QName>, KeyStatement, KeyEffectiveStatement> ctx,
final QNameModule targetModule) {
- final Builder<SchemaNodeIdentifier> builder = ImmutableSet.builder();
+ final Builder<QName> builder = ImmutableSet.builder();
boolean replaced = false;
- for (final SchemaNodeIdentifier arg : ctx.getStatementArgument()) {
- final QName qname = arg.getLastComponent();
+ for (final QName qname : ctx.getArgument()) {
if (!targetModule.equals(qname.getModule())) {
- final QName newQname = ctx.getFromNamespace(QNameCacheNamespace.class, qname.withModule(targetModule));
- builder.add(SchemaNodeIdentifier.SAME.createChild(newQname));
+ final QName newQname = qname.bindTo(targetModule).intern();
+ builder.add(newQname);
replaced = true;
} else {
- builder.add(arg);
+ builder.add(qname);
}
}
// This makes sure we reuse the collection when a grouping is
// instantiated in the same module
- return replaced ? builder.build() : ctx.getStatementArgument();
+ return replaced ? builder.build() : ctx.argument();
}
@Override
- public KeyStatement createDeclared(final StmtContext<Collection<SchemaNodeIdentifier>, KeyStatement, ?> ctx) {
- return new KeyStatementImpl(ctx);
+ protected SubstatementValidator getSubstatementValidator() {
+ return SUBSTATEMENT_VALIDATOR;
}
@Override
- public EffectiveStatement<Collection<SchemaNodeIdentifier>, KeyStatement> createEffective(
- final StmtContext<Collection<SchemaNodeIdentifier>, KeyStatement,
- EffectiveStatement<Collection<SchemaNodeIdentifier>, KeyStatement>> ctx) {
- return new KeyEffectiveStatementImpl(ctx);
+ protected KeyStatement createDeclared(final StmtContext<Set<QName>, KeyStatement, ?> ctx,
+ final ImmutableList<? extends DeclaredStatement<?>> substatements) {
+ return new RegularKeyStatement(ctx.getRawArgument(), ctx.getArgument(), substatements);
}
@Override
- protected SubstatementValidator getSubstatementValidator() {
- return SUBSTATEMENT_VALIDATOR;
+ protected KeyStatement createEmptyDeclared(final StmtContext<Set<QName>, KeyStatement, ?> ctx) {
+ return new EmptyKeyStatement(ctx.getRawArgument(), ctx.getArgument());
+ }
+
+ @Override
+ protected KeyEffectiveStatement createEffective(final Current<Set<QName>, KeyStatement> stmt,
+ final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
+ final Set<QName> arg = stmt.getArgument();
+ final KeyStatement declared = stmt.declared();
+ if (substatements.isEmpty()) {
+ return arg.equals(declared.argument()) ? new EmptyLocalKeyEffectiveStatement(declared)
+ : new EmptyForeignKeyEffectiveStatement(declared, arg);
+ }
+
+ return arg.equals(declared.argument()) ? new RegularLocalKeyEffectiveStatement(declared, substatements)
+ : new RegularForeignKeyEffectiveStatement(declared, arg, substatements);
+ }
+
+ static @NonNull Object maskSet(final @NonNull Set<QName> set) {
+ return set.size() == 1 ? set.iterator().next() : set;
+ }
+
+ @SuppressWarnings("unchecked")
+ static @NonNull Set<QName> unmaskSet(final @NonNull Object masked) {
+ if (masked instanceof Set) {
+ return (Set<QName>) masked;
+ }
+ verify(masked instanceof QName, "Unexpected argument %s", masked);
+ return ImmutableSet.of((QName) masked);
}
-}
\ No newline at end of file
+}