Generate DataObject codec implementation

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

diff --git a/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/KeyedListNodeCodecContext.java b/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/KeyedListNodeCodecContext.java
index b7413d375269c0ad4e1a7b7e16c67f7aca69c9b4..0da9d7a9921fd81b14d2fe74c42103ef5ff1d443 100644 (file)
--- a/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/KeyedListNodeCodecContext.java
+++ b/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/KeyedListNodeCodecContext.java
@@ -9,6 +9,7 @@ package org.opendaylight.mdsal.binding.dom.codec.impl;
 
 import static org.opendaylight.mdsal.binding.spec.naming.BindingMapping.IDENTIFIABLE_KEY_NAME;
 
+import java.lang.reflect.Method;
 import java.util.List;
 import org.opendaylight.yangtools.yang.binding.DataObject;
 import org.opendaylight.yangtools.yang.binding.Identifiable;
@@ -25,13 +26,14 @@ final class KeyedListNodeCodecContext<D extends DataObject & Identifiable<?>> ex
     private final IdentifiableItemCodec codec;
 
     KeyedListNodeCodecContext(final DataContainerCodecPrototype<ListSchemaNode> prototype) {
-        super(prototype);
+        super(prototype, keyMethod(prototype.getBindingClass()));
+        this.codec = factory().getPathArgumentCodec(getBindingClass(), getSchema());
+    }
 
-        final Class<D> bindingClass = getBindingClass();
-        this.codec = factory().getPathArgumentCodec(bindingClass, getSchema());
+    private static Method keyMethod(final Class<?> bindingClass) {
         try {
             // This just verifies the method is present
-            bindingClass.getMethod(IDENTIFIABLE_KEY_NAME);
+            return bindingClass.getMethod(IDENTIFIABLE_KEY_NAME);
         } catch (NoSuchMethodException e) {
             throw new IllegalStateException("Required method not available", e);
         }