*/
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.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;
.addOptional(YangStmtMapping.STATUS)
.build();
private static final ExtensionStatementSupport INSTANCE = new ExtensionStatementSupport();
+ private static final ThreadLocal<Map<Current<?, ?>, ExtensionEffectiveStatementImpl>> TL_BUILDERS =
+ new ThreadLocal<>();
private ExtensionStatementSupport() {
- super(YangStmtMapping.EXTENSION);
+ super(YangStmtMapping.EXTENSION, StatementPolicy.reject());
}
public static ExtensionStatementSupport getInstance() {
return StmtContextUtils.parseIdentifier(ctx, value);
}
- @Override
- public ExtensionStatement createDeclared(final StmtContext<QName, ExtensionStatement,?> ctx) {
- return new ExtensionStatementImpl(ctx);
- }
-
- @Override
- public ExtensionEffectiveStatement createEffective(
- final StmtContext<QName, ExtensionStatement, ExtensionEffectiveStatement> ctx) {
- return ExtensionEffectiveStatementImpl.create(ctx);
- }
-
@Override
public void onStatementDefinitionDeclared(
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<Current<?, ?>, ExtensionEffectiveStatementImpl> tl = TL_BUILDERS.get();
+ if (tl == null) {
+ tl = new IdentityHashMap<>();
+ TL_BUILDERS.set(tl);
+ }
+
+ final ExtensionEffectiveStatementImpl existing = tl.get(stmt);
+ if (existing != null) {
+ // Implies non-empty map, no cleanup necessary
+ return existing;
+ }
+
+ try {
+ final ExtensionEffectiveStatementImpl created = new ExtensionEffectiveStatementImpl(stmt.declared(),
+ stmt.wrapSchemaPath());
+ verify(tl.put(stmt, created) == null);
+ try {
+ return super.createEffective(stmt, declaredSubstatements, effectiveSubstatements);
+ } finally {
+ verify(tl.remove(stmt) == 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), "No build state found for %s", stmt);
+ ret.setSubstatements(substatements);
+ return ret;
+ }
}
\ No newline at end of file