BUG-6757: revert fix for BUG-4456
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / rfc6020 / ExtensionStatementImpl.java
1 /*
2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
3  *
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
7  */
8 package org.opendaylight.yangtools.yang.parser.stmt.rfc6020;
9
10 import java.util.HashSet;
11 import java.util.Set;
12 import org.opendaylight.yangtools.yang.common.QName;
13 import org.opendaylight.yangtools.yang.model.api.Rfc6020Mapping;
14 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
15 import org.opendaylight.yangtools.yang.model.api.stmt.ArgumentStatement;
16 import org.opendaylight.yangtools.yang.model.api.stmt.DescriptionStatement;
17 import org.opendaylight.yangtools.yang.model.api.stmt.ExtensionStatement;
18 import org.opendaylight.yangtools.yang.model.api.stmt.ReferenceStatement;
19 import org.opendaylight.yangtools.yang.model.api.stmt.StatusStatement;
20 import org.opendaylight.yangtools.yang.parser.spi.ExtensionNamespace;
21 import org.opendaylight.yangtools.yang.parser.spi.SubstatementValidator;
22 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractDeclaredStatement;
23 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
24 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
25 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
26 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.effective.ExtensionEffectiveStatementImpl;
27
28 public class ExtensionStatementImpl extends AbstractDeclaredStatement<QName> implements ExtensionStatement {
29     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(Rfc6020Mapping
30             .EXTENSION)
31             .add(Rfc6020Mapping.ARGUMENT, 0, 1)
32             .add(Rfc6020Mapping.DESCRIPTION, 0, 1)
33             .add(Rfc6020Mapping.REFERENCE, 0, 1)
34             .add(Rfc6020Mapping.STATUS, 0, 1)
35             .build();
36
37     protected ExtensionStatementImpl(final StmtContext<QName, ExtensionStatement,?> context) {
38         super(context);
39     }
40
41     public static class Definition extends AbstractStatementSupport<QName,ExtensionStatement,EffectiveStatement<QName,ExtensionStatement>> {
42         private static final ThreadLocal<Set<StmtContext<?, ?, ?>>> BUILDING = new ThreadLocal<>();
43
44         public Definition() {
45             super(Rfc6020Mapping.EXTENSION);
46         }
47
48         @Override
49         public QName parseArgumentValue(final StmtContext<?,?,?> ctx, final String value) {
50             return Utils.qNameFromArgument(ctx, value);
51         }
52
53         @Override
54         public ExtensionStatement createDeclared(final StmtContext<QName, ExtensionStatement,?> ctx) {
55             return new ExtensionStatementImpl(ctx);
56         }
57
58         @Override
59         public EffectiveStatement<QName,ExtensionStatement> createEffective(
60                 final StmtContext<QName,ExtensionStatement, EffectiveStatement<QName,ExtensionStatement>> ctx) {
61             Set<StmtContext<?, ?, ?>> building = BUILDING.get();
62             if (building == null) {
63                 building = new HashSet<>();
64                 BUILDING.set(building);
65             }
66
67             SourceException.throwIf(building.contains(ctx), ctx.getStatementSourceReference(),
68                 "Extension %s references itself", ctx.getStatementArgument());
69
70             building.add(ctx);
71             try {
72                 return new ExtensionEffectiveStatementImpl(ctx);
73             } finally {
74                 building.remove(ctx);
75                 if (building.isEmpty()) {
76                     BUILDING.remove();
77                 }
78             }
79         }
80
81         @Override
82         public void onStatementDefinitionDeclared(final StmtContext.Mutable<QName, ExtensionStatement, EffectiveStatement<QName, ExtensionStatement>> stmt) {
83             stmt.addContext(ExtensionNamespace.class, stmt.getStatementArgument(), stmt);
84         }
85
86         @Override
87         public void onFullDefinitionDeclared(final StmtContext.Mutable<QName, ExtensionStatement,
88                 EffectiveStatement<QName, ExtensionStatement>> stmt) {
89             super.onFullDefinitionDeclared(stmt);
90             SUBSTATEMENT_VALIDATOR.validate(stmt);
91         }
92     }
93
94     @Override
95     public StatusStatement getStatus() {
96         return firstDeclared(StatusStatement.class);
97     }
98
99     @Override
100     public DescriptionStatement getDescription() {
101         return firstDeclared(DescriptionStatement.class);
102     }
103
104     @Override
105     public ReferenceStatement getReference() {
106         return firstDeclared(ReferenceStatement.class);
107     }
108
109     @Override
110     public ArgumentStatement getArgument() {
111         return firstDeclared(ArgumentStatement.class);
112     }
113
114 }