Split out BindingDataObjectCodecTreeNode.streamChild()

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

diff --git a/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/SchemaRootCodecContext.java b/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/SchemaRootCodecContext.java
index 8b80d4152ba5e6b6f7f51a5bef96aab6d44cc41a..bc1b2d4b8d53a3548b273a2886cd24c6d06d7876 100644 (file)
--- a/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/SchemaRootCodecContext.java
+++ b/binding/mdsal-binding-dom-codec/src/main/java/org/opendaylight/mdsal/binding/dom/codec/impl/SchemaRootCodecContext.java
@@ -225,10 +225,12 @@ final class SchemaRootCodecContext<D extends DataObject> extends DataContainerCo
         return getType().getEffectiveModelContext();
     }
 
-    @SuppressWarnings("unchecked")
     @Override
+    @SuppressWarnings("unchecked")
     public <C extends DataObject> DataContainerCodecContext<C, ?> streamChild(final Class<C> childClass) {
-        return (DataContainerCodecContext<C, ?>) getOrRethrow(childrenByClass, childClass);
+        final DataContainerCodecContext<?, ?> result = Notification.class.isAssignableFrom(childClass)
+            ? getNotificationImpl(childClass) : getOrRethrow(childrenByClass, childClass);
+        return (DataContainerCodecContext<C, ?>) result;
     }
 
     @Override