Update StoreTreeNode methods

[yangtools.git] / yang / yang-data-api / src / main / java / org / opendaylight / yangtools / yang / data / api / schema / tree / spi / LazyMutableContainerNode.java

diff --git a/yang/yang-data-api/src/main/java/org/opendaylight/yangtools/yang/data/api/schema/tree/spi/LazyMutableContainerNode.java b/yang/yang-data-api/src/main/java/org/opendaylight/yangtools/yang/data/api/schema/tree/spi/LazyMutableContainerNode.java
index 437edeed8a324852bd8fc46489fb4bcd4caa1cb3..e0b3e0198e3fd7c21b2ee861107717b110b9237b 100644 (file)
--- a/yang/yang-data-api/src/main/java/org/opendaylight/yangtools/yang/data/api/schema/tree/spi/LazyMutableContainerNode.java
+++ b/yang/yang-data-api/src/main/java/org/opendaylight/yangtools/yang/data/api/schema/tree/spi/LazyMutableContainerNode.java
@@ -8,7 +8,6 @@
 package org.opendaylight.yangtools.yang.data.api.schema.tree.spi;
 
 import java.util.Map;
-import java.util.Optional;
 import org.opendaylight.yangtools.util.MapAdaptor;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
 
@@ -26,12 +25,8 @@ final class LazyMutableContainerNode extends AbstractMutableContainerNode {
     }
 
     @Override
-    public Optional<TreeNode> getChild(final PathArgument childId) {
-        final TreeNode modified = getModifiedChild(childId);
-        if (modified != null) {
-            return Optional.of(modified);
-        }
-
-        return Optional.ofNullable(AbstractContainerNode.getChildFromData(getData(), childId, getVersion()));
+    public TreeNode childByArg(final PathArgument arg) {
+        final TreeNode modified = getModifiedChild(arg);
+        return modified != null ? modified : AbstractContainerNode.getChildFromData(getData(), arg, getVersion());
     }
 }