Convert yang-data-impl to a JPMS module
[yangtools.git] / yang / yang-data-impl / src / main / java / org / opendaylight / yangtools / yang / data / impl / schema / tree / AbstractModifiedNodeBasedCandidateNode.java
index 4788029b8a3d1cd3367f8c65b3f35fcbc7cf06dd..25e44a74fe66935791c2d792f24811abc2633511 100644 (file)
@@ -1,36 +1,39 @@
 /*
- * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
+ * Copyright (c) 2015, 2016 Cisco Systems, Inc. and others.  All rights reserved.
  *
  * This program and the accompanying materials are made available under the
  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
  * and is available at http://www.eclipse.org/legal/epl-v10.html
  */
 package org.opendaylight.yangtools.yang.data.impl.schema.tree;
-import com.google.common.base.Function;
-import com.google.common.base.Optional;
-import com.google.common.base.Preconditions;
-import com.google.common.base.Verify;
+
+import static com.google.common.base.Verify.verifyNotNull;
+import static java.util.Objects.requireNonNull;
+
 import com.google.common.collect.Collections2;
+import com.google.common.collect.ImmutableList;
 import java.util.Collection;
-import javax.annotation.Nonnull;
-import javax.annotation.Nullable;
+import java.util.Optional;
+import org.eclipse.jdt.annotation.NonNull;
+import org.eclipse.jdt.annotation.Nullable;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
+import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNodeContainer;
 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidateNode;
+import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidateNodes;
 import org.opendaylight.yangtools.yang.data.api.schema.tree.ModificationType;
 import org.opendaylight.yangtools.yang.data.api.schema.tree.spi.TreeNode;
 
