Push out YangInstanceIdentifier FIXMEs
[yangtools.git] / yang / yang-data-api / src / main / java / org / opendaylight / yangtools / yang / data / api / YangInstanceIdentifier.java
index 127db2ddbf0f836ac20e88a92175baa8a7f0971f..ffcf3f34670fd933e5530496f5ada13cbb347b1b 100644 (file)
@@ -7,33 +7,45 @@
  */
 package org.opendaylight.yangtools.yang.data.api;
 
+import static com.google.common.base.Preconditions.checkArgument;
+import static com.google.common.base.Verify.verify;
+import static java.util.Objects.requireNonNull;
+
 import com.google.common.annotations.Beta;
-import com.google.common.base.Optional;
-import com.google.common.base.Preconditions;
 import com.google.common.cache.CacheBuilder;
 import com.google.common.cache.CacheLoader;
 import com.google.common.cache.LoadingCache;
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableMap;
 import com.google.common.collect.ImmutableSet;
 import com.google.common.collect.Iterables;
+import com.google.common.collect.Sets;
+import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
 import java.io.Serializable;
 import java.lang.reflect.Array;
+import java.util.AbstractMap.SimpleImmutableEntry;
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collection;
+import java.util.Deque;
 import java.util.Iterator;
 import java.util.List;
 import java.util.Map;
 import java.util.Map.Entry;
 import java.util.Objects;
+import java.util.Optional;
 import java.util.Set;
 import java.util.concurrent.atomic.AtomicReferenceFieldUpdater;
-import javax.annotation.Nonnull;
-import javax.annotation.Nullable;
+import java.util.function.Function;
+import org.eclipse.jdt.annotation.NonNull;
+import org.eclipse.jdt.annotation.Nullable;
 import org.opendaylight.yangtools.concepts.Builder;
 import org.opendaylight.yangtools.concepts.Immutable;
 import org.opendaylight.yangtools.concepts.Path;
 import org.opendaylight.yangtools.util.HashCodeBuilder;
 import org.opendaylight.yangtools.util.ImmutableOffsetMap;
 import org.opendaylight.yangtools.util.SharedSingletonMap;
+import org.opendaylight.yangtools.util.SingletonSet;
 import org.opendaylight.yangtools.yang.common.QName;
 import org.opendaylight.yangtools.yang.common.QNameModule;
 import org.opendaylight.yangtools.yang.data.api.schema.LeafSetEntryNode;
@@ -46,18 +58,18 @@ import org.opendaylight.yangtools.yang.data.api.schema.LeafSetEntryNode;
  * which conceptually is XPath expression minimized to uniquely identify element
  * in data tree which conforms to constraints maintained by YANG Model,
  * effectively this makes Instance Identifier a path to element in data tree.
- * </p>
+ *
  * <p>
  * Constraints put in YANG specification on instance-identifier allowed it to be
  * effectively represented in Java and it's evaluation does not require
  * full-blown XPath processor.
- * </p>
- * <h3>Path Arguments</h3>
+ *
  * <p>
+ * <h3>Path Arguments</h3>
  * Path to the node represented in instance identifier consists of
  * {@link PathArgument} which carries necessary information to uniquely identify
  * node on particular level in the subtree.
- * </p>
+ *
  * <ul>
  * <li>{@link NodeIdentifier} - Identifier of node, which has cardinality
  * <code>0..1</code> in particular subtree in data tree.</li>
@@ -69,16 +81,12 @@ import org.opendaylight.yangtools.yang.data.api.schema.LeafSetEntryNode;
  * <code>augmentation</code> node.</li>
  * </ul>
  *
- *
  * @see <a href="http://tools.ietf.org/html/rfc6020#section-9.13">RFC6020</a>
  */
+// FIXME: 6.0.0: this concept needs to be moved to yang-common, as parser components need the ability to refer
+//               to data nodes -- most notably XPath expressions and {@code default} statement arguments need to be able
+//               to represent these.
 public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentifier>, Immutable, Serializable {
-    /**
-     * An empty {@link YangInstanceIdentifier}. It corresponds to the path of the conceptual
-     * root of the YANG namespace.
-     */
-    public static final YangInstanceIdentifier EMPTY = FixedYangInstanceIdentifier.EMPTY_INSTANCE;
-
     private static final AtomicReferenceFieldUpdater<YangInstanceIdentifier, String> TOSTRINGCACHE_UPDATER =
             AtomicReferenceFieldUpdater.newUpdater(YangInstanceIdentifier.class, String.class, "toStringCache");
     private static final long serialVersionUID = 4L;
@@ -91,13 +99,25 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
         this.hash = hash;
     }
 
