Refactor PathArgument to DataObjectStep
[mdsal.git] / binding / mdsal-binding-dom-codec / src / main / java / org / opendaylight / mdsal / binding / dom / codec / impl / AugmentationCodecPrototype.java
index dccf09ae3ca6f02174d0c9020c80e7639a2a35cd..5f10f1abc5d8e6321da7a9aa781f2eb58ff2523a 100644 (file)
@@ -13,16 +13,16 @@ import com.google.common.collect.ImmutableSet;
 import org.eclipse.jdt.annotation.NonNull;
 import org.opendaylight.mdsal.binding.runtime.api.AugmentRuntimeType;
 import org.opendaylight.yangtools.yang.binding.DataObject;
-import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.Item;
+import org.opendaylight.yangtools.yang.binding.NodeStep;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
 
 final class AugmentationCodecPrototype extends CommonDataObjectCodecPrototype<AugmentRuntimeType> {
     private final @NonNull ImmutableSet<NodeIdentifier> childArgs;
 
     @SuppressWarnings("unchecked")
-    AugmentationCodecPrototype(final Class<?> cls, final AugmentRuntimeType type, final CodecContextFactory factory,
-            final ImmutableSet<NodeIdentifier> childArgs) {
-        super(Item.of((Class<? extends DataObject>) cls), type, factory);
+    AugmentationCodecPrototype(final @NonNull Class<?> cls, final AugmentRuntimeType type,
+            final CodecContextFactory factory, final ImmutableSet<NodeIdentifier> childArgs) {
+        super(new NodeStep<>((Class<? extends DataObject>) cls), type, factory);
         this.childArgs = requireNonNull(childArgs);
     }