BUG-4456: add RecursiveExtensionResolver
[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 org.opendaylight.yangtools.yang.common.QName;
11 import org.opendaylight.yangtools.yang.model.api.Rfc6020Mapping;
12 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
13 import org.opendaylight.yangtools.yang.model.api.stmt.ArgumentStatement;
14 import org.opendaylight.yangtools.yang.model.api.stmt.DescriptionStatement;
15 import org.opendaylight.yangtools.yang.model.api.stmt.ExtensionStatement;
16 import org.opendaylight.yangtools.yang.model.api.stmt.ReferenceStatement;
17 import org.opendaylight.yangtools.yang.model.api.stmt.StatusStatement;
18 import org.opendaylight.yangtools.yang.parser.spi.ExtensionNamespace;
19 import org.opendaylight.yangtools.yang.parser.spi.SubstatementValidator;
20 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractDeclaredStatement;
21 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
22 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
23 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
24 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.effective.ExtensionEffectiveStatementImpl;
25
26 public class ExtensionStatementImpl extends AbstractDeclaredStatement<QName> implements ExtensionStatement {
27     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(Rfc6020Mapping
28             .EXTENSION)
29             .add(Rfc6020Mapping.ARGUMENT, 0, 1)
30             .add(Rfc6020Mapping.DESCRIPTION, 0, 1)
31             .add(Rfc6020Mapping.REFERENCE, 0, 1)
32             .add(Rfc6020Mapping.STATUS, 0, 1)
33             .build();
34
35     protected ExtensionStatementImpl(final StmtContext<QName, ExtensionStatement,?> context) {
36         super(context);
37     }
38
39     public static class Definition extends AbstractStatementSupport<QName,ExtensionStatement,EffectiveStatement<QName,ExtensionStatement>> {
40         public Definition() {
41             super(Rfc6020Mapping.EXTENSION);
42         }
43
44         @Override
45         public QName parseArgumentValue(final StmtContext<?,?,?> ctx, final String value) {
46             return Utils.qNameFromArgument(ctx, value);
47         }
48
49         @Override
50         public ExtensionStatement createDeclared(final StmtContext<QName, ExtensionStatement,?> ctx) {
51             return new ExtensionStatementImpl(ctx);
52         }
53
54         @Override
55         public EffectiveStatement<QName,ExtensionStatement> createEffective(
56                 final StmtContext<QName,ExtensionStatement ,EffectiveStatement<QName,ExtensionStatement>> ctx) {
57
58             // Look at the thread-local leak in case we are invoked recursively
59             final ExtensionEffectiveStatementImpl existing = RecursiveObjectLeaker.lookup(ctx,
60                 ExtensionEffectiveStatementImpl.class);
61             if (existing != null) {
62                 // Careful! this not fully initialized!
63                 return existing;
64             }
65
66             RecursiveObjectLeaker.beforeConstructor(ctx);
67             try {
68                 // This result is fine, we know it has been completely initialized
69                 return new ExtensionEffectiveStatementImpl(ctx);
70             } finally {
71                 RecursiveObjectLeaker.afterConstructor(ctx);
72             }
73         }
74
75         @Override
76         public void onStatementDefinitionDeclared(final StmtContext.Mutable<QName, ExtensionStatement, EffectiveStatement<QName, ExtensionStatement>> stmt) throws SourceException {
77             stmt.addContext(ExtensionNamespace.class, stmt.getStatementArgument(), stmt);
78         }
79
80         @Override
81         public void onFullDefinitionDeclared(final StmtContext.Mutable<QName, ExtensionStatement,
82                 EffectiveStatement<QName, ExtensionStatement>> stmt) {
83             super.onFullDefinitionDeclared(stmt);
84             SUBSTATEMENT_VALIDATOR.validate(stmt);
85         }
86     }
87
88     @Override
89     public StatusStatement getStatus() {
90         return firstDeclared(StatusStatement.class);
91     }
92
93     @Override
94     public DescriptionStatement getDescription() {
95         return firstDeclared(DescriptionStatement.class);
96     }
97
98     @Override
99     public ReferenceStatement getReference() {
100         return firstDeclared(ReferenceStatement.class);
101     }
102
103     @Override
104     public ArgumentStatement getArgument() {
105         return firstDeclared(ArgumentStatement.class);
106     }
107
108 }