Update ChoiceNodeCodecContext

[mdsal.git] / binding / mdsal-binding-dom-codec / src / main / java / org / opendaylight / mdsal / binding / dom / codec / impl / CaseNodeCodecContext.java

diff --git a/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/CaseNodeCodecContext.java b/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/CaseNodeCodecContext.java
index ac68f6a929a2ff209105249986aeaa0df6979228..1041319e434a06ff412b838037d08dbe90d4defa 100644 (file)
--- a/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/CaseNodeCodecContext.java
+++ b/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/CaseNodeCodecContext.java
@@ -10,17 +10,26 @@ package org.opendaylight.mdsal.binding.dom.codec.impl;
 import com.google.common.base.Preconditions;
 import java.util.List;
 import org.opendaylight.yangtools.yang.binding.DataObject;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.Item;
 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.PathArgument;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
-import org.opendaylight.yangtools.yang.model.api.ChoiceCaseNode;
+import org.opendaylight.yangtools.yang.model.api.CaseSchemaNode;
+import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
 
-final class CaseNodeCodecContext<D extends DataObject> extends DataObjectCodecContext<D,ChoiceCaseNode> {
-    public CaseNodeCodecContext(final DataContainerCodecPrototype<ChoiceCaseNode> prototype) {
+final class CaseNodeCodecContext<D extends DataObject> extends DataObjectCodecContext<D, CaseSchemaNode> {
+    CaseNodeCodecContext(final DataContainerCodecPrototype<CaseSchemaNode> prototype) {
         super(prototype);
     }
 
+    @Override
+    @SuppressWarnings({ "unchecked", "rawtypes" })
+    Item<?> createBindingArg(final Class<?> childClass, final DataSchemaNode childSchema) {
+        return childSchema.isAddedByUses() ? Item.of((Class)getBindingClass(), (Class)childClass)
+                : Item.of((Class<? extends DataObject>) childClass);
+    }
+
     @Override
     protected void addYangPathArgument(final PathArgument arg,
             final List<YangInstanceIdentifier.PathArgument> builder) {
@@ -39,8 +48,7 @@ final class CaseNodeCodecContext<D extends DataObject> extends DataObjectCodecCo
     }
 
     @Override
-    public YangInstanceIdentifier.PathArgument serializePathArgument(
-            final PathArgument arg) {
+    public YangInstanceIdentifier.PathArgument serializePathArgument(final PathArgument arg) {
         Preconditions.checkArgument(arg == null);
         return null;
     }
@@ -50,4 +58,4 @@ final class CaseNodeCodecContext<D extends DataObject> extends DataObjectCodecCo
         Preconditions.checkArgument(arg == null);
         return null;
     }
-}
\ No newline at end of file
+}