*/
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 org.eclipse.jdt.annotation.NonNull;
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.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.REFERENCE)
.addOptional(YangStmtMapping.STATUS)
.build();
+ private static final ExtensionStatementSupport INSTANCE = new ExtensionStatementSupport();
+ private static final ThreadLocal<Map<StmtContext<?, ?, ?>, ExtensionEffectiveStatementImpl>> TL_BUILDERS =
+ new ThreadLocal<>();
- public ExtensionStatementSupport() {
+ private ExtensionStatementSupport() {
super(YangStmtMapping.EXTENSION);
}
- @Override
- public QName parseArgumentValue(final StmtContext<?,?,?> ctx, final String value) {
- return StmtContextUtils.qnameFromArgument(ctx, value);
- }
-
- @Override
- public ExtensionStatement createDeclared(final StmtContext<QName, ExtensionStatement,?> ctx) {
- return new ExtensionStatementImpl(ctx);
+ public static ExtensionStatementSupport getInstance() {
+ return INSTANCE;
}
@Override
- public EffectiveStatement<QName, ExtensionStatement> createEffective(
- final StmtContext<QName, ExtensionStatement, EffectiveStatement<QName,ExtensionStatement>> ctx) {
- return ExtensionEffectiveStatementImpl.create(ctx);
+ public QName parseArgumentValue(final StmtContext<?,?,?> ctx, final String value) {
+ return StmtContextUtils.parseIdentifier(ctx, value);
}
@Override
public void onStatementDefinitionDeclared(
- final Mutable<QName, ExtensionStatement, EffectiveStatement<QName, ExtensionStatement>> stmt) {
+ final Mutable<QName, ExtensionStatement, ExtensionEffectiveStatement> stmt) {
super.onStatementDefinitionDeclared(stmt);
- QName stmtName = stmt.getStatementArgument();
+ QName stmtName = stmt.coerceStatementArgument();
if (OpenConfigStatements.OPENCONFIG_VERSION.getStatementName().isEqualWithoutRevision(stmtName)) {
stmtName = stmtName.withoutRevision();
}
YinElementStatement.class);
stmt.addToNs(StatementDefinitionNamespace.class, stmt.getStatementArgument(),
- new ModelDefinedStatementSupport(new ModelDefinedStatementDefinition(stmt.getStatementArgument(),
+ new ModelDefinedStatementSupport(new ModelDefinedStatementDefinition(stmt.coerceStatementArgument(),
argument != null ? argument.getStatementArgument() : null,
- yinElement != null ? yinElement.getStatementArgument() : false)));
+ yinElement != null && yinElement.coerceStatementArgument())));
}
@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.coerceStatementArgument(), substatements);
+ }
+
+ @Override
+ protected ExtensionStatement createEmptyDeclared(final StmtContext<QName, ExtensionStatement, ?> ctx) {
+ return new EmptyExtensionStatement(ctx.coerceStatementArgument());
+ }
+
+ @Override
+ public ExtensionEffectiveStatement createEffective(
+ final StmtContext<QName, ExtensionStatement, ExtensionEffectiveStatement> ctx) {
+ Map<StmtContext<?, ?, ?>, ExtensionEffectiveStatementImpl> tl = TL_BUILDERS.get();
+ if (tl == null) {
+ tl = new IdentityHashMap<>();
+ TL_BUILDERS.set(tl);
+ }
+
+ final ExtensionEffectiveStatementImpl existing = tl.get(ctx);
+ if (existing != null) {
+ // Implies non-empty map, no cleanup necessary
+ return existing;
+ }
+
+ try {
+ final ExtensionEffectiveStatementImpl created = new ExtensionEffectiveStatementImpl(ctx.buildDeclared(),
+ ctx.getSchemaPath().get());
+ verify(tl.put(ctx, created) == null);
+ try {
+ return super.createEffective(ctx);
+ } finally {
+ verify(tl.remove(ctx) == created);
+
+ }
+ } finally {
+ if (tl.isEmpty()) {
+ TL_BUILDERS.remove();
+ }
+ }
+ }
+
+ @Override
+ protected ExtensionEffectiveStatement createEffective(
+ final StmtContext<QName, ExtensionStatement, ExtensionEffectiveStatement> ctx,
+ final ExtensionStatement declared,
+ final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
+ return finishCreate(ctx, substatements);
+ }
+
+ @Override
+ protected ExtensionEffectiveStatement createEmptyEffective(
+ final StmtContext<QName, ExtensionStatement, ExtensionEffectiveStatement> ctx,
+ final ExtensionStatement declared) {
+ return finishCreate(ctx, ImmutableList.of());
+ }
+
+ private static @NonNull ExtensionEffectiveStatement finishCreate(final StmtContext<?, ?, ?> ctx,
+ final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
+ final ExtensionEffectiveStatementImpl ret = verifyNotNull(verifyNotNull(TL_BUILDERS.get(),
+ "Statement build state not initialized").get(ctx), "No build state found for %s", ctx);
+ ret.setSubstatements(substatements);
+ return ret;
+ }
}
\ No newline at end of file