Add SchemaPath->SchemaNodeIdentifier conversions
[yangtools.git] / yang / yang-model-api / src / main / java / org / opendaylight / yangtools / yang / model / api / SchemaPath.java
index 6a72b0efbd7ae3957203ad579348f485a377cdb9..267e9786e4cd515d221074e051293b3b01955908 100644 (file)
@@ -7,30 +7,37 @@
  */
 package org.opendaylight.yangtools.yang.model.api;
 
-import com.google.common.base.Objects;
-import com.google.common.base.Objects.ToStringHelper;
-import com.google.common.base.Preconditions;
+import static com.google.common.base.Preconditions.checkArgument;
+import static com.google.common.base.Preconditions.checkState;
+import static java.util.Objects.requireNonNull;
+
+import com.google.common.base.MoreObjects;
+import com.google.common.base.MoreObjects.ToStringHelper;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.Iterables;
-
+import com.google.common.collect.UnmodifiableIterator;
 import java.util.Arrays;
-import java.util.Collections;
 import java.util.List;
-
+import java.util.NoSuchElementException;
+import java.util.Objects;
+import org.eclipse.jdt.annotation.NonNull;
 import org.opendaylight.yangtools.concepts.Immutable;
-import org.opendaylight.yangtools.util.HashCodeBuilder;
 import org.opendaylight.yangtools.yang.common.QName;
