Fix yang-data extension definition
[yangtools.git] / parser / rfc8040-parser-support / src / main / java / org / opendaylight / yangtools / rfc8040 / parser / YangDataEffectiveStatementImpl.java
1 /*
2  * Copyright (c) 2017 Pantheon Technologies, s.r.o. 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.rfc8040.parser;
9
10 import static com.google.common.base.Verify.verify;
11 import static java.util.Objects.requireNonNull;
12
13 import com.google.common.annotations.Beta;
14 import com.google.common.collect.ImmutableList;
15 import java.util.Map;
16 import java.util.Optional;
17 import org.eclipse.jdt.annotation.NonNull;
18 import org.opendaylight.yangtools.rfc8040.model.api.YangDataEffectiveStatement;
19 import org.opendaylight.yangtools.rfc8040.model.api.YangDataSchemaNode;
20 import org.opendaylight.yangtools.rfc8040.model.api.YangDataStatement;
21 import org.opendaylight.yangtools.yang.common.QName;
22 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
23 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
24 import org.opendaylight.yangtools.yang.model.api.meta.IdentifierNamespace;
25 import org.opendaylight.yangtools.yang.model.api.stmt.DataTreeAwareEffectiveStatement;
26 import org.opendaylight.yangtools.yang.model.api.stmt.DataTreeEffectiveStatement;
27 import org.opendaylight.yangtools.yang.model.spi.meta.AbstractEffectiveUnknownSchmemaNode;
28 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.DataNodeContainerMixin;
29 import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
30
31 @Beta
32 final class YangDataEffectiveStatementImpl extends AbstractEffectiveUnknownSchmemaNode<String, YangDataStatement>
33         implements YangDataEffectiveStatement, YangDataSchemaNode, DataNodeContainerMixin<String, YangDataStatement> {
34     private final @NonNull DataSchemaNode child;
35
36     YangDataEffectiveStatementImpl(final Current<String, YangDataStatement> stmt,
37              final ImmutableList<? extends EffectiveStatement<?, ?>> substatements, final DataSchemaNode child) {
38         super(stmt.declared(), stmt.argument(), stmt.history(), substatements);
39         this.child = requireNonNull(child);
40     }
41
42     @Override
43     public QName getQName() {
44         return child.getQName();
45     }
46
47     @Override
48     public DataSchemaNode dataChildByName(final QName name) {
49         return name.equals(child.getQName()) ? child : null;
50     }
51
52     @Override
53     public YangDataEffectiveStatement asEffectiveStatement() {
54         return this;
55     }
56
57     @Override
58     protected <K, V, N extends IdentifierNamespace<K, V>> Optional<? extends Map<K, V>> getNamespaceContents(
59             final Class<N> namespace) {
60         if (SchemaTreeNamespace.class.equals(namespace)) {
61             return castChild();
62         }
63         if (DataTreeNamespace.class.equals(namespace)) {
64             if (child instanceof DataTreeEffectiveStatement) {
65                 return castChild();
66             }
67
68             // A schema tree statement which *has to* know about data tree -- just forward it
69             verify(child instanceof DataTreeAwareEffectiveStatement, "Unexpected child %s", child);
70             return Optional.of(((DataTreeAwareEffectiveStatement<?, ?>) child).getAll(namespace));
71         }
72         return super.getNamespaceContents(namespace);
73     }
74
75     @SuppressWarnings("unchecked")
76     private <K, V> Optional<Map<K, V>> castChild() {
77         return Optional.of((Map<K, V>) Map.of(child.getQName(), child));
78     }
79 }