Improve ImmutableMapEntryNodeBuilder identifier handling

[yangtools.git] / yang / yang-data-impl / src / main / java / org / opendaylight / yangtools / yang / data / impl / schema / builder / impl / ImmutableMapEntryNodeBuilder.java

diff --git a/yang/yang-data-impl/src/main/java/org/opendaylight/yangtools/yang/data/impl/schema/builder/impl/ImmutableMapEntryNodeBuilder.java b/yang/yang-data-impl/src/main/java/org/opendaylight/yangtools/yang/data/impl/schema/builder/impl/ImmutableMapEntryNodeBuilder.java
index af8368f902280c940031ac2c2ae04655446bc442..44d6930e2cfdda30088fdf820d9cada6b9f08096 100644 (file)
--- a/yang/yang-data-impl/src/main/java/org/opendaylight/yangtools/yang/data/impl/schema/builder/impl/ImmutableMapEntryNodeBuilder.java
+++ b/yang/yang-data-impl/src/main/java/org/opendaylight/yangtools/yang/data/impl/schema/builder/impl/ImmutableMapEntryNodeBuilder.java
@@ -43,7 +43,7 @@ public class ImmutableMapEntryNodeBuilder
     protected ImmutableMapEntryNodeBuilder(final ImmutableMapEntryNode node) {
         super(node);
         this.childrenQNamesToPaths = new LinkedHashMap<>();
-        fillQnames(node.body(), childrenQNamesToPaths);
+        fillQNames(node.body(), childrenQNamesToPaths);
     }
 
     public static @NonNull DataContainerNodeBuilder<NodeIdentifierWithPredicates, MapEntryNode> create() {
@@ -64,38 +64,31 @@ public class ImmutableMapEntryNodeBuilder
         return new ImmutableMapEntryNodeBuilder((ImmutableMapEntryNode)node);
     }
 
-    private static void fillQnames(final Iterable<DataContainerChild> iterable, final Map<QName, PathArgument> out) {
-        for (final DataContainerChild childId : iterable) {
-            final PathArgument identifier = childId.getIdentifier();
-
-            // Augmentation nodes cannot be keys, and do not have to be present in childrenQNamesToPaths map
-            if (isAugment(identifier)) {
-                continue;
-            }
+    private static void fillQNames(final Iterable<DataContainerChild> iterable, final Map<QName, PathArgument> out) {
+        for (final DataContainerChild child : iterable) {
+            putQName(out, child);
+        }
+    }
 
-            out.put(childId.getNodeType(), identifier);
+    private static void putQName(final Map<QName, PathArgument> map, final DataContainerChild child) {
+        // Augmentation nodes cannot be keys, and do not have to be present in childrenQNamesToPaths map
+        final PathArgument identifier = child.getIdentifier();
+        if (!(identifier instanceof AugmentationIdentifier)) {
+            map.put(identifier.getNodeType(), identifier);
         }
     }
 
     @Override
     public DataContainerNodeBuilder<NodeIdentifierWithPredicates, MapEntryNode> withValue(
             final Collection<DataContainerChild> withValue) {
-        fillQnames(withValue, childrenQNamesToPaths);
+        fillQNames(withValue, childrenQNamesToPaths);
         return super.withValue(withValue);
     }
 
-    private static boolean isAugment(final PathArgument identifier) {
-        return identifier instanceof AugmentationIdentifier;
-    }
-
     @Override
     public DataContainerNodeBuilder<NodeIdentifierWithPredicates, MapEntryNode> withChild(
             final DataContainerChild child) {
-        // Augmentation nodes cannot be keys, and do not have to be present in childrenQNamesToPaths map
-        if (!isAugment(child.getIdentifier())) {
-            childrenQNamesToPaths.put(child.getNodeType(), child.getIdentifier());
-        }
-
+        putQName(childrenQNamesToPaths, child);
         return super.withChild(child);
     }