+import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier;
+import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier.Absolute;
+import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier.Descendant;
 
 /**
  * Represents unique path to the every node inside the module.
  */
 public abstract class SchemaPath implements Immutable {
+
     /**
      * An absolute SchemaPath.
      */
     private static final class AbsoluteSchemaPath extends SchemaPath {
-        private AbsoluteSchemaPath(final Iterable<QName> path, final int hash) {
-            super(path, hash);
+        private AbsoluteSchemaPath(final SchemaPath parent, final QName qname) {
+            super(parent, qname);
         }
 
         @Override
@@ -39,8 +46,8 @@ public abstract class SchemaPath implements Immutable {
         }
 
         @Override
-        protected SchemaPath createInstance(final Iterable<QName> path, final int hash) {
-            return new AbsoluteSchemaPath(path, hash);
+        public AbsoluteSchemaPath createChild(final QName element) {
+            return new AbsoluteSchemaPath(this, requireNonNull(element));
         }
     }
 
@@ -48,8 +55,8 @@ public abstract class SchemaPath implements Immutable {
      * A relative SchemaPath.
      */
     private static final class RelativeSchemaPath extends SchemaPath {
-        private RelativeSchemaPath(final Iterable<QName> path, final int hash) {
-            super(path, hash);
+        private RelativeSchemaPath(final SchemaPath parent, final QName qname) {
+            super(parent, qname);
         }
 
         @Override
@@ -58,78 +65,78 @@ public abstract class SchemaPath implements Immutable {
         }
 
         @Override
-        protected SchemaPath createInstance(final Iterable<QName> path, final int hash) {
-            return new RelativeSchemaPath(path, hash);
+        public RelativeSchemaPath createChild(final QName element) {
+            return new RelativeSchemaPath(this, requireNonNull(element));
         }
     }
 
     /**
      * Shared instance of the conceptual root schema node.
      */
-    public static final SchemaPath ROOT = new AbsoluteSchemaPath(Collections.<QName>emptyList(), Boolean.TRUE.hashCode());
+    public static final @NonNull SchemaPath ROOT = new AbsoluteSchemaPath(null, null);
 
     /**
      * Shared instance of the "same" relative schema node.
      */
-    public static final SchemaPath SAME = new RelativeSchemaPath(Collections.<QName>emptyList(), Boolean.FALSE.hashCode());
+    public static final @NonNull SchemaPath SAME = new RelativeSchemaPath(null, null);
 
     /**
-     * List of QName instances which represents complete path to the node.
+     * Parent path.
      */
-    private final Iterable<QName> path;
+    private final SchemaPath parent;
 
     /**
-     * Cached hash code. We can use this since we are immutable.
+     * This component.
      */
-    private final int hash;
+    private final QName qname;
 
     /**
-     * Cached legacy path, filled-in when {@link #getPath()} or {@link #getPathTowardsRoot()}
-     * is invoked.
+     * Cached hash code. We can use this since we are immutable.
      */
-    private ImmutableList<QName> legacyPath;
+    private final int hash;
 
-    private ImmutableList<QName> getLegacyPath() {
-        if (legacyPath == null) {
-            legacyPath = ImmutableList.copyOf(path);
+    SchemaPath(final SchemaPath parent, final QName qname) {
+        this.parent = parent;
+        this.qname = qname;
+
+        int tmp = Objects.hashCode(parent);
+        if (qname != null) {
+            tmp = tmp * 31 + qname.hashCode();
         }
 
-        return legacyPath;
+        hash = tmp;
     }
 
     /**
-     * Returns the complete path to schema node.
+     * Constructs new instance of this class with the concrete path.
      *
-     * @return list of <code>QName</code> instances which represents complete
-     *         path to schema node
+     * @param path
+     *            list of QName instances which specifies exact path to the
+     *            module node
+     * @param absolute
+     *            boolean value which specifies if the path is absolute or
+     *            relative
      *
-     * @deprecated Use {@link #getPathFromRoot()} instead.
+     * @return A SchemaPath instance.
      */
-    @Deprecated
-    public List<QName> getPath() {
-        return getLegacyPath();
-    }
-
-    protected SchemaPath(final Iterable<QName> path, final int hash) {
-        this.path = Preconditions.checkNotNull(path);
-        this.hash = hash;
+    public static @NonNull SchemaPath create(final Iterable<QName> path, final boolean absolute) {
+        return (absolute ? ROOT : SAME).createChild(path);
     }
 
     /**
      * Constructs new instance of this class with the concrete path.
      *
-     * @param path
-     *            list of QName instances which specifies exact path to the
-     *            module node
      * @param absolute
      *            boolean value which specifies if the path is absolute or
      *            relative
+     * @param element
+     *            a single QName which specifies exact path to the
+     *            module node
      *
      * @return A SchemaPath instance.
      */
-    public static SchemaPath create(final Iterable<QName> path, final boolean absolute) {
-        final SchemaPath parent = absolute ? ROOT : SAME;
-        return parent.createChild(path);
+    public static @NonNull SchemaPath create(final boolean absolute, final QName element) {
+        return (absolute ? ROOT : SAME).createChild(element);
     }
 
     /**
@@ -144,56 +151,47 @@ public abstract class SchemaPath implements Immutable {
      *
      * @return A SchemaPath instance.
      */
-    public static SchemaPath create(final boolean absolute, final QName... path) {
+    public static @NonNull SchemaPath create(final boolean absolute, final QName... path) {
         return create(Arrays.asList(path), absolute);
     }
 
     /**
-     * Create a new instance.
+     * Create a child path based on concatenation of this path and a relative path.
      *
-     * @param path path from root
-     * @param hash intended hash code
-     * @return A new SchemaPath instance
+     * @param relative Relative path
+     * @return A new child path
      */
-    protected abstract SchemaPath createInstance(Iterable<QName> path, int hash);
-
-    private SchemaPath trustedCreateChild(final Iterable<QName> relative) {
+    public @NonNull SchemaPath createChild(final Iterable<QName> relative) {
         if (Iterables.isEmpty(relative)) {
             return this;
         }
 
-        final HashCodeBuilder<QName> b = new HashCodeBuilder<>(hash);
-        for (QName p : relative) {
-            b.addArgument(p);
+        SchemaPath parentPath = this;
+        for (QName item : relative) {
+            parentPath = parentPath.createChild(item);
         }
 
-        return createInstance(Iterables.concat(path, relative), b.toInstance());
+        return parentPath;
     }
 
     /**
      * Create a child path based on concatenation of this path and a relative path.
      *
-     * @param relative Relative path
+     * @param relative Relative SchemaPath
      * @return A new child path
      */
-    public SchemaPath createChild(final Iterable<QName> relative) {
-        if (Iterables.isEmpty(relative)) {
-            return this;
-        }
-
-        return trustedCreateChild(ImmutableList.copyOf(relative));
+    public @NonNull SchemaPath createChild(final SchemaPath relative) {
+        checkArgument(!relative.isAbsolute(), "Child creation requires relative path");
+        return createChild(relative.getPathFromRoot());
     }
 
     /**
-     * Create a child path based on concatenation of this path and a relative path.
+     * Create a child path based on concatenation of this path and an additional path element.
      *
-     * @param relative Relative SchemaPath
+     * @param element Relative SchemaPath elements
      * @return A new child path
      */
-    public SchemaPath createChild(final SchemaPath relative) {
-        Preconditions.checkArgument(!relative.isAbsolute(), "Child creation requires relative path");
-        return trustedCreateChild(relative.path);
-    }
+    public abstract @NonNull SchemaPath createChild(QName element);
 
     /**
      * Create a child path based on concatenation of this path and additional
@@ -202,7 +200,7 @@ public abstract class SchemaPath implements Immutable {
      * @param elements Relative SchemaPath elements
      * @return A new child path
      */
-    public SchemaPath createChild(final QName... elements) {
+    public @NonNull SchemaPath createChild(final QName... elements) {
         return createChild(Arrays.asList(elements));
     }
 
@@ -214,8 +212,11 @@ public abstract class SchemaPath implements Immutable {
      * @return list of <code>qname</code> instances which represents
      *         path from the root to the schema node.
      */
-    public Iterable<QName> getPathFromRoot() {
-        return path;
+    public List<QName> getPathFromRoot() {
+        if (qname == null) {
+            return ImmutableList.of();
+        }
+        return parent == null ? ImmutableList.of(qname) : new PathFromRoot(this);
     }
 
     /**
@@ -226,7 +227,25 @@ public abstract class SchemaPath implements Immutable {
      *         path from the schema node towards the root.
      */
     public Iterable<QName> getPathTowardsRoot() {
-        return getLegacyPath().reverse();
+        return () -> new UnmodifiableIterator<>() {
+            private SchemaPath current = SchemaPath.this;
+
+            @Override
+            public boolean hasNext() {
+                return current.parent != null;
+            }
+
+            @Override
+            public QName next() {
+                if (current.parent != null) {
+                    final QName ret = current.qname;
+                    current = current.parent;
+                    return ret;
+                }
+
+                throw new NoSuchElementException("No more elements available");
+            }
+        };
     }
 
     /**
@@ -235,13 +254,16 @@ public abstract class SchemaPath implements Immutable {
      * @return Parent path, null if this SchemaPath is already toplevel.
      */
     public SchemaPath getParent() {
-        final int size = Iterables.size(path);
-        if (size != 0) {
-            final SchemaPath parent = isAbsolute() ? ROOT : SAME;
-            return parent.trustedCreateChild(Iterables.limit(path, size - 1));
-        } else {
-            return null;
-        }
+        return parent;
+    }
+
+    /**
+     * Get the last component of this path.
+     *
+     * @return The last component of this path.
+     */
+    public final QName getLastComponent() {
+        return qname;
     }
 
     /**
@@ -252,6 +274,46 @@ public abstract class SchemaPath implements Immutable {
      */
     public abstract boolean isAbsolute();
 
+    /**
+     * Return this path as a {@link SchemaNodeIdentifier}.
+     *
+     * @return A SchemaNodeIdentifier.
+     * @throws IllegalStateException if this path is empty
+     */
+    public final SchemaNodeIdentifier asSchemaNodeIdentifier() {
+        checkState(qname != null, "Cannot convert empty %s", this);
+        final List<QName> path = getPathFromRoot();
+        return isAbsolute() ? Absolute.of(path) : Descendant.of(path);
+    }
+
+    /**
+     * Return this path as an {@link Absolute} SchemaNodeIdentifier.
+     *
+     * @return An SchemaNodeIdentifier.
+     * @throws IllegalStateException if this path is empty or is not absolute.
+     */
+    public final Absolute asAbsolute() {
+        final SchemaNodeIdentifier ret = asSchemaNodeIdentifier();
+        if (ret instanceof Absolute) {
+            return (Absolute) ret;
+        }
+        throw new IllegalStateException("Path " + this + " is relative");
+    }
+
+    /**
+     * Return this path as an {@link Descendant} SchemaNodeIdentifier.
+     *
+     * @return An SchemaNodeIdentifier.
+     * @throws IllegalStateException if this path is empty or is not relative.
+     */
+    public final Descendant asDescendant() {
+        final SchemaNodeIdentifier ret = asSchemaNodeIdentifier();
+        if (ret instanceof Descendant) {
+            return (Descendant) ret;
+        }
+        throw new IllegalStateException("Path " + this + " is absolute");
+    }
+
     @Override
     public final int hashCode() {
         return hash;
@@ -268,16 +330,16 @@ public abstract class SchemaPath implements Immutable {
         if (getClass() != obj.getClass()) {
             return false;
         }
-        SchemaPath other = (SchemaPath) obj;
-        return Iterables.elementsEqual(path, other.path);
+        final SchemaPath other = (SchemaPath) obj;
+        return Objects.equals(qname, other.qname) && Objects.equals(parent, other.parent);
     }
 
     @Override
     public final String toString() {
-        return addToStringAttributes(Objects.toStringHelper(this)).toString();
+        return addToStringAttributes(MoreObjects.toStringHelper(this)).toString();
     }
 
     protected ToStringHelper addToStringAttributes(final ToStringHelper toStringHelper) {
-        return toStringHelper.add("path", path);
+        return toStringHelper.add("path", getPathFromRoot());
     }
 }