-    @Nonnull abstract YangInstanceIdentifier createRelativeIdentifier(int skipFromRoot);
-    @Nonnull abstract Collection<PathArgument> tryPathArguments();
-    @Nonnull abstract Collection<PathArgument> tryReversePathArguments();
+    /**
+     * Return An empty {@link YangInstanceIdentifier}. It corresponds to the path of the conceptual root of the YANG
+     * namespace.
+     *
+     * @return An empty YangInstanceIdentifier
+     */
+    public static @NonNull YangInstanceIdentifier empty() {
+        return FixedYangInstanceIdentifier.EMPTY_INSTANCE;
+    }
+
+    abstract @NonNull YangInstanceIdentifier createRelativeIdentifier(int skipFromRoot);
+
+    abstract @Nullable Collection<PathArgument> tryPathArguments();
+
+    abstract @Nullable Collection<PathArgument> tryReversePathArguments();
 
     /**
      * Check if this instance identifier has empty path arguments, e.g. it is
-     * empty and corresponds to {@link #EMPTY}.
+     * empty and corresponds to {@link #empty()}.
      *
      * @return True if this instance identifier is empty, false otherwise.
      */
@@ -110,15 +130,15 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
      * @return A optimized equivalent instance.
      */
     @Beta
-    public abstract YangInstanceIdentifier toOptimized();
+    public abstract @NonNull YangInstanceIdentifier toOptimized();
 
     /**
      * Return the conceptual parent {@link YangInstanceIdentifier}, which has
      * one item less in {@link #getPathArguments()}.
      *
-     * @return Parent {@link YangInstanceIdentifier}, or null if this object is {@link #EMPTY}.
+     * @return Parent {@link YangInstanceIdentifier}, or null if this object is {@link #empty()}.
      */
-    @Nullable public abstract YangInstanceIdentifier getParent();
+    public abstract @Nullable YangInstanceIdentifier getParent();
 
     /**
      * Return the ancestor {@link YangInstanceIdentifier} with a particular depth, e.g. number of path arguments.
@@ -127,14 +147,14 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
      * @return Ancestor {@link YangInstanceIdentifier}
      * @throws IllegalArgumentException if the specified depth is negative or is greater than the depth of this object.
      */
-   @Nonnull public abstract YangInstanceIdentifier getAncestor(int depth);
+    public abstract @NonNull YangInstanceIdentifier getAncestor(int depth);
 
     /**
      * Returns an ordered iteration of path arguments.
      *
      * @return Immutable iteration of path arguments.
      */
-    public abstract List<PathArgument> getPathArguments();
+    public abstract @NonNull List<PathArgument> getPathArguments();
 
     /**
      * Returns an iterable of path arguments in reverse order. This is useful
@@ -142,7 +162,7 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
      *
      * @return Immutable iterable of path arguments in reverse order.
      */
-    public abstract List<PathArgument> getReversePathArguments();
+    public abstract @NonNull List<PathArgument> getReversePathArguments();
 
     /**
      * Returns the last PathArgument. This is equivalent of iterating
@@ -152,9 +172,9 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
      */
     public abstract PathArgument getLastPathArgument();
 
