2 * Copyright (c) 2015 Cisco Systems, Inc. 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.data.util;
10 import com.google.common.collect.FluentIterable;
11 import com.google.common.collect.ImmutableSet;
12 import java.util.HashSet;
13 import java.util.List;
14 import java.util.Optional;
16 import javax.annotation.Nullable;
17 import org.eclipse.jdt.annotation.NonNull;
18 import org.opendaylight.yangtools.concepts.Identifiable;
19 import org.opendaylight.yangtools.yang.common.QName;
20 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
21 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.AugmentationIdentifier;
22 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
23 import org.opendaylight.yangtools.yang.model.api.AnyXmlSchemaNode;
24 import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
25 import org.opendaylight.yangtools.yang.model.api.AugmentationTarget;
26 import org.opendaylight.yangtools.yang.model.api.CaseSchemaNode;
27 import org.opendaylight.yangtools.yang.model.api.ChoiceSchemaNode;
28 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
29 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
30 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
31 import org.opendaylight.yangtools.yang.model.api.LeafListSchemaNode;
32 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
33 import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
34 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
35 import org.opendaylight.yangtools.yang.model.api.SchemaNode;
36 import org.opendaylight.yangtools.yang.model.util.EffectiveAugmentationSchema;
39 * Schema derived data providing necessary information for mapping between
40 * {@link org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode} and serialization format defined in RFC6020,
41 * since the mapping is not one-to-one.
43 * @param <T> Path Argument type
45 public abstract class DataSchemaContextNode<T extends PathArgument> implements Identifiable<T> {
46 private final DataSchemaNode dataSchemaNode;
47 private final T identifier;
49 protected DataSchemaContextNode(final T identifier, final SchemaNode schema) {
50 this.identifier = identifier;
51 if (schema instanceof DataSchemaNode) {
52 this.dataSchemaNode = (DataSchemaNode) schema;
54 this.dataSchemaNode = null;
59 public T getIdentifier() {
63 public boolean isMixin() {
67 public boolean isKeyedEntry() {
71 public abstract boolean isLeaf();
73 protected Set<QName> getQNameIdentifiers() {
74 return ImmutableSet.of(identifier.getNodeType());
78 * Find a child node identifier by its {@link PathArgument}.
80 * @param child Child path argument
83 @Nullable public abstract DataSchemaContextNode<?> getChild(PathArgument child);
85 @Nullable public abstract DataSchemaContextNode<?> getChild(QName child);
87 @Nullable public DataSchemaNode getDataSchemaNode() {
88 return dataSchemaNode;
92 * Find a child node as identified by a {@link YangInstanceIdentifier} relative to this node.
94 * @param path Path towards the child node
95 * @return Child node if present, or empty when corresponding child is not found.
96 * @throws NullPointerException if {@code path} is null
98 public final @NonNull Optional<@NonNull DataSchemaContextNode<?>> findChild(
99 final @NonNull YangInstanceIdentifier path) {
100 DataSchemaContextNode<?> currentOp = this;
101 for (PathArgument arg : path.getPathArguments()) {
102 currentOp = currentOp.getChild(arg);
103 if (currentOp == null) {
104 return Optional.empty();
107 return Optional.of(currentOp);
110 static DataSchemaNode findChildSchemaNode(final DataNodeContainer parent, final QName child) {
111 DataSchemaNode potential = parent.getDataChildByName(child);
112 if (potential == null) {
113 Iterable<ChoiceSchemaNode> choices = FluentIterable.from(
114 parent.getChildNodes()).filter(ChoiceSchemaNode.class);
115 potential = findChoice(choices, child);
120 static DataSchemaContextNode<?> fromSchemaAndQNameChecked(final DataNodeContainer schema, final QName child) {
121 DataSchemaNode result = findChildSchemaNode(schema, child);
122 // We try to look up if this node was added by augmentation
123 if (result != null && schema instanceof DataSchemaNode && result.isAugmenting()) {
124 return fromAugmentation(schema, (AugmentationTarget) schema, result);
126 return fromDataSchemaNode(result);
129 // FIXME: this looks like it should be a Predicate on a stream with findFirst()
130 private static ChoiceSchemaNode findChoice(final Iterable<ChoiceSchemaNode> choices, final QName child) {
131 for (ChoiceSchemaNode choice : choices) {
132 // FIXME: this looks weird: what are we looking for again?
133 for (CaseSchemaNode caze : choice.getCases().values()) {
134 if (findChildSchemaNode(caze, child) != null) {
142 public static AugmentationIdentifier augmentationIdentifierFrom(final AugmentationSchemaNode augmentation) {
143 ImmutableSet.Builder<QName> potentialChildren = ImmutableSet.builder();
144 for (DataSchemaNode child : augmentation.getChildNodes()) {
145 potentialChildren.add(child.getQName());
147 return new AugmentationIdentifier(potentialChildren.build());
150 static DataNodeContainer augmentationProxy(final AugmentationSchemaNode augmentation,
151 final DataNodeContainer schema) {
152 Set<DataSchemaNode> children = new HashSet<>();
153 for (DataSchemaNode augNode : augmentation.getChildNodes()) {
154 children.add(schema.getDataChildByName(augNode.getQName()));
156 return new EffectiveAugmentationSchema(augmentation, children);
160 * Returns a DataContextNodeOperation for provided child node
163 * If supplied child is added by Augmentation this operation returns a
164 * DataContextNodeOperation for augmentation, otherwise returns a
165 * DataContextNodeOperation for child as call for
166 * {@link #fromDataSchemaNode(DataSchemaNode)}.
168 @Nullable static DataSchemaContextNode<?> fromAugmentation(final DataNodeContainer parent,
169 final AugmentationTarget parentAug, final DataSchemaNode child) {
170 AugmentationSchemaNode augmentation = null;
171 for (AugmentationSchemaNode aug : parentAug.getAvailableAugmentations()) {
172 DataSchemaNode potential = aug.getDataChildByName(child.getQName());
173 if (potential != null) {
178 if (augmentation != null) {
179 return new AugmentationContextNode(augmentation, parent);
181 return fromDataSchemaNode(child);
184 @Nullable public static DataSchemaContextNode<?> fromDataSchemaNode(final DataSchemaNode potential) {
185 if (potential instanceof ContainerSchemaNode) {
186 return new ContainerContextNode((ContainerSchemaNode) potential);
187 } else if (potential instanceof ListSchemaNode) {
188 return fromListSchemaNode((ListSchemaNode) potential);
189 } else if (potential instanceof LeafSchemaNode) {
190 return new LeafContextNode((LeafSchemaNode) potential);
191 } else if (potential instanceof ChoiceSchemaNode) {
192 return new ChoiceNodeContextNode((ChoiceSchemaNode) potential);
193 } else if (potential instanceof LeafListSchemaNode) {
194 return fromLeafListSchemaNode((LeafListSchemaNode) potential);
195 } else if (potential instanceof AnyXmlSchemaNode) {
196 return new AnyXmlContextNode((AnyXmlSchemaNode) potential);
201 private static DataSchemaContextNode<?> fromListSchemaNode(final ListSchemaNode potential) {
202 List<QName> keyDefinition = potential.getKeyDefinition();
203 if (keyDefinition == null || keyDefinition.isEmpty()) {
204 return new UnkeyedListMixinContextNode(potential);
206 if (potential.isUserOrdered()) {
207 return new OrderedMapMixinContextNode(potential);
209 return new UnorderedMapMixinContextNode(potential);
212 private static DataSchemaContextNode<?> fromLeafListSchemaNode(final LeafListSchemaNode potential) {
213 if (potential.isUserOrdered()) {
214 return new OrderedLeafListMixinContextNode(potential);
216 return new UnorderedLeafListMixinContextNode(potential);
219 public static DataSchemaContextNode<?> from(final SchemaContext ctx) {
220 return new ContainerContextNode(ctx);