Bump upstreams

[bgpcep.git] / bgp / rib-impl / src / main / java / org / opendaylight / protocol / bgp / rib / impl / CodecsImpl.java

diff --git a/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/CodecsImpl.java b/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/CodecsImpl.java
index d2f6d158c62b53f9c7d288c9d880a700048d0ad9..84419ed8bfac9dc0eba9b9e1df455591953a041b 100644 (file)
--- a/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/CodecsImpl.java
+++ b/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/CodecsImpl.java
@@ -16,6 +16,7 @@ import java.util.Set;
 import org.opendaylight.mdsal.binding.dom.codec.api.BindingCodecTree;
 import org.opendaylight.mdsal.binding.dom.codec.api.BindingDataObjectCodecTreeNode;
 import org.opendaylight.mdsal.binding.dom.codec.api.BindingNormalizedNodeCachingCodec;
+import org.opendaylight.mdsal.binding.dom.codec.api.CommonDataObjectCodecTreeNode;
 import org.opendaylight.protocol.bgp.rib.impl.spi.Codecs;
 import org.opendaylight.protocol.bgp.rib.spi.RIBSupport;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.ClusterId;
@@ -96,24 +97,26 @@ public final class CodecsImpl implements Codecs {
     @SuppressWarnings("unchecked")
     public void onCodecTreeUpdated(final BindingCodecTree tree) {
 
-        @SuppressWarnings("rawtypes")
-        final BindingDataObjectCodecTreeNode tableCodecContext = tree.getSubtreeCodec(TABLE_BASE_II);
+        final CommonDataObjectCodecTreeNode<Tables> codecContext = tree.getSubtreeCodec(TABLE_BASE_II);
+        if (!(codecContext instanceof BindingDataObjectCodecTreeNode tableCodecContext)) {
+            throw new IllegalStateException("Unexpected table codec " + codecContext);
+        }
+
         final BindingDataObjectCodecTreeNode<? extends Route> routeListCodec = tableCodecContext
-            .streamChild(Routes.class)
-            .streamChild(ribSupport.routesCaseClass())
-            .streamChild(ribSupport.routesContainerClass())
-            .streamChild(ribSupport.routesListClass());
+            .getStreamChild(Routes.class)
+            .getStreamChild(ribSupport.routesCaseClass())
+            .getStreamChild(ribSupport.routesContainerClass())
+            .getStreamDataObject(ribSupport.routesListClass());
 
-        attributesCodec = routeListCodec.streamChild(Attributes.class)
+        attributesCodec = routeListCodec.getStreamDataObject(Attributes.class)
             .createCachingCodec(cacheableAttributes);
 
-        final var attrCodec = tree.streamChild(Update.class)
-            .streamChild(Attributes.class);
-        reachNlriCodec = attrCodec.streamChild(AttributesReach.class)
-            .streamChild(MpReachNlri.class)
+        final var attrCodec = tree.getStreamChild(Update.class).getStreamChild(Attributes.class);
+        reachNlriCodec = attrCodec.getStreamChild(AttributesReach.class)
+            .getStreamDataObject(MpReachNlri.class)
             .createCachingCodec(ribSupport.cacheableNlriObjects());
-        unreachNlriCodec = attrCodec.streamChild(AttributesUnreach.class)
-            .streamChild(MpUnreachNlri.class)
+        unreachNlriCodec = attrCodec.getStreamChild(AttributesUnreach.class)
+            .getStreamDataObject(MpUnreachNlri.class)
             .createCachingCodec(ribSupport.cacheableNlriObjects());
     }