-    public static YangInstanceIdentifier create(final Iterable<? extends PathArgument> path) {
+    public static @NonNull YangInstanceIdentifier create(final Iterable<? extends PathArgument> path) {
         if (Iterables.isEmpty(path)) {
-            return EMPTY;
+            return empty();
         }
 
         final HashCodeBuilder<PathArgument> hash = new HashCodeBuilder<>();
@@ -165,21 +185,42 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
         return FixedYangInstanceIdentifier.create(path, hash.build());
     }
 
-    public static YangInstanceIdentifier create(final PathArgument... path) {
+    public static @NonNull YangInstanceIdentifier create(final PathArgument... path) {
         // We are forcing a copy, since we cannot trust the user
         return create(Arrays.asList(path));
     }
 
-    @Override
-    public final int hashCode() {
-        /*
-         * The caching is safe, since the object contract requires
-         * immutability of the object and all objects referenced from this
-         * object.
-         * Used lists, maps are immutable. Path Arguments (elements) are also
-         * immutable, since the PathArgument contract requires immutability.
-         */
-        return hash;
+    /**
+     * Create a {@link YangInstanceIdentifier} by taking a snapshot of provided path and iterating it backwards.
+     *
+     * @param pathTowardsRoot Path towards root
+     * @return A {@link YangInstanceIdentifier} instance
+     * @throws NullPointerException if {@code pathTowardsRoot} or any of its members is null
+     */
+    public static @NonNull YangInstanceIdentifier createReverse(final Deque<PathArgument> pathTowardsRoot) {
+        final ImmutableList.Builder<PathArgument> builder = ImmutableList.builderWithExpectedSize(
+            pathTowardsRoot.size());
+        pathTowardsRoot.descendingIterator().forEachRemaining(builder::add);
+        return YangInstanceIdentifier.create(builder.build());
+    }
+
+    /**
+     * Create a {@link YangInstanceIdentifier} by walking specified stack backwards and extracting path components
+     * from it.
+     *
+     * @param stackTowardsRoot Stack towards root,
+     * @return A {@link YangInstanceIdentifier} instance
+     * @throws NullPointerException if {@code pathTowardsRoot} is null
+     */
+    public static <T> @NonNull YangInstanceIdentifier createReverse(final Deque<? extends T> stackTowardsRoot,
+            final Function<T, PathArgument> function) {
+        final ImmutableList.Builder<PathArgument> builder = ImmutableList.builderWithExpectedSize(
+            stackTowardsRoot.size());
+        final Iterator<? extends T> it = stackTowardsRoot.descendingIterator();
+        while (it.hasNext()) {
+            builder.add(function.apply(it.next()));
+        }
+        return YangInstanceIdentifier.create(builder.build());
     }
 
     boolean pathArgumentsEqual(final YangInstanceIdentifier other) {
@@ -203,23 +244,22 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
     }
 
     /**
-     * Constructs a new Instance Identifier with new {@link NodeIdentifier} added to the end of path arguments
+     * Constructs a new Instance Identifier with new {@link NodeIdentifier} added to the end of path arguments.
      *
      * @param name QName of {@link NodeIdentifier}
      * @return Instance Identifier with additional path argument added to the end.
      */
-    public final YangInstanceIdentifier node(final QName name) {
+    public final @NonNull YangInstanceIdentifier node(final QName name) {
         return node(new NodeIdentifier(name));
     }
 
     /**
-     *
-     * Constructs a new Instance Identifier with new {@link PathArgument} added to the end of path arguments
+     * Constructs a new Instance Identifier with new {@link PathArgument} added to the end of path arguments.
      *
      * @param arg Path argument which should be added to the end
      * @return Instance Identifier with additional path argument added to the end.
      */
-    public final YangInstanceIdentifier node(final PathArgument arg) {
+    public final @NonNull YangInstanceIdentifier node(final PathArgument arg) {
         return new StackedYangInstanceIdentifier(this, arg, HashCodeBuilder.nextHashCode(hash, arg));
     }
 
@@ -234,20 +274,20 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
      */
     public Optional<YangInstanceIdentifier> relativeTo(final YangInstanceIdentifier ancestor) {
         if (this == ancestor) {
-            return Optional.of(EMPTY);
+            return Optional.of(empty());
         }
         if (ancestor.isEmpty()) {
             return Optional.of(this);
         }
 
-        final Iterator<?> lit = getPathArguments().iterator();
-        final Iterator<?> oit = ancestor.getPathArguments().iterator();
+        final Iterator<PathArgument> lit = getPathArguments().iterator();
+        final Iterator<PathArgument> oit = ancestor.getPathArguments().iterator();
         int common = 0;
 
         while (oit.hasNext()) {
             // Ancestor is not really an ancestor
             if (!lit.hasNext() || !lit.next().equals(oit.next())) {
-                return Optional.absent();
+                return Optional.empty();
             }
 
             ++common;
@@ -257,7 +297,7 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
             return Optional.of(this);
         }
         if (!lit.hasNext()) {
-            return Optional.of(EMPTY);
+            return Optional.of(empty());
         }
 
         return Optional.of(createRelativeIdentifier(common));
@@ -269,9 +309,9 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
             return true;
         }
 
-        Preconditions.checkArgument(other != null, "other should not be null");
-        final Iterator<?> lit = getPathArguments().iterator();
-        final Iterator<?> oit = other.getPathArguments().iterator();
+        checkArgument(other != null, "other should not be null");
+        final Iterator<PathArgument> lit = getPathArguments().iterator();
+        final Iterator<PathArgument> oit = other.getPathArguments().iterator();
 
         while (lit.hasNext()) {
             if (!oit.hasNext()) {
@@ -315,6 +355,20 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
         return ret;
     }
 
+    @Override
+    public final int hashCode() {
+        /*
+         * The caching is safe, since the object contract requires
+         * immutability of the object and all objects referenced from this
+         * object.
+         * Used lists, maps are immutable. Path Arguments (elements) are also
+         * immutable, since the PathArgument contract requires immutability.
+         */
+        return hash;
+    }
+
+    @SuppressFBWarnings(value = "UPM_UNCALLED_PRIVATE_METHOD",
+            justification = "https://github.com/spotbugs/spotbugs/issues/811")
     private static int hashCode(final Object value) {
         if (value == null) {
             return 0;
@@ -337,15 +391,20 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
         return Objects.hashCode(value);
     }
 
+    final Object writeReplace() {
+        return new YIDv1(this);
+    }
+
     // Static factories & helpers
 
     /**
-     * Returns a new InstanceIdentifier with only one path argument of type {@link NodeIdentifier} with supplied QName
+     * Returns a new InstanceIdentifier with only one path argument of type {@link NodeIdentifier} with supplied
+     * QName.
      *
      * @param name QName of first node identifier
      * @return Instance Identifier with only one path argument of type {@link NodeIdentifier}
      */
-    public static YangInstanceIdentifier of(final QName name) {
+    public static @NonNull YangInstanceIdentifier of(final QName name) {
         return create(new NodeIdentifier(name));
     }
 
@@ -354,33 +413,34 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
      *
      * @return new builder for InstanceIdentifier with empty path arguments.
      */
-    public static InstanceIdentifierBuilder builder() {
+    public static @NonNull InstanceIdentifierBuilder builder() {
         return new YangInstanceIdentifierBuilder();
     }
 
     /**
-     *
      * Returns new builder for InstanceIdentifier with path arguments copied from original instance identifier.
      *
      * @param origin InstanceIdentifier from which path arguments are copied.
      * @return new builder for InstanceIdentifier with path arguments copied from original instance identifier.
      */
-    public static InstanceIdentifierBuilder builder(final YangInstanceIdentifier origin) {
+    public static @NonNull InstanceIdentifierBuilder builder(final YangInstanceIdentifier origin) {
         return new YangInstanceIdentifierBuilder(origin.getPathArguments(), origin.hashCode());
     }
 
     /**
-     * Path argument / component of InstanceIdentifier
-     *
+     * Path argument / component of InstanceIdentifier.
      * Path argument uniquely identifies node in data tree on particular
      * level.
+     *
      * <p>
      * This interface itself is used as common parent for actual
      * path arguments types and should not be implemented by user code.
+     *
      * <p>
      * Path arguments SHOULD contain only minimum of information
      * required to uniquely identify node on particular subtree level.
      *
+     * <p>
      * For actual path arguments types see:
      * <ul>
      * <li>{@link NodeIdentifier} - Identifier of container or leaf
@@ -391,15 +451,12 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
      */
     public interface PathArgument extends Comparable<PathArgument>, Immutable, Serializable {
         /**
-         * If applicable returns unique QName of data node as defined in YANG
-         * Schema.
-         *
-         * This method may return null, if the corresponding schema node, does
-         * not have QName associated, such as in cases of augmentations.
+         * Returns unique QName of data node as defined in YANG Schema, if available.
          *
          * @return Node type
+         * @throws UnsupportedOperationException if node type is not applicable, for example in case of an augmentation.
          */
-        QName getNodeType();
+        @NonNull QName getNodeType();
 
         /**
          * Return the string representation of this object for use in context
@@ -410,17 +467,16 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
          * @param previous Previous path argument
          * @return String representation
          */
-        String toRelativeString(PathArgument previous);
+        @NonNull String toRelativeString(PathArgument previous);
     }
 
-    private static abstract class AbstractPathArgument implements PathArgument {
+    private abstract static class AbstractPathArgument implements PathArgument {
         private static final long serialVersionUID = -4546547994250849340L;
-        private final QName nodeType;
-        private transient int hashValue;
-        private transient volatile boolean hashGuard = false;
+        private final @NonNull QName nodeType;
+        private transient volatile int hashValue;
 
         protected AbstractPathArgument(final QName nodeType) {
-            this.nodeType = Preconditions.checkNotNull(nodeType);
+            this.nodeType = requireNonNull(nodeType);
         }
 
         @Override
@@ -429,22 +485,19 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
         }
 
         @Override
+        @SuppressWarnings("checkstyle:parameterName")
         public int compareTo(final PathArgument o) {
             return nodeType.compareTo(o.getNodeType());
         }
 
         protected int hashCodeImpl() {
-            return 31 + getNodeType().hashCode();
+            return nodeType.hashCode();
         }
 
         @Override
         public final int hashCode() {
-            if (!hashGuard) {
-                hashValue = hashCodeImpl();
-                hashGuard = true;
-            }
-
-            return hashValue;
+            int local;
+            return (local = hashValue) != 0 ? local : (hashValue = hashCodeImpl());
         }
 
         @Override
@@ -467,7 +520,7 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
         @Override
         public String toRelativeString(final PathArgument previous) {
             if (previous instanceof AbstractPathArgument) {
-                final QNameModule mod = ((AbstractPathArgument)previous).getNodeType().getModule();
+                final QNameModule mod = previous.getNodeType().getModule();
                 if (getNodeType().getModule().equals(mod)) {
                     return getNodeType().getLocalName();
                 }
@@ -475,6 +528,8 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
 
             return getNodeType().toString();
         }
+
+        abstract Object writeReplace();
     }
 
     /**
@@ -502,82 +557,291 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
          * @param node Node's QName
          * @return A {@link NodeIdentifier}
          */
-        public static NodeIdentifier create(final QName node) {
+        public static @NonNull NodeIdentifier create(final QName node) {
             return CACHE.getUnchecked(node);
         }
+
+        @Override
+        Object writeReplace() {
+            return new NIv1(this);
+        }
     }
 
     /**
      * Composite path argument identifying a {@link org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode} leaf
      * overall data tree.
      */
-    public static final class NodeIdentifierWithPredicates extends AbstractPathArgument {
-        private static final long serialVersionUID = -4787195606494761540L;
+    public abstract static class NodeIdentifierWithPredicates extends AbstractPathArgument {
+        @Beta
+        public static final class Singleton extends NodeIdentifierWithPredicates {
+            private static final long serialVersionUID = 1L;
+
+            private final @NonNull QName key;
+            private final @NonNull Object value;
+
+            Singleton(final QName node, final QName key, final Object value) {
+                super(node);
+                this.key = requireNonNull(key);
+                this.value = requireNonNull(value);
+            }
 
-        private final Map<QName, Object> keyValues;
+            @Override
+            public SingletonSet<Entry<QName, Object>> entrySet() {
+                return SingletonSet.of(singleEntry());
+            }
 
-        public NodeIdentifierWithPredicates(final QName node, final Map<QName, Object> keyValues) {
-            super(node);
-            // Retains ImmutableMap for empty maps. For larger sizes uses a shared key set.
-            this.keyValues = ImmutableOffsetMap.unorderedCopyOf(keyValues);
-        }
+            @Override
+            public SingletonSet<QName> keySet() {
+                return SingletonSet.of(key);
+            }
 
-        public NodeIdentifierWithPredicates(final QName node, final QName key, final Object value) {
-            super(node);
-            this.keyValues = SharedSingletonMap.unorderedOf(key, value);
-        }
+            @Override
+            public boolean containsKey(final QName qname) {
+                return key.equals(requireNonNull(qname));
+            }
 
-        public Map<QName, Object> getKeyValues() {
-            return keyValues;
-        }
+            @Override
+            public SingletonSet<Object> values() {
+                return SingletonSet.of(value);
+            }
 
-        @Override
-        protected int hashCodeImpl() {
-            final int prime = 31;
-            int result = super.hashCodeImpl();
-            result = prime * result;
+            @Override
+            public int size() {
+                return 1;
+            }
 
-            for (Entry<QName, Object> entry : keyValues.entrySet()) {
-                result += Objects.hashCode(entry.getKey()) + YangInstanceIdentifier.hashCode(entry.getValue());
+            @Override
+            public ImmutableMap<QName, Object> asMap() {
+                return ImmutableMap.of(key, value);
+            }
+
+            /**
+             * Return the single entry contained in this object. This is equivalent to
+             * {@code entrySet().iterator().next()}.
+             *
+             * @return A single entry.
+             */
+            public @NonNull Entry<QName, Object> singleEntry() {
+                return new SimpleImmutableEntry<>(key, value);
+            }
+
+            @Override
+            boolean equalMapping(final NodeIdentifierWithPredicates other) {
+                final Singleton single = (Singleton) other;
+                return key.equals(single.key) && Objects.deepEquals(value, single.value);
+            }
+
+            @Override
+            Object keyValue(final QName qname) {
+                return key.equals(qname) ? value : null;
             }
-            return result;
         }
 
-        @Override
-        public boolean equals(final Object obj) {
-            if (!super.equals(obj)) {
-                return false;
+        private static final class Regular extends NodeIdentifierWithPredicates {
+            private static final long serialVersionUID = 1L;
+
+            private final @NonNull Map<QName, Object> keyValues;
+
+            Regular(final QName node, final Map<QName, Object> keyValues) {
+                super(node);
+                this.keyValues = requireNonNull(keyValues);
             }
 
-            final Map<QName, Object> otherKeyValues = ((NodeIdentifierWithPredicates) obj).keyValues;
+            @Override
+            public Set<Entry<QName, Object>> entrySet() {
+                return keyValues.entrySet();
+            }
 
-            // TODO: benchmark to see if just calling equals() on the two maps is not faster
-            if (keyValues == otherKeyValues) {
-                return true;
+            @Override
+            public Set<QName> keySet() {
+                return keyValues.keySet();
             }
-            if (keyValues.size() != otherKeyValues.size()) {
-                return false;
+
+            @Override
+            public boolean containsKey(final QName qname) {
+                return keyValues.containsKey(requireNonNull(qname));
             }
 
-            for (Entry<QName, Object> entry : keyValues.entrySet()) {
-                if (!otherKeyValues.containsKey(entry.getKey())
-                        || !Objects.deepEquals(entry.getValue(), otherKeyValues.get(entry.getKey()))) {
+            @Override
+            public Collection<Object> values() {
+                return keyValues.values();
+            }
+
+            @Override
+            public int size() {
+                return keyValues.size();
+            }
+
+            @Override
+            public Map<QName, Object> asMap() {
+                return keyValues;
+            }
+
+            @Override
+            Object keyValue(final QName qname) {
+                return keyValues.get(qname);
+            }
 
+            @Override
+            boolean equalMapping(final NodeIdentifierWithPredicates other) {
+                final Map<QName, Object> otherKeyValues = ((Regular) other).keyValues;
+                // TODO: benchmark to see if just calling equals() on the two maps is not faster
+                if (keyValues == otherKeyValues) {
+                    return true;
+                }
+                if (keyValues.size() != otherKeyValues.size()) {
                     return false;
                 }
+
+                for (Entry<QName, Object> entry : entrySet()) {
+                    final Object otherValue = otherKeyValues.get(entry.getKey());
+                    if (otherValue == null || !Objects.deepEquals(entry.getValue(), otherValue)) {
+                        return false;
+                    }
+                }
+
+                return true;
             }
+        }
 
-            return true;
+        private static final long serialVersionUID = -4787195606494761540L;
+
+        NodeIdentifierWithPredicates(final QName node) {
+            super(node);
+        }
+
+        public static @NonNull NodeIdentifierWithPredicates of(final QName node) {
+            return new Regular(node, ImmutableMap.of());
+        }
+
+        public static @NonNull NodeIdentifierWithPredicates of(final QName node, final QName key, final Object value) {
+            return new Singleton(node, key, value);
+        }
+
+        public static @NonNull NodeIdentifierWithPredicates of(final QName node, final Entry<QName, Object> entry) {
+            return of(node, entry.getKey(), entry.getValue());
+        }
+
+        public static @NonNull NodeIdentifierWithPredicates of(final QName node, final Map<QName, Object> keyValues) {
+            return keyValues.size() == 1 ? of(keyValues, node)
+                    // Retains ImmutableMap for empty maps. For larger sizes uses a shared key set.
+                    : new Regular(node, ImmutableOffsetMap.unorderedCopyOf(keyValues));
+        }
+
+        public static @NonNull NodeIdentifierWithPredicates of(final QName node,
+                final ImmutableOffsetMap<QName, Object> keyValues) {
+            return keyValues.size() == 1 ? of(keyValues, node) : new Regular(node, keyValues);
+        }
+
+        @Deprecated
+        public static @NonNull NodeIdentifierWithPredicates of(final QName node,
+                final SharedSingletonMap<QName, Object> keyValues) {
+            return of(node, keyValues.getEntry());
+        }
+
+        private static @NonNull NodeIdentifierWithPredicates of(final Map<QName, Object> keyValues, final QName node) {
+            return of(node, keyValues.entrySet().iterator().next());
+        }
+
+        /**
+         * Return the set of predicates keys and values. Keys are guaranteeed to be unique.
+         *
+         * @return Predicate set.
+         */
+        @Beta
+        public abstract @NonNull Set<Entry<QName, Object>> entrySet();
+
+        /**
+         * Return the predicate key in the iteration order of {@link #entrySet()}.
+         *
+         * @return Predicate values.
+         */
+        @Beta
+        public abstract @NonNull Set<QName> keySet();
+
+        /**
+         * Determine whether a particular predicate key is present.
+         *
+         * @param key Predicate key
+         * @return True if the predicate is present, false otherwise
+         * @throws NullPointerException if {@code key} is null
+         */
+        @Beta
+        public abstract boolean containsKey(QName key);
+
+        /**
+         * Return the predicate values in the iteration order of {@link #entrySet()}.
+         *
+         * @return Predicate values.
+         */
+        @Beta
+        public abstract @NonNull Collection<Object> values();
+
+        @Beta
+        public final @Nullable Object getValue(final QName key) {
+            return keyValue(requireNonNull(key));
         }
 
+        @Beta
+        public final <T> @Nullable T getValue(final QName key, final Class<T> valueClass) {
+            return valueClass.cast(getValue(key));
+        }
+
+        /**
+         * Return the number of predicates present.
+         *
+         * @return The number of predicates present.
+         */
+        @Beta
+        public abstract int size();
+
+        /**
+         * A Map-like view of this identifier's predicates. The view is expected to be stable and effectively-immutable.
+         *
+         * @return Map of predicates.
+         * @deprecated This method in a provisional one. It can be used in the code base, but users requiring it should
+         *             contact <a href="mailto:yangtools-dev@lists.opendaylight.org">yangtools-dev</a> for migration
+         *             guidelines. Callers are strongly encouraged to explore {@link #entrySet()}, {@link #size()},
+         *             {@link #values()} and {@link #keySet()} as an alternative.
+         */
+        @Beta
+        @Deprecated
+        // FIXME: 6.0.0: evaluate the real usefulness of this. The problem here is Map.hashCode() and Map.equals(),
+        //               which limits our options.
+        public abstract @NonNull Map<QName, Object> asMap();
+
         @Override
-        public String toString() {
-            return super.toString() + '[' + keyValues + ']';
+        protected final int hashCodeImpl() {
+            int result = 31 * super.hashCodeImpl();
+            for (Entry<QName, Object> entry : entrySet()) {
+                result += entry.getKey().hashCode() + YangInstanceIdentifier.hashCode(entry.getValue());
+            }
+            return result;
         }
 
         @Override
-        public String toRelativeString(final PathArgument previous) {
-            return super.toRelativeString(previous) + '[' + keyValues + ']';
+        @SuppressWarnings("checkstyle:equalsHashCode")
+        public final boolean equals(final Object obj) {
+            return super.equals(obj) && equalMapping((NodeIdentifierWithPredicates) obj);
+        }
+
+        abstract boolean equalMapping(NodeIdentifierWithPredicates other);
+
+        abstract @Nullable Object keyValue(@NonNull QName qname);
+
+        @Override
+        public final String toString() {
+            return super.toString() + '[' + asMap() + ']';
+        }
+
+        @Override
+        public final String toRelativeString(final PathArgument previous) {
+            return super.toRelativeString(previous) + '[' + asMap() + ']';
+        }
+
+        @Override
+        final Object writeReplace() {
+            return new NIPv2(this);
         }
     }
 
@@ -601,13 +865,11 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
 
         @Override
         protected int hashCodeImpl() {
-            final int prime = 31;
-            int result = super.hashCodeImpl();
-            result = prime * result + YangInstanceIdentifier.hashCode(value);
-            return result;
+            return 31 * super.hashCodeImpl() + YangInstanceIdentifier.hashCode(value);
         }
 
         @Override
+        @SuppressWarnings("checkstyle:equalsHashCode")
         public boolean equals(final Object obj) {
             if (!super.equals(obj)) {
                 return false;
@@ -625,12 +887,18 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
         public String toRelativeString(final PathArgument previous) {
             return super.toRelativeString(previous) + '[' + value + ']';
         }
+
+        @Override
+        Object writeReplace() {
+            return new NIVv1(this);
+        }
     }
 
     /**
-     * Composite path argument identifying a {@link org.opendaylight.yangtools.yang.data.api.schema.AugmentationNode} node in
-     * particular subtree.
+     * Composite path argument identifying a {@link org.opendaylight.yangtools.yang.data.api.schema.AugmentationNode}
+     * node in particular subtree.
      *
+     * <p>
      * Augmentation is uniquely identified by set of all possible child nodes.
      * This is possible
      * to identify instance of augmentation,
@@ -638,12 +906,20 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
      * statement must not add multiple nodes from same namespace
      * / module to the target node.
      *
-     *
      * @see <a href="http://tools.ietf.org/html/rfc6020#section-7.15">RFC6020</a>
      */
     public static final class AugmentationIdentifier implements PathArgument {
         private static final long serialVersionUID = -8122335594681936939L;
-        private final ImmutableSet<QName> childNames;
+
+        private static final LoadingCache<ImmutableSet<QName>, AugmentationIdentifier> CACHE = CacheBuilder.newBuilder()
+                .weakValues().build(new CacheLoader<ImmutableSet<QName>, AugmentationIdentifier>() {
+                    @Override
+                    public AugmentationIdentifier load(final ImmutableSet<QName> key) {
+                        return new AugmentationIdentifier(key);
+                    }
+                });
+
+        private final @NonNull ImmutableSet<QName> childNames;
 
         @Override
         public QName getNodeType() {
@@ -652,9 +928,19 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
         }
 
         /**
+         * Construct new augmentation identifier using supplied set of possible
+         * child nodes.
          *
+         * @param childNames
+         *            Set of possible child nodes.
+         */
+        public AugmentationIdentifier(final ImmutableSet<QName> childNames) {
+            this.childNames = requireNonNull(childNames);
+        }
+
+        /**
          * Construct new augmentation identifier using supplied set of possible
-         * child nodes
+         * child nodes.
          *
          * @param childNames
          *            Set of possible child nodes.
@@ -664,11 +950,34 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
         }
 
         /**
-         * Returns set of all possible child nodes
+         * Return an AugmentationIdentifier for a particular set of QNames. Unlike the constructor, this factory method
+         * uses a global instance cache, resulting in object reuse for equal inputs.
+         *
+         * @param childNames Set of possible child nodes
+         * @return An {@link AugmentationIdentifier}
+         */
+        public static @NonNull AugmentationIdentifier create(final ImmutableSet<QName> childNames) {
+            return CACHE.getUnchecked(childNames);
+        }
+
+        /**
+         * Return an AugmentationIdentifier for a particular set of QNames. Unlike the constructor, this factory method
+         * uses a global instance cache, resulting in object reuse for equal inputs.
+         *
+         * @param childNames Set of possible child nodes
+         * @return An {@link AugmentationIdentifier}
+         */
+        public static @NonNull AugmentationIdentifier create(final Set<QName> childNames) {
+            final AugmentationIdentifier existing = CACHE.getIfPresent(childNames);
+            return existing != null ? existing : create(ImmutableSet.copyOf(childNames));
+        }
+
+        /**
+         * Returns set of all possible child nodes.
          *
          * @return set of all possible child nodes.
          */
-        public Set<QName> getPossibleChildNames() {
+        public @NonNull Set<QName> getPossibleChildNames() {
             return childNames;
         }
 
@@ -683,15 +992,15 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
         }
 
         @Override
-        public boolean equals(final Object o) {
-            if (this == o) {
+        public boolean equals(final Object obj) {
+            if (this == obj) {
                 return true;
             }
-            if (!(o instanceof AugmentationIdentifier)) {
+            if (!(obj instanceof AugmentationIdentifier)) {
                 return false;
             }
 
-            AugmentationIdentifier that = (AugmentationIdentifier) o;
+            AugmentationIdentifier that = (AugmentationIdentifier) obj;
             return childNames.equals(that.childNames);
         }
 
@@ -701,6 +1010,7 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
         }
 
         @Override
+        @SuppressWarnings("checkstyle:parameterName")
         public int compareTo(final PathArgument o) {
             if (!(o instanceof AugmentationIdentifier)) {
                 return -1;
@@ -710,33 +1020,42 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
             int thisSize = childNames.size();
             int otherSize = otherChildNames.size();
             if (thisSize == otherSize) {
-                Iterator<QName> otherIterator = otherChildNames.iterator();
-                for (QName name : childNames) {
-                    int c = name.compareTo(otherIterator.next());
-                    if (c != 0) {
-                        return c;
-                    }
+                // Quick Set-based comparison
+                if (childNames.equals(otherChildNames)) {
+                    return 0;
                 }
-                return 0;
+
+                // We already know the sets are not equal, but have equal size, hence the sets differ in their elements,
+                // but potentially share a common set of elements. The most consistent way of comparing them is using
+                // total ordering defined by QName's compareTo. Hence convert both sets to lists ordered
+                // by QName.compareTo() and decide on the first differing element.
+                final List<QName> diff = new ArrayList<>(Sets.symmetricDifference(childNames, otherChildNames));
+                verify(!diff.isEmpty(), "Augmentation identifiers %s and %s report no difference", this, o);
+                diff.sort(QName::compareTo);
+                return childNames.contains(diff.get(0)) ? -1 : 1;
             } else if (thisSize < otherSize) {
                 return 1;
             } else {
                 return -1;
             }
         }
+
+        private Object writeReplace() {
+            return new AIv1(this);
+        }
     }
 
     /**
-     * Fluent Builder of Instance Identifier instances
+     * Fluent Builder of Instance Identifier instances.
      */
     public interface InstanceIdentifierBuilder extends Builder<YangInstanceIdentifier> {
         /**
-         * Adds a {@link PathArgument} to to path arguments of resulting instance identifier.
+         * Adds a {@link PathArgument} to path arguments of resulting instance identifier.
          *
          * @param arg A {@link PathArgument} to be added
          * @return this builder
          */
-        InstanceIdentifierBuilder node(PathArgument arg);
+        @NonNull InstanceIdentifierBuilder node(PathArgument arg);
 
         /**
          * Adds {@link NodeIdentifier} with supplied QName to path arguments of resulting instance identifier.
@@ -744,16 +1063,17 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
          * @param nodeType QName of {@link NodeIdentifier} which will be added
          * @return this builder
          */
-        InstanceIdentifierBuilder node(QName nodeType);
+        @NonNull InstanceIdentifierBuilder node(QName nodeType);
 
         /**
-         * Adds {@link NodeIdentifierWithPredicates} with supplied QName and key values to path arguments of resulting instance identifier.
+         * Adds {@link NodeIdentifierWithPredicates} with supplied QName and key values to path arguments of resulting
+         * instance identifier.
          *
          * @param nodeType QName of {@link NodeIdentifierWithPredicates} which will be added
          * @param keyValues Map of key components and their respective values for {@link NodeIdentifierWithPredicates}
          * @return this builder
          */
-        InstanceIdentifierBuilder nodeWithKey(QName nodeType, Map<QName, Object> keyValues);
+        @NonNull InstanceIdentifierBuilder nodeWithKey(QName nodeType, Map<QName, Object> keyValues);
 
         /**
          * Adds {@link NodeIdentifierWithPredicates} with supplied QName and key, value.
@@ -763,11 +1083,32 @@ public abstract class YangInstanceIdentifier implements Path<YangInstanceIdentif
          * @param value value of key which will be added
          * @return this builder
          */
-        InstanceIdentifierBuilder nodeWithKey(QName nodeType, QName key, Object value);
+        @NonNull InstanceIdentifierBuilder nodeWithKey(QName nodeType, QName key, Object value);
 
         /**
+         * Adds a collection of {@link PathArgument}s to path arguments of resulting instance identifier.
          *
-         * Builds an {@link YangInstanceIdentifier} with path arguments from this builder
+         * @param args {@link PathArgument}s to be added
+         * @return this builder
+         * @throws NullPointerException if any of the arguments is null
+         */
+        @Beta
+        @NonNull InstanceIdentifierBuilder append(Collection<? extends PathArgument> args);
+
+        /**
+         * Adds a collection of {@link PathArgument}s to path arguments of resulting instance identifier.
+         *
+         * @param args {@link PathArgument}s to be added
+         * @return this builder
+         * @throws NullPointerException if any of the arguments is null
+         */
+        @Beta
+        default @NonNull InstanceIdentifierBuilder append(final PathArgument... args) {
+            return append(Arrays.asList(args));
+        }
+
+        /**
+         * Builds an {@link YangInstanceIdentifier} with path arguments from this builder.
          *
          * @return {@link YangInstanceIdentifier}
          */