*/
package org.opendaylight.yangtools.yang.parser.rfc7950.stmt.extension;
+import static com.google.common.base.Verify.verify;
+import static com.google.common.base.Verify.verifyNotNull;
+
+import com.google.common.collect.ImmutableList;
+import java.util.IdentityHashMap;
+import java.util.Map;
+import java.util.stream.Stream;
import org.opendaylight.yangtools.openconfig.model.api.OpenConfigStatements;
import org.opendaylight.yangtools.yang.common.QName;
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.ArgumentStatement;
+import org.opendaylight.yangtools.yang.model.api.stmt.ExtensionEffectiveStatement;
import org.opendaylight.yangtools.yang.model.api.stmt.ExtensionStatement;
import org.opendaylight.yangtools.yang.model.api.stmt.YinElementStatement;
+import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.BaseQNameStatementSupport;
import org.opendaylight.yangtools.yang.parser.spi.ExtensionNamespace;
-import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractQNameStatementSupport;
+import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
import org.opendaylight.yangtools.yang.parser.spi.meta.StatementDefinitionNamespace;
import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
public final class ExtensionStatementSupport
- extends AbstractQNameStatementSupport<ExtensionStatement, EffectiveStatement<QName, ExtensionStatement>> {
+ extends BaseQNameStatementSupport<ExtensionStatement, ExtensionEffectiveStatement> {
private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(YangStmtMapping
.EXTENSION)
.addOptional(YangStmtMapping.ARGUMENT)
.addOptional(YangStmtMapping.STATUS)
.build();
private static final ExtensionStatementSupport INSTANCE = new ExtensionStatementSupport();
+ // FIXME: YANGTOOLS-1185: use EffectiveStmtCtx.Current as key
+ private static final ThreadLocal<Map<StmtContext<?, ?, ?>, ExtensionEffectiveStatementImpl>> TL_BUILDERS =
+ new ThreadLocal<>();
private ExtensionStatementSupport() {
super(YangStmtMapping.EXTENSION);
return StmtContextUtils.parseIdentifier(ctx, value);
}
- @Override
- public ExtensionStatement createDeclared(final StmtContext<QName, ExtensionStatement,?> ctx) {
- return new ExtensionStatementImpl(ctx);
- }
-
- @Override
- public EffectiveStatement<QName, ExtensionStatement> createEffective(
- final StmtContext<QName, ExtensionStatement, EffectiveStatement<QName,ExtensionStatement>> ctx) {
- return ExtensionEffectiveStatementImpl.create(ctx);
- }
-
@Override
public void onStatementDefinitionDeclared(
- final Mutable<QName, ExtensionStatement, EffectiveStatement<QName, ExtensionStatement>> stmt) {
+ final Mutable<QName, ExtensionStatement, ExtensionEffectiveStatement> stmt) {
super.onStatementDefinitionDeclared(stmt);
- QName stmtName = stmt.coerceStatementArgument();
+ QName stmtName = stmt.getArgument();
if (OpenConfigStatements.OPENCONFIG_VERSION.getStatementName().isEqualWithoutRevision(stmtName)) {
stmtName = stmtName.withoutRevision();
}
final StmtContext<Boolean, ?, ?> yinElement = StmtContextUtils.findFirstDeclaredSubstatement(stmt,
YinElementStatement.class);
- stmt.addToNs(StatementDefinitionNamespace.class, stmt.getStatementArgument(),
- new ModelDefinedStatementSupport(new ModelDefinedStatementDefinition(stmt.getStatementArgument(),
- argument != null ? argument.getStatementArgument() : null,
- yinElement != null && yinElement.getStatementArgument())));
+ stmt.addToNs(StatementDefinitionNamespace.class, stmt.argument(),
+ new ModelDefinedStatementSupport(new ModelDefinedStatementDefinition(stmt.getArgument(),
+ argument != null ? argument.argument() : null, yinElement != null && yinElement.getArgument())));
}
@Override
protected SubstatementValidator getSubstatementValidator() {
return SUBSTATEMENT_VALIDATOR;
}
+
+ @Override
+ protected ExtensionStatement createDeclared(final StmtContext<QName, ExtensionStatement, ?> ctx,
+ final ImmutableList<? extends DeclaredStatement<?>> substatements) {
+ return new RegularExtensionStatement(ctx.getArgument(), substatements);
+ }
+
+ @Override
+ protected ExtensionStatement createEmptyDeclared(final StmtContext<QName, ExtensionStatement, ?> ctx) {
+ return new EmptyExtensionStatement(ctx.getArgument());
+ }
+
+ @Override
+ public ExtensionEffectiveStatement createEffective(final Current<QName, ExtensionStatement> stmt,
+ final Stream<? extends StmtContext<?, ?, ?>> declaredSubstatements,
+ final Stream<? extends StmtContext<?, ?, ?>> effectiveSubstatements) {
+ Map<StmtContext<?, ?, ?>, ExtensionEffectiveStatementImpl> tl = TL_BUILDERS.get();
+ if (tl == null) {
+ tl = new IdentityHashMap<>();
+ TL_BUILDERS.set(tl);
+ }
+
+ final StmtContext<?, ?, ?> ctx = stmt.caerbannog();
+ final ExtensionEffectiveStatementImpl existing = tl.get(ctx);
+ if (existing != null) {
+ // Implies non-empty map, no cleanup necessary
+ return existing;
+ }
+
+ try {
+ final ExtensionEffectiveStatementImpl created = new ExtensionEffectiveStatementImpl(stmt.declared(),
+ stmt.getSchemaPath());
+ verify(tl.put(ctx, created) == null);
+ try {
+ return super.createEffective(stmt, declaredSubstatements, effectiveSubstatements);
+ } finally {
+ verify(tl.remove(ctx) == created);
+ }
+ } finally {
+ if (tl.isEmpty()) {
+ TL_BUILDERS.remove();
+ }
+ }
+ }
+
+ @Override
+ protected ExtensionEffectiveStatement createEffective(final Current<QName, ExtensionStatement> stmt,
+ final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
+ final ExtensionEffectiveStatementImpl ret = verifyNotNull(verifyNotNull(TL_BUILDERS.get(),
+ "Statement build state not initialized").get(stmt.caerbannog()), "No build state found for %s", stmt);
+ ret.setSubstatements(substatements);
+ return ret;
+ }
}
\ No newline at end of file