2 * Copyright (c) 2017 Pantheon Technologies, s.r.o. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.yangtools.yang.parser.rfc7950.stmt.extension;
10 import static com.google.common.base.Verify.verify;
11 import static com.google.common.base.Verify.verifyNotNull;
13 import com.google.common.collect.ImmutableList;
14 import java.util.IdentityHashMap;
16 import org.eclipse.jdt.annotation.NonNull;
17 import org.opendaylight.yangtools.openconfig.model.api.OpenConfigStatements;
18 import org.opendaylight.yangtools.yang.common.QName;
19 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
20 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
21 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
22 import org.opendaylight.yangtools.yang.model.api.stmt.ArgumentStatement;
23 import org.opendaylight.yangtools.yang.model.api.stmt.ExtensionEffectiveStatement;
24 import org.opendaylight.yangtools.yang.model.api.stmt.ExtensionStatement;
25 import org.opendaylight.yangtools.yang.model.api.stmt.YinElementStatement;
26 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.BaseQNameStatementSupport;
27 import org.opendaylight.yangtools.yang.parser.spi.ExtensionNamespace;
28 import org.opendaylight.yangtools.yang.parser.spi.meta.StatementDefinitionNamespace;
29 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
30 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
31 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils;
32 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
34 public final class ExtensionStatementSupport
35 extends BaseQNameStatementSupport<ExtensionStatement, ExtensionEffectiveStatement> {
36 private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(YangStmtMapping
38 .addOptional(YangStmtMapping.ARGUMENT)
39 .addOptional(YangStmtMapping.DESCRIPTION)
40 .addOptional(YangStmtMapping.REFERENCE)
41 .addOptional(YangStmtMapping.STATUS)
43 private static final ExtensionStatementSupport INSTANCE = new ExtensionStatementSupport();
44 private static final ThreadLocal<Map<StmtContext<?, ?, ?>, ExtensionEffectiveStatementImpl>> TL_BUILDERS =
47 private ExtensionStatementSupport() {
48 super(YangStmtMapping.EXTENSION);
51 public static ExtensionStatementSupport getInstance() {
56 public QName parseArgumentValue(final StmtContext<?,?,?> ctx, final String value) {
57 return StmtContextUtils.parseIdentifier(ctx, value);
61 public void onStatementDefinitionDeclared(
62 final Mutable<QName, ExtensionStatement, ExtensionEffectiveStatement> stmt) {
63 super.onStatementDefinitionDeclared(stmt);
65 QName stmtName = stmt.coerceStatementArgument();
66 if (OpenConfigStatements.OPENCONFIG_VERSION.getStatementName().isEqualWithoutRevision(stmtName)) {
67 stmtName = stmtName.withoutRevision();
70 stmt.addContext(ExtensionNamespace.class, stmtName, stmt);
72 final StmtContext<QName, ?, ?> argument = StmtContextUtils.findFirstDeclaredSubstatement(stmt,
73 ArgumentStatement.class);
74 final StmtContext<Boolean, ?, ?> yinElement = StmtContextUtils.findFirstDeclaredSubstatement(stmt,
75 YinElementStatement.class);
77 stmt.addToNs(StatementDefinitionNamespace.class, stmt.getStatementArgument(),
78 new ModelDefinedStatementSupport(new ModelDefinedStatementDefinition(stmt.getStatementArgument(),
79 argument != null ? argument.getStatementArgument() : null,
80 yinElement != null && yinElement.coerceStatementArgument())));
84 protected SubstatementValidator getSubstatementValidator() {
85 return SUBSTATEMENT_VALIDATOR;
89 protected ExtensionStatement createDeclared(final StmtContext<QName, ExtensionStatement, ?> ctx,
90 final ImmutableList<? extends DeclaredStatement<?>> substatements) {
91 return new RegularExtensionStatement(ctx.coerceStatementArgument(), substatements);
95 protected ExtensionStatement createEmptyDeclared(final StmtContext<QName, ExtensionStatement, ?> ctx) {
96 return new EmptyExtensionStatement(ctx.coerceStatementArgument());
100 public ExtensionEffectiveStatement createEffective(
101 final StmtContext<QName, ExtensionStatement, ExtensionEffectiveStatement> ctx) {
102 Map<StmtContext<?, ?, ?>, ExtensionEffectiveStatementImpl> tl = TL_BUILDERS.get();
104 tl = new IdentityHashMap<>();
108 final ExtensionEffectiveStatementImpl existing = tl.get(ctx);
109 if (existing != null) {
110 // Implies non-empty map, no cleanup necessary
115 final ExtensionEffectiveStatementImpl created = new ExtensionEffectiveStatementImpl(ctx.buildDeclared(),
116 ctx.getSchemaPath().get());
117 verify(tl.put(ctx, created) == null);
119 return super.createEffective(ctx);
121 verify(tl.remove(ctx) == created);
126 TL_BUILDERS.remove();
132 protected ExtensionEffectiveStatement createEffective(
133 final StmtContext<QName, ExtensionStatement, ExtensionEffectiveStatement> ctx,
134 final ExtensionStatement declared,
135 final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
136 return finishCreate(ctx, substatements);
140 protected ExtensionEffectiveStatement createEmptyEffective(
141 final StmtContext<QName, ExtensionStatement, ExtensionEffectiveStatement> ctx,
142 final ExtensionStatement declared) {
143 return finishCreate(ctx, ImmutableList.of());
146 private static @NonNull ExtensionEffectiveStatement finishCreate(final StmtContext<?, ?, ?> ctx,
147 final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
148 final ExtensionEffectiveStatementImpl ret = verifyNotNull(verifyNotNull(TL_BUILDERS.get(),
149 "Statement build state not initialized").get(ctx), "No build state found for %s", ctx);
150 ret.setSubstatements(substatements);