Reparent ChoiceCodecContext

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

diff --git a/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/ChoiceCodecPrototype.java b/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/ChoiceCodecPrototype.java
index 7fc1f3cca996e21e7bc85f71bdb89645f8a39944..f5ac3d670281375ecb2f900ec64b2109d906162f 100644 (file)
--- a/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/ChoiceCodecPrototype.java
+++ b/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/ChoiceCodecPrototype.java
@@ -7,20 +7,40 @@
  */
 package org.opendaylight.mdsal.binding.dom.codec.impl;
 
+import static java.util.Objects.requireNonNull;
+
+import org.eclipse.jdt.annotation.NonNull;
 import org.opendaylight.mdsal.binding.runtime.api.ChoiceRuntimeType;
-import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.Item;
+import org.opendaylight.yangtools.yang.binding.ChoiceIn;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
 
 /**
  * A prototype for {@link ChoiceCodecContext}.
  */
-final class ChoiceCodecPrototype extends DataObjectCodecPrototype<ChoiceRuntimeType> {
-    ChoiceCodecPrototype(final Item<?> item, final ChoiceRuntimeType type, final CodecContextFactory factory) {
-        super(item, NodeIdentifier.create(type.statement().argument()), type, factory);
+final class ChoiceCodecPrototype<T extends ChoiceIn<?>>
+        extends DataContainerPrototype<ChoiceCodecContext<T>, ChoiceRuntimeType> {
+    private final @NonNull NodeIdentifier yangArg;
+    private final @NonNull Class<T> javaClass;
+
+    ChoiceCodecPrototype(final CodecContextFactory contextFactory, final ChoiceRuntimeType runtimeType,
+            final Class<T> javaClass) {
+        super(contextFactory, runtimeType);
+        this.javaClass = requireNonNull(javaClass);
+        yangArg = NodeIdentifier.create(runtimeType.statement().argument());
+    }
+
+    @Override
+    Class<T> javaClass() {
+        return javaClass;
+    }
+
+    @Override
+    NodeIdentifier yangArg() {
+        return yangArg;
     }
 
     @Override
-    ChoiceCodecContext<?> createInstance() {
+    ChoiceCodecContext<T> createInstance() {
         return new ChoiceCodecContext<>(this);
     }
 }