Add YangInstanceIdentifierBuilder.node(PathArgument)

[yangtools.git] / yang / yang-data-api / src / main / java / org / opendaylight / yangtools / yang / data / api / YangInstanceIdentifierBuilder.java

diff --git a/yang/yang-data-api/src/main/java/org/opendaylight/yangtools/yang/data/api/YangInstanceIdentifierBuilder.java b/yang/yang-data-api/src/main/java/org/opendaylight/yangtools/yang/data/api/YangInstanceIdentifierBuilder.java
index 55f03af8009edf1265e826a40f97d42f37bd554f..63aec35904c598a933e56e8a193063db7480dfcc 100644 (file)
--- a/yang/yang-data-api/src/main/java/org/opendaylight/yangtools/yang/data/api/YangInstanceIdentifierBuilder.java
+++ b/yang/yang-data-api/src/main/java/org/opendaylight/yangtools/yang/data/api/YangInstanceIdentifierBuilder.java
@@ -6,6 +6,7 @@
  */
 package org.opendaylight.yangtools.yang.data.api;
 
+import com.google.common.base.Preconditions;
 import com.google.common.collect.Lists;
 import java.util.ArrayList;
 import java.util.List;
@@ -31,28 +32,30 @@ final class YangInstanceIdentifierBuilder implements InstanceIdentifierBuilder {
         this.hash = new HashCodeBuilder<>(hash);
     }
 
-    @Override
-    public InstanceIdentifierBuilder node(final QName nodeType) {
-        final PathArgument arg = new NodeIdentifier(nodeType);
+    private InstanceIdentifierBuilder addArgument(final PathArgument arg) {
         path.add(arg);
         hash.addArgument(arg);
         return this;
     }
 
+    @Override
+    public InstanceIdentifierBuilder node(final PathArgument arg) {
+        return addArgument(Preconditions.checkNotNull(arg));
+    }
+
+    @Override
+    public InstanceIdentifierBuilder node(final QName nodeType) {
+        return addArgument(new NodeIdentifier(nodeType));
+    }
+
     @Override
     public InstanceIdentifierBuilder nodeWithKey(final QName nodeType, final QName key, final Object value) {
-        final PathArgument arg = new NodeIdentifierWithPredicates(nodeType, key, value);
-        path.add(arg);
-        hash.addArgument(arg);
-        return this;
+        return addArgument(new NodeIdentifierWithPredicates(nodeType, key, value));
     }
 
     @Override
     public InstanceIdentifierBuilder nodeWithKey(final QName nodeType, final Map<QName, Object> keyValues) {
-        final PathArgument arg = new NodeIdentifierWithPredicates(nodeType, keyValues);
-        path.add(arg);
-        hash.addArgument(arg);
-        return this;
+        return addArgument(new NodeIdentifierWithPredicates(nodeType, keyValues));
     }
 
     @Override