b016c748b828d43a264fd1fc691810f0ce83706d
[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / AbstractUndeclaredEffectiveStatement.java
1 /*
2  * Copyright (c) 2020 PANTHEON.tech, 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.yang.parser.rfc7950.stmt;
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 com.google.common.collect.ImmutableMap;
16 import java.util.Map;
17 import java.util.Optional;
18 import org.eclipse.jdt.annotation.NonNull;
19 import org.opendaylight.yangtools.yang.common.QName;
20 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
21 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
22 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
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.meta.StatementSource;
26 import org.opendaylight.yangtools.yang.model.api.stmt.DataTreeAwareEffectiveStatement;
27 import org.opendaylight.yangtools.yang.model.api.stmt.DataTreeEffectiveStatement;
28 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeAwareEffectiveStatement;
29 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeEffectiveStatement;
30 import org.opendaylight.yangtools.yang.parser.spi.source.StatementSourceReference;
31
32 @Beta
33 public abstract class AbstractUndeclaredEffectiveStatement<A, D extends DeclaredStatement<A>>
34         extends AbstractEffectiveStatement<A, D>  {
35     @Override
36     public final StatementSource getStatementSource() {
37         return StatementSource.CONTEXT;
38     }
39
40     @Override
41     public final D getDeclared() {
42         return null;
43     }
44
45     /**
46      * Base stateless superclass form {@link SchemaTreeAwareEffectiveStatement}s. It maintains the contents of schema
47      * tree namespace based of effective substatements.
48      *
49      * @param <A> Argument type ({@link Void} if statement does not have argument.)
50      * @param <D> Class representing declared version of this statement.
51      * @param <E> Class representing effective version of this statement.
52      */
53     public abstract static class WithSchemaTree<A, D extends DeclaredStatement<A>,
54             E extends SchemaTreeAwareEffectiveStatement<A, D>> extends AbstractUndeclaredEffectiveStatement<A, D> {
55         @Override
56         @SuppressWarnings("unchecked")
57         protected <K, V, N extends IdentifierNamespace<K, V>> Optional<? extends Map<K, V>> getNamespaceContents(
58                 final Class<N> namespace) {
59             if (SchemaTreeAwareEffectiveStatement.Namespace.class.equals(namespace)) {
60                 return Optional.of((Map<K, V>) schemaTreeNamespace());
61             }
62             return super.getNamespaceContents(namespace);
63         }
64
65         /**
66          * Indexing support for {@link DataNodeContainer#findDataChildByName(QName)}.
67          */
68         protected final Optional<DataSchemaNode> findDataSchemaNode(final QName name) {
69             // Only DataNodeContainer subclasses should be calling this method
70             verify(this instanceof DataNodeContainer);
71             final SchemaTreeEffectiveStatement<?> child = schemaTreeNamespace().get(requireNonNull(name));
72             return child instanceof DataSchemaNode ? Optional.of((DataSchemaNode) child) : Optional.empty();
73         }
74
75         protected abstract Map<QName, SchemaTreeEffectiveStatement<?>> schemaTreeNamespace();
76     }
77
78     /**
79      * Base stateless superclass form {@link DataTreeAwareEffectiveStatement}s. It maintains the contents of data tree
80      * namespace based of effective substatements.
81      *
82      * @param <A> Argument type ({@link Void} if statement does not have argument.)
83      * @param <D> Class representing declared version of this statement.
84      * @param <E> Class representing effective version of this statement.
85      */
86     public abstract static class WithDataTree<A, D extends DeclaredStatement<A>,
87             E extends DataTreeAwareEffectiveStatement<A, D>> extends WithSchemaTree<A, D, E> {
88         @Override
89         @SuppressWarnings("unchecked")
90         protected <K, V, N extends IdentifierNamespace<K, V>> Optional<? extends Map<K, V>> getNamespaceContents(
91                 final Class<N> namespace) {
92             if (DataTreeAwareEffectiveStatement.Namespace.class.equals(namespace)) {
93                 return Optional.of((Map<K, V>) dataTreeNamespace());
94             }
95             return super.getNamespaceContents(namespace);
96         }
97
98         protected abstract Map<QName, DataTreeEffectiveStatement<?>> dataTreeNamespace();
99     }
100
101     /**
102      * Stateful version of {@link WithSchemaTree}. Schema tree namespace is eagerly instantiated (and checked).
103      *
104      * @param <A> Argument type ({@link Void} if statement does not have argument.)
105      * @param <D> Class representing declared version of this statement.
106      * @param <E> Class representing effective version of this statement.
107      */
108     public abstract static class DefaultWithSchemaTree<A, D extends DeclaredStatement<A>,
109             E extends SchemaTreeAwareEffectiveStatement<A, D>> extends WithSchemaTree<A, D, E> {
110         public abstract static class WithSubstatements<A, D extends DeclaredStatement<A>,
111                 E extends SchemaTreeAwareEffectiveStatement<A, D>> extends DefaultWithSchemaTree<A, D, E> {
112             private final @NonNull Object substatements;
113
114             protected WithSubstatements(final ImmutableList<? extends EffectiveStatement<?, ?>> substatements,
115                     final StatementSourceReference ref) {
116                 super(substatements, ref);
117                 this.substatements = maskList(substatements);
118             }
119
120             @Override
121             public final ImmutableList<? extends EffectiveStatement<?, ?>> effectiveSubstatements() {
122                 return unmaskList(substatements);
123             }
124         }
125
126         private final @NonNull ImmutableMap<QName, SchemaTreeEffectiveStatement<?>> schemaTree;
127
128         protected DefaultWithSchemaTree(final ImmutableList<? extends EffectiveStatement<?, ?>> substatements,
129                 final StatementSourceReference ref) {
130             this.schemaTree = ImmutableMap.copyOf(createSchemaTreeNamespace(ref, substatements));
131         }
132
133         @Override
134         protected final Map<QName, SchemaTreeEffectiveStatement<?>> schemaTreeNamespace() {
135             return schemaTree;
136         }
137     }
138
139     /**
140      * Stateful version of {@link WithDataTree}. Schema tree and data tree namespaces are eagerly instantiated
141      * (and checked).
142      *
143      * @param <A> Argument type ({@link Void} if statement does not have argument.)
144      * @param <D> Class representing declared version of this statement.
145      * @param <E> Class representing effective version of this statement.
146      */
147     public abstract static class DefaultWithDataTree<A, D extends DeclaredStatement<A>,
148             E extends DataTreeAwareEffectiveStatement<A, D>> extends WithDataTree<A, D, E> {
149         public abstract static class WithSubstatements<A, D extends DeclaredStatement<A>,
150                 E extends DataTreeAwareEffectiveStatement<A, D>> extends DefaultWithDataTree<A, D, E> {
151             private final @NonNull Object substatements;
152
153             protected WithSubstatements(final ImmutableList<? extends EffectiveStatement<?, ?>> substatements,
154                     final StatementSourceReference ref) {
155                 super(substatements, ref);
156                 this.substatements = maskList(substatements);
157             }
158
159             @Override
160             public final ImmutableList<? extends EffectiveStatement<?, ?>> effectiveSubstatements() {
161                 return unmaskList(substatements);
162             }
163         }
164
165         private final @NonNull ImmutableMap<QName, SchemaTreeEffectiveStatement<?>> schemaTree;
166         private final @NonNull ImmutableMap<QName, DataTreeEffectiveStatement<?>> dataTree;
167
168         protected DefaultWithDataTree(final ImmutableList<? extends EffectiveStatement<?, ?>> substatements,
169                 final StatementSourceReference ref) {
170             final Map<QName, SchemaTreeEffectiveStatement<?>> schema = createSchemaTreeNamespace(ref, substatements);
171             this.schemaTree = ImmutableMap.copyOf(schema);
172             this.dataTree = createDataTreeNamespace(ref, schema.values(), schemaTree);
173         }
174
175         @Override
176         protected final Map<QName, SchemaTreeEffectiveStatement<?>> schemaTreeNamespace() {
177             return schemaTree;
178         }
179
180         @Override
181         protected final Map<QName, DataTreeEffectiveStatement<?>> dataTreeNamespace() {
182             return dataTree;
183         }
184     }
185 }