-
 abstract class AbstractModifiedNodeBasedCandidateNode implements DataTreeCandidateNode {
     private final ModifiedNode mod;
     private final TreeNode newMeta;
     private final TreeNode oldMeta;
 
-    protected AbstractModifiedNodeBasedCandidateNode(final ModifiedNode mod,
-            final TreeNode oldMeta, final TreeNode newMeta) {
+    protected AbstractModifiedNodeBasedCandidateNode(final ModifiedNode mod, final TreeNode oldMeta,
+            final TreeNode newMeta) {
         this.newMeta = newMeta;
         this.oldMeta = oldMeta;
-        this.mod = Preconditions.checkNotNull(mod);
+        this.mod = requireNonNull(mod);
     }
 
     protected final ModifiedNode getMod() {
@@ -45,73 +48,105 @@ abstract class AbstractModifiedNodeBasedCandidateNode implements DataTreeCandida
         return oldMeta;
     }
 
-    private static final TreeNode childMeta(final TreeNode parent, final PathArgument id) {
-        if (parent != null) {
-            return parent.getChild(id).orNull();
-        } else {
-            return null;
+    private static TreeNode childMeta(final TreeNode parent, final PathArgument id) {
+        return parent == null ? null : parent.getChild(id).orElse(null);
+    }
+
+    private static boolean canHaveChildren(final @Nullable TreeNode oldMeta, final @Nullable TreeNode newMeta) {
+        if (oldMeta != null) {
+            return oldMeta.getData() instanceof NormalizedNodeContainer;
+        }
+        if (newMeta != null) {
+            return newMeta.getData() instanceof NormalizedNodeContainer;
         }
+        return false;
     }
 
-    private DataTreeCandidateNode childNode(final ModifiedNode input) {
-        final PathArgument id = input.getIdentifier();
-        return createChildNode(input, childMeta(oldMeta, id), childMeta(newMeta, id));
+    @SuppressWarnings("unchecked")
+    private static NormalizedNodeContainer<?, PathArgument, NormalizedNode<?, ?>> getContainer(
+            final @Nullable TreeNode meta) {
+        return meta == null ? null : (NormalizedNodeContainer<?, PathArgument, NormalizedNode<?, ?>>)meta.getData();
     }
 
-    private static DataTreeCandidateNode createChildNode(@Nonnull final ModifiedNode input, @Nullable final TreeNode oldMeta, @Nullable final TreeNode newMeta) {
-        if(oldMeta == null) {
-            return RecursiveModificationCandidateNode.initialWriteNode(newMeta.getData());
-        }
-        if(newMeta == null) {
-            return RecursiveModificationCandidateNode.deleteNode(oldMeta.getData());
-        }
-        return new ChildNode(input, oldMeta, newMeta);
+    private ChildNode childNode(final ModifiedNode childMod) {
+        final PathArgument id = childMod.getIdentifier();
+        return new ChildNode(childMod, childMeta(oldMeta, id), childMeta(newMeta, id));
     }
 
     @Override
     public Collection<DataTreeCandidateNode> getChildNodes() {
-        return Collections2.transform(mod.getChildren(), new Function<ModifiedNode, DataTreeCandidateNode>() {
-            @Override
-            public DataTreeCandidateNode apply(final ModifiedNode input) {
-                return childNode(input);
-            }
-        });
+        switch (mod.getModificationType()) {
+            case APPEARED:
+            case DISAPPEARED:
+            case SUBTREE_MODIFIED:
+                return Collections2.transform(mod.getChildren(), this::childNode);
+            case UNMODIFIED:
+                // Unmodified node, but we still need to resolve potential children. canHaveChildren returns
+                // false if both arguments are null.
+                if (!canHaveChildren(oldMeta, newMeta)) {
+                    return ImmutableList.of();
+                }
+
+                return Collections2.transform(getContainer(newMeta != null ? newMeta : oldMeta).getValue(),
+                    DataTreeCandidateNodes::unmodified);
+            case DELETE:
+            case WRITE:
+                // This is unusual, the user is requesting we follow into an otherwise-terminal node.
+                // We need to fudge things based on before/after data to correctly fake the expectations.
+                if (!canHaveChildren(oldMeta, newMeta)) {
+                    return ImmutableList.of();
+                }
+                return DataTreeCandidateNodes.containerDelta(getContainer(oldMeta), getContainer(newMeta));
+            default:
+                throw new IllegalArgumentException("Unhandled modification type " + mod.getModificationType());
+        }
     }
 
     @Override
     public ModificationType getModificationType() {
-        return Verify.verifyNotNull(mod.getModificationType(), "Node %s does not have resolved modification type", mod);
+        return verifyNotNull(mod.getModificationType(), "Node %s does not have resolved modification type", mod);
     }
 
-    private Optional<NormalizedNode<?, ?>> optionalData(final TreeNode meta) {
-        if (meta != null) {
-            return Optional.<NormalizedNode<?,?>>of(meta.getData());
-        } else {
-            return Optional.absent();
-        }
+    private static @NonNull Optional<NormalizedNode<?, ?>> optionalData(final TreeNode meta) {
+        return meta == null ? Optional.empty() : Optional.of(meta.getData());
     }
 
     @Override
-    public Optional<NormalizedNode<?, ?>> getDataAfter() {
+    public final Optional<NormalizedNode<?, ?>> getDataAfter() {
         return optionalData(newMeta);
     }
 
     @Override
-    public Optional<NormalizedNode<?, ?>> getDataBefore() {
+    public final Optional<NormalizedNode<?, ?>> getDataBefore() {
         return optionalData(oldMeta);
     }
 
     @Override
-    public DataTreeCandidateNode getModifiedChild(final PathArgument identifier) {
-        final Optional<ModifiedNode> childMod = mod.getChild(identifier);
-        if (childMod.isPresent()) {
-            return childNode(childMod.get());
+    public final Optional<DataTreeCandidateNode> getModifiedChild(final PathArgument identifier) {
+        switch (mod.getModificationType()) {
+            case APPEARED:
+            case DISAPPEARED:
+            case SUBTREE_MODIFIED:
+                return mod.getChild(identifier).map(this::childNode);
+            case UNMODIFIED:
+                if (!canHaveChildren(oldMeta, newMeta)) {
+                    return Optional.empty();
+                }
+                return getContainer(newMeta != null ? newMeta : oldMeta).getChild(identifier)
+                        .map(DataTreeCandidateNodes::unmodified);
+            case DELETE:
+            case WRITE:
+                if (!canHaveChildren(oldMeta, newMeta)) {
+                    return Optional.empty();
+                }
+                return DataTreeCandidateNodes.containerDelta(getContainer(oldMeta), getContainer(newMeta), identifier);
+            default:
+                throw new IllegalArgumentException("Unhandled modification type " + mod.getModificationType());
         }
-        return null;
     }
 
-    static final class ChildNode extends AbstractModifiedNodeBasedCandidateNode {
-        public ChildNode(final ModifiedNode mod, final TreeNode oldMeta, final TreeNode newMeta) {
+    private static final class ChildNode extends AbstractModifiedNodeBasedCandidateNode {
+        ChildNode(final ModifiedNode mod, final TreeNode oldMeta, final TreeNode newMeta) {
             super(mod, oldMeta, newMeta);
         }
 
@@ -120,4 +155,10 @@ abstract class AbstractModifiedNodeBasedCandidateNode implements DataTreeCandida
             return getMod().getIdentifier();
         }
     }
-}
\ No newline at end of file
+
+    @Override
+    public String toString() {
+        return this.getClass().getSimpleName() + "{mod = " + this.mod + ", oldMeta = " + this.oldMeta + ", newMeta = "
+                + this.newMeta + "}";
+    }
+}