Remove NormalizedNode.getNodeType()
[yangtools.git] / yang / yang-data-impl / src / main / java / org / opendaylight / yangtools / yang / data / impl / leafref / LeafRefValidation.java
index 522bfe2152e2cc895edcd29e42bd60acb9e2a74c..5dc9938b69ff2da062d663d77d1355c21b5ecd36 100644 (file)
@@ -21,7 +21,6 @@ import java.util.Map;
 import java.util.Map.Entry;
 import java.util.Optional;
 import java.util.Set;
-import java.util.function.Consumer;
 import java.util.function.Predicate;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
@@ -54,15 +53,15 @@ public final class LeafRefValidation {
 
     private final Set<LeafRefContext> validatedLeafRefCtx = new HashSet<>();
     private final List<String> errorsMessages = new ArrayList<>();
-    private final NormalizedNode<?, ?> root;
+    private final NormalizedNode root;
 
-    private LeafRefValidation(final NormalizedNode<?, ?> root) {
+    private LeafRefValidation(final NormalizedNode root) {
         this.root = root;
     }
 
     public static void validate(final DataTreeCandidate tree, final LeafRefContext rootLeafRefCtx)
             throws LeafRefDataValidationFailedException {
-        final Optional<NormalizedNode<?, ?>> root = tree.getRootNode().getDataAfter();
+        final Optional<NormalizedNode> root = tree.getRootNode().getDataAfter();
         if (root.isPresent()) {
             new LeafRefValidation(root.get()).validateChildren(rootLeafRefCtx, tree.getRootNode().getChildNodes());
         }
@@ -131,7 +130,7 @@ public final class LeafRefValidation {
         final QName childQName = childNode.getIdentifier().getNodeType();
         LeafRefContext childReferencingCtx = referencingCtx.getReferencingChildByName(childQName);
         if (childReferencingCtx == null) {
-            final NormalizedNode<?, ?> data = childNode.getDataAfter().get();
+            final NormalizedNode data = childNode.getDataAfter().get();
             if (data instanceof MapEntryNode || data instanceof UnkeyedListEntryNode) {
                 childReferencingCtx = referencingCtx;
             }
@@ -149,7 +148,7 @@ public final class LeafRefValidation {
         final QName childQName = childNode.getIdentifier().getNodeType();
         LeafRefContext childReferencedByCtx = referencedByCtx.getReferencedChildByName(childQName);
         if (childReferencedByCtx == null) {
-            final NormalizedNode<?, ?> data = childNode.getDataAfter().get();
+            final NormalizedNode data = childNode.getDataAfter().get();
             if (data instanceof MapEntryNode || data instanceof UnkeyedListEntryNode) {
                 childReferencedByCtx = referencedByCtx;
             }
@@ -158,30 +157,40 @@ public final class LeafRefValidation {
         return childReferencedByCtx;
     }
 
-    private void validateNodeData(final NormalizedNode<?, ?> node, final LeafRefContext referencedByCtx,
+    private void validateNodeData(final NormalizedNode node, final LeafRefContext referencedByCtx,
             final LeafRefContext referencingCtx, final ModificationType modificationType,
             final YangInstanceIdentifier current) {
-
         if (node instanceof LeafNode) {
-            final LeafNode<?> leaf = (LeafNode<?>) node;
-
-            if (referencedByCtx != null && referencedByCtx.isReferenced()) {
-                validateLeafRefTargetNodeData(leaf, referencedByCtx, modificationType);
-            }
-            if (referencingCtx != null && referencingCtx.isReferencing()) {
-                validateLeafRefNodeData(leaf, referencingCtx, modificationType, current);
-            }
-
-            return;
+            validateLeafNodeData((LeafNode<?>) node, referencedByCtx, referencingCtx, modificationType, current);
+        } else if (node instanceof LeafSetNode) {
+            validateLeafSetNodeData((LeafSetNode<?>) node, referencedByCtx, referencingCtx, modificationType, current);
+        } else if (node instanceof ChoiceNode) {
+            validateChoiceNodeData((ChoiceNode) node, referencedByCtx, referencingCtx, modificationType, current);
+        } else if (node instanceof DataContainerNode) {
+            validateDataContainerNodeData((DataContainerNode) node, referencedByCtx, referencingCtx, modificationType,
+                current);
+        } else if (node instanceof MapNode) {
+            validateMapNodeData((MapNode) node, referencedByCtx, referencingCtx, modificationType, current);
         }
+        // FIXME: check UnkeyedListNode case
+    }
 
-        if (node instanceof LeafSetNode) {
-            if (referencedByCtx == null && referencingCtx == null) {
-                return;
-            }
+    private void validateLeafNodeData(final LeafNode<?> node, final LeafRefContext referencedByCtx,
+            final LeafRefContext referencingCtx, final ModificationType modificationType,
+            final YangInstanceIdentifier current) {
+        if (referencedByCtx != null && referencedByCtx.isReferenced()) {
+            validateLeafRefTargetNodeData(node, referencedByCtx, modificationType);
+        }
+        if (referencingCtx != null && referencingCtx.isReferencing()) {
+            validateLeafRefNodeData(node, referencingCtx, modificationType, current);
+        }
+    }
 
-            final LeafSetNode<?> leafSet = (LeafSetNode<?>) node;
-            for (final NormalizedNode<?, ?> leafSetEntry : leafSet.getValue()) {
+    private void validateLeafSetNodeData(final LeafSetNode<?> node, final LeafRefContext referencedByCtx,
+            final LeafRefContext referencingCtx, final ModificationType modificationType,
+            final YangInstanceIdentifier current) {
+        if (referencedByCtx != null || referencingCtx != null) {
+            for (final NormalizedNode leafSetEntry : node.body()) {
                 if (referencedByCtx != null && referencedByCtx.isReferenced()) {
                     validateLeafRefTargetNodeData(leafSetEntry, referencedByCtx, modificationType);
                 }
@@ -189,141 +198,102 @@ public final class LeafRefValidation {
                     validateLeafRefNodeData(leafSetEntry, referencingCtx, modificationType, current);
                 }
             }
-
-            return;
         }
+    }
 
-        if (node instanceof ChoiceNode) {
-            final ChoiceNode choice = (ChoiceNode) node;
-            for (final DataContainerChild<? extends PathArgument, ?> dataContainerChild : choice.getValue()) {
-                final QName qname = dataContainerChild.getNodeType();
-
-                final LeafRefContext childReferencedByCtx;
-                if (referencedByCtx != null) {
-                    childReferencedByCtx = findReferencedByCtxUnderChoice(referencedByCtx, qname);
-                } else {
-                    childReferencedByCtx = null;
-                }
-
-                final LeafRefContext childReferencingCtx;
-                if (referencingCtx != null) {
-                    childReferencingCtx = findReferencingCtxUnderChoice(referencingCtx, qname);
-                } else {
-                    childReferencingCtx = null;
-                }
+    private void validateChoiceNodeData(final ChoiceNode node, final LeafRefContext referencedByCtx,
+            final LeafRefContext referencingCtx, final ModificationType modificationType,
+            final YangInstanceIdentifier current) {
+        for (final DataContainerChild child : node.body()) {
+            final QName qname = child.getIdentifier().getNodeType();
+            final LeafRefContext childReferencedByCtx = referencedByCtx == null ? null
+                    : findReferencedByCtxUnderChoice(referencedByCtx, qname);
+            final LeafRefContext childReferencingCtx = referencingCtx == null ? null
+                    : findReferencingCtxUnderChoice(referencingCtx, qname);
+            if (childReferencedByCtx != null || childReferencingCtx != null) {
+                validateNodeData(child, childReferencedByCtx, childReferencingCtx, modificationType,
+                    current.node(child.getIdentifier()));
+            }
+        }
+    }
 
-                if (childReferencedByCtx != null || childReferencingCtx != null) {
-                    final YangInstanceIdentifier childYangInstanceIdentifier = current
-                            .node(dataContainerChild.getIdentifier());
-                    validateNodeData(dataContainerChild, childReferencedByCtx,
-                            childReferencingCtx, modificationType, childYangInstanceIdentifier);
-                }
+    private void validateDataContainerNodeData(final DataContainerNode node, final LeafRefContext referencedByCtx,
+            final LeafRefContext referencingCtx, final ModificationType modificationType,
+            final YangInstanceIdentifier current) {
+        for (final DataContainerChild child : node.body()) {
+            if (child instanceof AugmentationNode) {
+                validateNodeData(child, referencedByCtx, referencingCtx, modificationType, current.node(
+                    child.getIdentifier()));
+                return;
             }
-        } else if (node instanceof DataContainerNode) {
-            final DataContainerNode<?> dataContainerNode = (DataContainerNode<?>) node;
 
-            for (final DataContainerChild<? extends PathArgument, ?> child : dataContainerNode.getValue()) {
+            validateChildNodeData(child, referencedByCtx, referencingCtx, modificationType, current);
+        }
+    }
+
+    private void validateMapNodeData(final MapNode node, final LeafRefContext referencedByCtx,
+            final LeafRefContext referencingCtx, final ModificationType modificationType,
+            final YangInstanceIdentifier current) {
+        for (final MapEntryNode mapEntry : node.asMap().values()) {
+            final YangInstanceIdentifier mapEntryIdentifier = current.node(mapEntry.getIdentifier());
+            for (final DataContainerChild child : mapEntry.body()) {
                 if (child instanceof AugmentationNode) {
-                    validateNodeData(child, referencedByCtx, referencingCtx, modificationType, current
-                        .node(child.getIdentifier()));
+                    validateNodeData(child, referencedByCtx, referencingCtx, modificationType, current.node(
+                        child.getIdentifier()));
                     return;
                 }
 
-                final QName qname = child.getNodeType();
-                final LeafRefContext childReferencedByCtx;
-                if (referencedByCtx != null) {
-                    childReferencedByCtx = referencedByCtx.getReferencedChildByName(qname);
-                } else {
-                    childReferencedByCtx = null;
-                }
-
-                final LeafRefContext childReferencingCtx;
-                if (referencingCtx != null) {
-                    childReferencingCtx = referencingCtx.getReferencingChildByName(qname);
-                } else {
-                    childReferencingCtx = null;
-                }
-
-                if (childReferencedByCtx != null || childReferencingCtx != null) {
-                    final YangInstanceIdentifier childYangInstanceIdentifier = current
-                            .node(child.getIdentifier());
-                    validateNodeData(child, childReferencedByCtx,
-                            childReferencingCtx, modificationType, childYangInstanceIdentifier);
-                }
-            }
-        } else if (node instanceof MapNode) {
-            final MapNode map = (MapNode) node;
-
-            for (final MapEntryNode mapEntry : map.getValue()) {
-                final YangInstanceIdentifier mapEntryYangInstanceIdentifier = current.node(mapEntry.getIdentifier());
-                for (final DataContainerChild<? extends PathArgument, ?> mapEntryNode : mapEntry.getValue()) {
-                    if (mapEntryNode instanceof AugmentationNode) {
-                        validateNodeData(mapEntryNode, referencedByCtx, referencingCtx, modificationType, current
-                            .node(mapEntryNode.getIdentifier()));
-                        return;
-                    }
-
-                    final QName qname = mapEntryNode.getNodeType();
-                    final LeafRefContext childReferencedByCtx;
-                    if (referencedByCtx != null) {
-                        childReferencedByCtx = referencedByCtx.getReferencedChildByName(qname);
-                    } else {
-                        childReferencedByCtx = null;
-                    }
-
-                    final LeafRefContext childReferencingCtx;
-                    if (referencingCtx != null) {
-                        childReferencingCtx = referencingCtx.getReferencingChildByName(qname);
-                    } else {
-                        childReferencingCtx = null;
-                    }
-
-                    if (childReferencedByCtx != null || childReferencingCtx != null) {
-                        validateNodeData(mapEntryNode, childReferencedByCtx, childReferencingCtx, modificationType,
-                                mapEntryYangInstanceIdentifier.node(mapEntryNode.getIdentifier()));
-                    }
-                }
+                validateChildNodeData(child, referencedByCtx, referencingCtx, modificationType, mapEntryIdentifier);
             }
         }
-        // FIXME: check UnkeyedListNode case
     }
 
-    private static LeafRefContext findReferencingCtxUnderChoice(
-            final LeafRefContext referencingCtx, final QName qname) {
+    private void validateChildNodeData(final DataContainerChild child, final LeafRefContext referencedByCtx,
+            final LeafRefContext referencingCtx, final ModificationType modificationType,
+            final YangInstanceIdentifier current) {
+        final QName qname = child.getIdentifier().getNodeType();
+        final LeafRefContext childReferencedByCtx = referencedByCtx == null ? null
+                : referencedByCtx.getReferencedChildByName(qname);
+        final LeafRefContext childReferencingCtx = referencingCtx == null ? null
+                : referencingCtx.getReferencingChildByName(qname);
+        if (childReferencedByCtx != null || childReferencingCtx != null) {
+            validateNodeData(child, childReferencedByCtx, childReferencingCtx, modificationType, current.node(
+                child.getIdentifier()));
+        }
+    }
 
+    private static LeafRefContext findReferencingCtxUnderChoice(final LeafRefContext referencingCtx,
+            final QName qname) {
         for (final LeafRefContext child : referencingCtx.getReferencingChilds().values()) {
             final LeafRefContext referencingChildByName = child.getReferencingChildByName(qname);
             if (referencingChildByName != null) {
                 return referencingChildByName;
             }
         }
-
         return null;
     }
 
-    private static LeafRefContext findReferencedByCtxUnderChoice(
-            final LeafRefContext referencedByCtx, final QName qname) {
-
+    private static LeafRefContext findReferencedByCtxUnderChoice(final LeafRefContext referencedByCtx,
+            final QName qname) {
         for (final LeafRefContext child : referencedByCtx.getReferencedByChilds().values()) {
             final LeafRefContext referencedByChildByName = child.getReferencedChildByName(qname);
             if (referencedByChildByName != null) {
                 return referencedByChildByName;
             }
         }
-
         return null;
     }
 
-    private void validateLeafRefTargetNodeData(final NormalizedNode<?, ?> leaf, final LeafRefContext
+    private void validateLeafRefTargetNodeData(final NormalizedNode leaf, final LeafRefContext
             referencedByCtx, final ModificationType modificationType) {
         if (!validatedLeafRefCtx.add(referencedByCtx)) {
             LOG.trace("Operation [{}] validate data of leafref TARGET node: name[{}] = value[{}] -> SKIP: Already "
-                    + "validated", modificationType, referencedByCtx.getNodeName(), leaf.getValue());
+                    + "validated", modificationType, referencedByCtx.getNodeName(), leaf.body());
             return;
         }
 
         LOG.trace("Operation [{}] validate data of leafref TARGET node: name[{}] = value[{}]", modificationType,
-            referencedByCtx.getNodeName(), leaf.getValue());
+            referencedByCtx.getNodeName(), leaf.body());
         final Set<LeafRefContext> leafRefs = referencedByCtx.getAllReferencedByLeafRefCtxs().values().stream()
                 .filter(LeafRefContext::isReferencing).collect(Collectors.toSet());
         if (leafRefs.isEmpty()) {
@@ -340,11 +310,11 @@ public final class LeafRefValidation {
 
                 LOG.debug("Invalid leafref value [{}] allowed values {} by validation of leafref TARGET node: {} path "
                         + "of invalid LEAFREF node: {} leafRef target path: {} {}", leafRefsValue,
-                        leafRefTargetNodeValues, leaf.getNodeType(), leafRefContext.getCurrentNodePath(),
+                        leafRefTargetNodeValues, leaf.getIdentifier(), leafRefContext.getCurrentNodePath(),
                         leafRefContext.getAbsoluteLeafRefTargetPath(), FAILED);
                 errorsMessages.add(String.format("Invalid leafref value [%s] allowed values %s by validation of leafref"
                         + " TARGET node: %s path of invalid LEAFREF node: %s leafRef target path: %s %s", leafRefsValue,
-                        leafRefTargetNodeValues, leaf.getNodeType(), leafRefContext.getCurrentNodePath(),
+                        leafRefTargetNodeValues, leaf.getIdentifier(), leafRefContext.getCurrentNodePath(),
                         leafRefContext.getAbsoluteLeafRefTargetPath(),
                         FAILED));
             });
@@ -355,42 +325,42 @@ public final class LeafRefValidation {
         return computeValues(root, createPath(context.getLeafRefNodePath()), null);
     }
 
-    private void validateLeafRefNodeData(final NormalizedNode<?, ?> leaf, final LeafRefContext referencingCtx,
+    private void validateLeafRefNodeData(final NormalizedNode leaf, final LeafRefContext referencingCtx,
             final ModificationType modificationType, final YangInstanceIdentifier current) {
         final Set<Object> values = computeValues(root, createPath(referencingCtx.getAbsoluteLeafRefTargetPath()),
             current);
-        if (values.contains(leaf.getValue())) {
+        if (values.contains(leaf.body())) {
             LOG.debug("Operation [{}] validate data of LEAFREF node: name[{}] = value[{}] {}", modificationType,
-                referencingCtx.getNodeName(), leaf.getValue(), SUCCESS);
+                referencingCtx.getNodeName(), leaf.body(), SUCCESS);
             return;
         }
 
         LOG.debug("Operation [{}] validate data of LEAFREF node: name[{}] = value[{}] {}", modificationType,
-            referencingCtx.getNodeName(), leaf.getValue(), FAILED);
+            referencingCtx.getNodeName(), leaf.body(), FAILED);
         LOG.debug("Invalid leafref value [{}] allowed values {} of LEAFREF node: {} leafRef target path: {}",
-            leaf.getValue(), values, leaf.getNodeType(), referencingCtx.getAbsoluteLeafRefTargetPath());
+            leaf.body(), values, leaf.getIdentifier(), referencingCtx.getAbsoluteLeafRefTargetPath());
         errorsMessages.add(String.format("Invalid leafref value [%s] allowed values %s of LEAFREF node: %s leafRef "
-                + "target path: %s", leaf.getValue(), values, leaf.getNodeType(),
+                + "target path: %s", leaf.body(), values, leaf.getIdentifier(),
                 referencingCtx.getAbsoluteLeafRefTargetPath()));
     }
 
-    private Set<Object> computeValues(final NormalizedNode<?, ?> node, final Deque<QNameWithPredicate> path,
+    private Set<Object> computeValues(final NormalizedNode node, final Deque<QNameWithPredicate> path,
             final YangInstanceIdentifier current) {
         final HashSet<Object> values = new HashSet<>();
         addValues(values, node, ImmutableList.of(), path, current);
         return values;
     }
 
-    private void addValues(final Set<Object> values, final NormalizedNode<?, ?> node,
+    private void addValues(final Set<Object> values, final NormalizedNode node,
             final List<QNamePredicate> nodePredicates, final Deque<QNameWithPredicate> path,
             final YangInstanceIdentifier current) {
         if (node instanceof ValueNode) {
-            values.add(node.getValue());
+            values.add(node.body());
             return;
         }
         if (node instanceof LeafSetNode<?>) {
-            for (final NormalizedNode<?, ?> entry : ((LeafSetNode<?>) node).getValue()) {
-                values.add(entry.getValue());
+            for (final NormalizedNode entry : ((LeafSetNode<?>) node).body()) {
+                values.add(entry.body());
             }
             return;
         }
@@ -400,32 +370,36 @@ public final class LeafRefValidation {
             return;
         }
 
-        final QName qname = next.getQName();
-        final PathArgument pathArgument = new NodeIdentifier(qname);
+        final PathArgument pathArgument = new NodeIdentifier(next.getQName());
         if (node instanceof DataContainerNode) {
-            final DataContainerNode<?> dataContainerNode = (DataContainerNode<?>) node;
-            final Optional<DataContainerChild<?, ?>> child = dataContainerNode.getChild(pathArgument);
-            if (child.isPresent()) {
-                addNextValues(values, child.get(), next.getQNamePredicates(), path, current);
-            } else {
-                forEachChoice(dataContainerNode,
-                    choice -> addValues(values, choice, next.getQNamePredicates(), path, current));
-            }
+            processChildNode(values, (DataContainerNode) node, pathArgument, next.getQNamePredicates(), path, current);
         } else if (node instanceof MapNode) {
-            Stream<MapEntryNode> entries = ((MapNode) node).getValue().stream();
+            Stream<MapEntryNode> entries = ((MapNode) node).body().stream();
             if (!nodePredicates.isEmpty() && current != null) {
                 entries = entries.filter(createMapEntryPredicate(nodePredicates, current));
             }
 
-            entries.forEach(mapEntryNode -> {
-                final Optional<DataContainerChild<?, ?>> child = mapEntryNode.getChild(pathArgument);
-                if (child.isPresent()) {
-                    addNextValues(values, child.get(), next.getQNamePredicates(), path, current);
-                } else {
-                    forEachChoice(mapEntryNode,
-                        choice -> addValues(values, choice, next.getQNamePredicates(), path, current));
+            entries.forEach(entry -> processChildNode(values, entry, pathArgument, next.getQNamePredicates(), path,
+                current));
+        }
+    }
+
+    private void processChildNode(final Set<Object> values, final DataContainerNode parent,
+            final PathArgument arg, final List<QNamePredicate> nodePredicates, final Deque<QNameWithPredicate> path,
+            final YangInstanceIdentifier current) {
+        final DataContainerChild child = parent.childByArg(arg);
+        if (child == null) {
+            // FIXME: YANGTOOLS-901. We have SchemaContext nearby, hence we should be able to cache how to get
+            //        to the leaf with with specified QName, without having to iterate through Choices/Augmentations.
+            //        That perhaps means we should not have QNameWithPredicates, but NodeIdentifierWithPredicates as
+            //        the path specification.
+            for (final DataContainerChild mixin : parent.body()) {
+                if (mixin instanceof AugmentationNode || mixin instanceof ChoiceNode) {
+                    addValues(values, mixin, nodePredicates, path, current);
                 }
-            });
+            }
+        } else {
+            addNextValues(values, child, nodePredicates, path, current);
         }
     }
 
@@ -438,7 +412,7 @@ public final class LeafRefValidation {
         }
 
         return mapEntry -> {
-            for (final Entry<QName, Object> entryKeyValue : mapEntry.getIdentifier().getKeyValues().entrySet()) {
+            for (final Entry<QName, Object> entryKeyValue : mapEntry.getIdentifier().entrySet()) {
                 final Set<?> allowedValues = keyValues.get(entryKeyValue.getKey());
                 if (allowedValues != null && !allowedValues.contains(entryKeyValue.getValue())) {
                     return false;
@@ -448,7 +422,7 @@ public final class LeafRefValidation {
         };
     }
 
-    private void addNextValues(final Set<Object> values, final NormalizedNode<?, ?> node,
+    private void addNextValues(final Set<Object> values, final NormalizedNode node,
             final List<QNamePredicate> nodePredicates, final Deque<QNameWithPredicate> path,
             final YangInstanceIdentifier current) {
         final QNameWithPredicate element = path.pop();
@@ -459,14 +433,6 @@ public final class LeafRefValidation {
         }
     }
 
-    private static void forEachChoice(final DataContainerNode<?> node, final Consumer<ChoiceNode> consumer) {
-        for (final DataContainerChild<? extends PathArgument, ?> child : node.getValue()) {
-            if (child instanceof ChoiceNode) {
-                consumer.accept((ChoiceNode) child);
-            }
-        }
-    }
-
     private Set<?> getPathKeyExpressionValues(final LeafRefPath predicatePathKeyExpression,
             final YangInstanceIdentifier current) {
         return findParentNode(Optional.of(root), current).map(parent -> {
@@ -476,9 +442,9 @@ public final class LeafRefValidation {
         }).orElse(ImmutableSet.of());
     }
 
-    private static Optional<NormalizedNode<?, ?>> findParentNode(
-            final Optional<NormalizedNode<?, ?>> root, final YangInstanceIdentifier path) {
-        Optional<NormalizedNode<?, ?>> currentNode = root;
+    private static Optional<NormalizedNode> findParentNode(
+            final Optional<NormalizedNode> root, final YangInstanceIdentifier path) {
+        Optional<NormalizedNode> currentNode = root;
         final Iterator<PathArgument> pathIterator = path.getPathArguments().iterator();
         while (pathIterator.hasNext()) {
             final PathArgument childPathArgument = pathIterator.next();