Bump upstreams to 2022.09
[bgpcep.git] / bgp / rib-impl / src / main / java / org / opendaylight / protocol / bgp / rib / impl / LocRibWriter.java
index 3cbd173c11b78eaa91231ea076e590824f3bb973..170e1b955519a39596030f338e2cc3db4056cd5f 100644 (file)
  */
 package org.opendaylight.protocol.bgp.rib.impl;
 
-import com.google.common.base.Optional;
-import com.google.common.base.Preconditions;
-import com.google.common.primitives.UnsignedInteger;
+import static com.google.common.base.Verify.verify;
+import static java.util.Objects.requireNonNull;
+import static org.opendaylight.protocol.bgp.rib.spi.RIBNodeIdentifiers.ATTRIBUTES_NID;
+import static org.opendaylight.protocol.bgp.rib.spi.RIBNodeIdentifiers.EFFRIBIN_NID;
+import static org.opendaylight.protocol.bgp.rib.spi.RIBNodeIdentifiers.LOCRIB_NID;
+import static org.opendaylight.protocol.bgp.rib.spi.RIBNodeIdentifiers.PEER_NID;
+import static org.opendaylight.protocol.bgp.rib.spi.RIBNodeIdentifiers.ROUTES_NID;
+import static org.opendaylight.protocol.bgp.rib.spi.RIBNodeIdentifiers.TABLES_NID;
+import static org.opendaylight.protocol.bgp.rib.spi.RIBNodeIdentifiers.UPTODATE_NID;
+
+import com.google.common.util.concurrent.FutureCallback;
+import com.google.common.util.concurrent.MoreExecutors;
+import java.util.ArrayList;
 import java.util.Collection;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
+import java.util.Map.Entry;
 import java.util.Set;
-import javax.annotation.Nonnull;
-import javax.annotation.concurrent.NotThreadSafe;
-import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
-import org.opendaylight.controller.md.sal.dom.api.ClusteredDOMDataTreeChangeListener;
-import org.opendaylight.controller.md.sal.dom.api.DOMDataTreeChangeService;
-import org.opendaylight.controller.md.sal.dom.api.DOMDataTreeIdentifier;
-import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
-import org.opendaylight.controller.md.sal.dom.api.DOMTransactionChain;
+import java.util.concurrent.atomic.LongAdder;
+import org.checkerframework.checker.lock.qual.GuardedBy;
+import org.eclipse.jdt.annotation.NonNull;
+import org.opendaylight.mdsal.common.api.CommitInfo;
+import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
+import org.opendaylight.mdsal.dom.api.ClusteredDOMDataTreeChangeListener;
+import org.opendaylight.mdsal.dom.api.DOMDataTreeChangeService;
+import org.opendaylight.mdsal.dom.api.DOMDataTreeIdentifier;
+import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteOperations;
+import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
+import org.opendaylight.mdsal.dom.api.DOMTransactionChain;
 import org.opendaylight.protocol.bgp.mode.api.PathSelectionMode;
 import org.opendaylight.protocol.bgp.mode.api.RouteEntry;
-import org.opendaylight.protocol.bgp.rib.impl.spi.RIBSupportContextRegistry;
-import org.opendaylight.protocol.bgp.rib.impl.stats.UnsignedInt32Counter;
-import org.opendaylight.protocol.bgp.rib.spi.ExportPolicyPeerTracker;
+import org.opendaylight.protocol.bgp.rib.impl.spi.RibOutRefresh;
+import org.opendaylight.protocol.bgp.rib.impl.state.rib.TotalPathsCounter;
+import org.opendaylight.protocol.bgp.rib.impl.state.rib.TotalPrefixesCounter;
+import org.opendaylight.protocol.bgp.rib.spi.BGPPeerTracker;
 import org.opendaylight.protocol.bgp.rib.spi.IdentifierUtils;
-import org.opendaylight.protocol.bgp.rib.spi.PeerExportGroup;
+import org.opendaylight.protocol.bgp.rib.spi.RIBNormalizedNodes;
 import org.opendaylight.protocol.bgp.rib.spi.RIBSupport;
-import org.opendaylight.protocol.bgp.rib.spi.RibSupportUtils;
-import org.opendaylight.protocol.bgp.rib.spi.RouterIds;
+import org.opendaylight.protocol.bgp.rib.spi.RouterId;
+import org.opendaylight.protocol.bgp.rib.spi.entry.ActualBestPathRoutes;
+import org.opendaylight.protocol.bgp.rib.spi.entry.AdvertizedRoute;
+import org.opendaylight.protocol.bgp.rib.spi.entry.StaleBestPathRoute;
+import org.opendaylight.protocol.bgp.rib.spi.policy.BGPRibRoutingPolicy;
+import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.AfiSafiType;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.PeerId;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.PeerRole;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.rib.LocRib;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.rib.Peer;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.rib.peer.EffectiveRibIn;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.Tables;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.tables.Attributes;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.tables.Routes;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.PeerId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.Tables;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.tables.Routes;
 import org.opendaylight.yangtools.concepts.ListenerRegistration;
-import org.opendaylight.yangtools.yang.common.QName;
+import org.opendaylight.yangtools.yang.binding.ChildOf;
+import org.opendaylight.yangtools.yang.binding.ChoiceIn;
+import org.opendaylight.yangtools.yang.binding.DataObject;
+import org.opendaylight.yangtools.yang.common.Uint32;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
-import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
-import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
+import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
-import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidate;
-import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidateNode;
-import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
+import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNodes;
+import org.opendaylight.yangtools.yang.data.tree.api.DataTreeCandidate;
+import org.opendaylight.yangtools.yang.data.tree.api.DataTreeCandidateNode;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-@NotThreadSafe
-final class LocRibWriter implements AutoCloseable, ClusteredDOMDataTreeChangeListener {
+// This class is NOT thread-safe
+final class LocRibWriter<C extends Routes & DataObject & ChoiceIn<Tables>, S extends ChildOf<? super C>>
+        implements AutoCloseable, RibOutRefresh, TotalPrefixesCounter, TotalPathsCounter,
+        ClusteredDOMDataTreeChangeListener {
 
     private static final Logger LOG = LoggerFactory.getLogger(LocRibWriter.class);
 
-    private static final LeafNode<Boolean> ATTRIBUTES_UPTODATE_TRUE = ImmutableNodes.leafNode(QName.create(Attributes.QNAME, "uptodate"), Boolean.TRUE);
-
-    private final Map<PathArgument, RouteEntry> routeEntries = new HashMap<>();
-    private final YangInstanceIdentifier locRibTarget;
-    private final DOMTransactionChain chain;
-    private final ExportPolicyPeerTracker exportPolicyPeerTracker;
-    private final NodeIdentifier attributesIdentifier;
-    private final Long ourAs;
-    private final RIBSupport ribSupport;
-    private final TablesKey localTablesKey;
-    private final ListenerRegistration<LocRibWriter> reg;
+    private final Map<String, RouteEntry<C, S>> routeEntries = new HashMap<>();
+    private final long ourAs;
+    private final RIBSupport<C, S> ribSupport;
+    private final DOMDataTreeChangeService dataBroker;
     private final PathSelectionMode pathSelectionMode;
-    private final UnsignedInt32Counter routeCounter;
-
-    private LocRibWriter(final RIBSupportContextRegistry registry, final DOMTransactionChain chain, final YangInstanceIdentifier target,
-        final Long ourAs, final DOMDataTreeChangeService service, final ExportPolicyPeerTracker exportPolicyPeerTracker, final TablesKey tablesKey,
-        @Nonnull final PathSelectionMode pathSelectionMode, final UnsignedInt32Counter routeCounter) {
-        this.chain = Preconditions.checkNotNull(chain);
-        final NodeIdentifierWithPredicates tableKey = RibSupportUtils.toYangTablesKey(tablesKey);
-        this.localTablesKey = tablesKey;
-        this.locRibTarget = YangInstanceIdentifier.create(target.node(LocRib.QNAME).node(Tables.QNAME).node(tableKey).getPathArguments());
-        this.ourAs = Preconditions.checkNotNull(ourAs);
-        this.ribSupport = registry.getRIBSupportContext(tablesKey).getRibSupport();
-        this.attributesIdentifier = this.ribSupport.routeAttributesIdentifier();
-        this.exportPolicyPeerTracker = exportPolicyPeerTracker;
+    private final LongAdder totalPathsCounter = new LongAdder();
+    private final LongAdder totalPrefixesCounter = new LongAdder();
+    private final RouteEntryDependenciesContainerImpl entryDep;
+    private final BGPPeerTracker peerTracker;
+    private final YangInstanceIdentifier ribIId;
+    private final YangInstanceIdentifier locRibTableIID;
+
+    private DOMTransactionChain chain;
+    @GuardedBy("this")
+    private ListenerRegistration<?> reg;
+
+    private LocRibWriter(final RIBSupport<C, S> ribSupport,
+            final DOMTransactionChain chain,
+            final YangInstanceIdentifier ribIId,
+            final Uint32 ourAs,
+            final DOMDataTreeChangeService dataBroker,
+            final BGPRibRoutingPolicy ribPolicies,
+            final BGPPeerTracker peerTracker,
+            final AfiSafiType afiSafiType,
+            final PathSelectionMode pathSelectionMode) {
+        this.chain = requireNonNull(chain);
+        this.ribIId = requireNonNull(ribIId);
+        this.ribSupport = requireNonNull(ribSupport);
+
+        this.locRibTableIID = ribIId.node(LOCRIB_NID).node(TABLES_NID).node(ribSupport.emptyTable().getIdentifier())
+            .toOptimized();
+        this.ourAs = ourAs.toJava();
+        this.dataBroker = requireNonNull(dataBroker);
+        this.peerTracker = peerTracker;
         this.pathSelectionMode = pathSelectionMode;
-        this.routeCounter = routeCounter;
 
-        final DOMDataWriteTransaction tx = this.chain.newWriteOnlyTransaction();
-        tx.merge(LogicalDatastoreType.OPERATIONAL, this.locRibTarget.node(Routes.QNAME), this.ribSupport.emptyRoutes());
-        tx.merge(LogicalDatastoreType.OPERATIONAL, this.locRibTarget.node(Attributes.QNAME).node(ATTRIBUTES_UPTODATE_TRUE.getNodeType()), ATTRIBUTES_UPTODATE_TRUE);
-        tx.submit();
+        this.entryDep = new RouteEntryDependenciesContainerImpl(this.ribSupport, this.peerTracker, ribPolicies,
+                afiSafiType, this.locRibTableIID);
+    }
+
+    public static <C extends Routes & DataObject & ChoiceIn<Tables>, S extends ChildOf<? super C>>
+                LocRibWriter<C, S> create(
+            final @NonNull RIBSupport<C, S> ribSupport,
+            final @NonNull AfiSafiType afiSafiType,
+            final @NonNull DOMTransactionChain chain,
+            final @NonNull YangInstanceIdentifier ribIId,
+            final @NonNull AsNumber ourAs,
+            final @NonNull DOMDataTreeChangeService dataBroker,
+            final BGPRibRoutingPolicy ribPolicies,
+            final @NonNull BGPPeerTracker peerTracker,
+            final @NonNull PathSelectionMode pathSelectionStrategy) {
+        final LocRibWriter<C, S> ret = new LocRibWriter<>(ribSupport, chain, ribIId, ourAs.getValue(), dataBroker,
+            ribPolicies, peerTracker, afiSafiType, pathSelectionStrategy);
+        ret.init();
+        return ret;
+    }
+
+    private synchronized void init() {
+        final DOMDataTreeWriteTransaction tx = this.chain.newWriteOnlyTransaction();
+        tx.put(LogicalDatastoreType.OPERATIONAL, this.locRibTableIID.node(ATTRIBUTES_NID).node(UPTODATE_NID),
+                RIBNormalizedNodes.ATTRIBUTES_UPTODATE_TRUE);
+        tx.commit().addCallback(new FutureCallback<CommitInfo>() {
+            @Override
+            public void onSuccess(final CommitInfo result) {
+                LOG.trace("Successful commit");
+            }
+
+            @Override
+            public void onFailure(final Throwable trw) {
+                LOG.error("Failed commit", trw);
+            }
+        }, MoreExecutors.directExecutor());
 
-        final YangInstanceIdentifier tableId = target.node(Peer.QNAME).node(Peer.QNAME).node(EffectiveRibIn.QNAME).node(Tables.QNAME).node(tableKey);
-        final DOMDataTreeIdentifier wildcard = new DOMDataTreeIdentifier(LogicalDatastoreType.OPERATIONAL, tableId);
-        this.reg = service.registerDataTreeChangeListener(wildcard, this);
+        this.reg = this.dataBroker.registerDataTreeChangeListener(new DOMDataTreeIdentifier(
+            LogicalDatastoreType.OPERATIONAL, ribIId.node(PEER_NID).node(PEER_NID).node(EFFRIBIN_NID).node(TABLES_NID)
+                .node(locRibTableIID.getLastPathArgument())), this);
     }
 
-    public static LocRibWriter create(@Nonnull final RIBSupportContextRegistry registry, @Nonnull final TablesKey tablesKey, @Nonnull final DOMTransactionChain chain,
-        @Nonnull final YangInstanceIdentifier target, @Nonnull final AsNumber ourAs, @Nonnull final DOMDataTreeChangeService service, @Nonnull final ExportPolicyPeerTracker ep,
-        @Nonnull final PathSelectionMode pathSelectionStrategy, @Nonnull final UnsignedInt32Counter routeCounter) {
-        return new LocRibWriter(registry, chain, target, ourAs.getValue(), service, ep, tablesKey, pathSelectionStrategy, routeCounter);
+    /**
+     * Re-initialize this LocRibWriter with new transaction chain.
+     *
+     * @param newChain new transaction chain
+     */
+    synchronized void restart(final @NonNull DOMTransactionChain newChain) {
+        requireNonNull(newChain);
+        close();
+        this.chain = newChain;
+        init();
     }
 
     @Override
-    public void close() {
-        this.reg.close();
-        // FIXME: wait for the chain to close? unfortunately RIBImpl is the listener, so that may require some work
-        this.chain.close();
+    public synchronized void close() {
+        if (this.reg != null) {
+            this.reg.close();
+            this.reg = null;
+        }
+        if (this.chain != null) {
+            this.chain.close();
+            this.chain = null;
+        }
     }
 
-    @Nonnull
-    private RouteEntry createEntry(final PathArgument routeId) {
-        final RouteEntry ret = this.pathSelectionMode.createRouteEntry(ribSupport.isComplexRoute());
+    private @NonNull RouteEntry<C, S> createEntry(final String routeId) {
+        final RouteEntry<C, S> ret = this.pathSelectionMode.createRouteEntry();
         this.routeEntries.put(routeId, ret);
+        this.totalPrefixesCounter.increment();
         LOG.trace("Created new entry for {}", routeId);
         return ret;
     }
 
+    @Override
+    public synchronized void onInitialData() {
+        // FIXME: we need to do something
+    }
+
     /**
      * We use two-stage processing here in hopes that we avoid duplicate
      * calculations when multiple peers have changed a particular entry.
@@ -127,12 +202,16 @@ final class LocRibWriter implements AutoCloseable, ClusteredDOMDataTreeChangeLis
      * @param changes on supported table
      */
     @Override
-    public void onDataTreeChanged(final Collection<DataTreeCandidate> changes) {
+    @SuppressWarnings("checkstyle:illegalCatch")
+    public synchronized void onDataTreeChanged(final List<DataTreeCandidate> changes) {
+        if (this.chain == null) {
+            LOG.trace("Chain closed, ignoring received data change {} to LocRib {}", changes, this);
+            return;
+        }
         LOG.trace("Received data change {} to LocRib {}", changes, this);
-
-        final DOMDataWriteTransaction tx = this.chain.newWriteOnlyTransaction();
+        final DOMDataTreeWriteTransaction tx = this.chain.newWriteOnlyTransaction();
         try {
-            final Map<RouteUpdateKey, RouteEntry> toUpdate = update(tx, changes);
+            final Map<RouteUpdateKey, RouteEntry<C, S>> toUpdate = update(tx, changes);
 
             if (!toUpdate.isEmpty()) {
                 walkThrough(tx, toUpdate.entrySet());
@@ -140,90 +219,184 @@ final class LocRibWriter implements AutoCloseable, ClusteredDOMDataTreeChangeLis
         } catch (final Exception e) {
             LOG.error("Failed to completely propagate updates {}, state is undefined", changes, e);
         } finally {
-            tx.submit();
+            tx.commit().addCallback(new FutureCallback<CommitInfo>() {
+                @Override
+                public void onSuccess(final CommitInfo result) {
+                    LOG.trace("Successful commit");
+                }
+
+                @Override
+                public void onFailure(final Throwable trw) {
+                    LOG.error("Failed commit", trw);
+                }
+            }, MoreExecutors.directExecutor());
         }
     }
 
-    private Map<RouteUpdateKey, RouteEntry> update(final DOMDataWriteTransaction tx, final Collection<DataTreeCandidate> changes) {
-        final Map<RouteUpdateKey, RouteEntry> ret = new HashMap<>();
-        changes.forEach(tc -> {
+    @SuppressWarnings("unchecked")
+    private Map<RouteUpdateKey, RouteEntry<C, S>> update(final DOMDataTreeWriteOperations tx,
+            final Collection<DataTreeCandidate> changes) {
+        final Map<RouteUpdateKey, RouteEntry<C, S>> ret = new HashMap<>();
+        for (final DataTreeCandidate tc : changes) {
             final DataTreeCandidateNode table = tc.getRootNode();
-            final YangInstanceIdentifier rootPath = tc.getRootPath();
-            final PeerId peerId = IdentifierUtils.peerKeyToPeerId(rootPath);
-            initializeTableWithExistentRoutes(table, peerId, rootPath, tx);
-            updateNodes(table, peerId, tx, ret);
-        });
+            final RouterId peerUuid = RouterId.forPeerId(IdentifierUtils.peerKeyToPeerId(tc.getRootPath()));
+
+            /*
+            Initialize Peer with routes under loc rib
+             */
+            if (!this.routeEntries.isEmpty() && table.getDataBefore().isEmpty()) {
+                final org.opendaylight.protocol.bgp.rib.spi.Peer toPeer
+                        = this.peerTracker.getPeer(peerUuid.getPeerId());
+                if (toPeer != null && toPeer.supportsTable(this.entryDep.getLocalTablesKey())) {
+                    LOG.debug("Peer {} table has been created, inserting existent routes", toPeer.getPeerId());
+                    final List<ActualBestPathRoutes<C, S>> routesToStore = new ArrayList<>();
+                    for (final Entry<String, RouteEntry<C, S>> entry : this.routeEntries.entrySet()) {
+                        final List<ActualBestPathRoutes<C, S>> filteredRoute = entry.getValue()
+                                .actualBestPaths(this.ribSupport, new RouteEntryInfoImpl(toPeer, entry.getKey()));
+                        routesToStore.addAll(filteredRoute);
+                    }
+                    toPeer.initializeRibOut(this.entryDep, routesToStore);
+                }
+            }
+            /*
+            Process new routes from Peer
+             */
+            updateNodes(table, peerUuid, tx, ret);
+        }
         return ret;
     }
 
-    private void initializeTableWithExistentRoutes(final DataTreeCandidateNode table, final PeerId peerIdOfNewPeer, final YangInstanceIdentifier rootPath,
-        final DOMDataWriteTransaction tx) {
-        if (!table.getDataBefore().isPresent() && this.exportPolicyPeerTracker.isTableSupported(peerIdOfNewPeer)) {
-            LOG.debug("Peer {} table has been created, inserting existent routes", peerIdOfNewPeer);
-            final PeerRole newPeerRole = this.exportPolicyPeerTracker.getRole(IdentifierUtils.peerPath(rootPath));
-            final PeerExportGroup peerGroup = this.exportPolicyPeerTracker.getPeerGroup(newPeerRole);
-            this.routeEntries.entrySet().forEach(entry -> entry.getValue().writeRoute(peerIdOfNewPeer, entry.getKey(),
-                rootPath.getParent().getParent().getParent(), peerGroup, this.localTablesKey, this.exportPolicyPeerTracker, this.ribSupport, tx));
+    private void updateNodes(final DataTreeCandidateNode table, final RouterId peerUuid,
+            final DOMDataTreeWriteOperations tx, final Map<RouteUpdateKey, RouteEntry<C, S>> routes) {
+        table.getModifiedChild(ATTRIBUTES_NID).flatMap(DataTreeCandidateNode::getDataAfter).ifPresent(newAttValue -> {
+            LOG.trace("Uptodate found for {}", newAttValue);
+            tx.put(LogicalDatastoreType.OPERATIONAL, this.locRibTableIID.node(ATTRIBUTES_NID), newAttValue);
+        });
+
+        table.getModifiedChild(ROUTES_NID).ifPresent(modifiedRoutes -> {
+            updateRoutesEntries(ribSupport.changedRoutes(modifiedRoutes), peerUuid, routes);
+        });
+    }
+
+    private void updateRoutesEntries(final Collection<DataTreeCandidateNode> collection,
+            final RouterId routerId, final Map<RouteUpdateKey, RouteEntry<C, S>> routes) {
+        for (final DataTreeCandidateNode route : collection) {
+            final PathArgument routeArg = route.getIdentifier();
+            if (!(routeArg instanceof NodeIdentifierWithPredicates routeId)) {
+                LOG.debug("Route {} already deleted", routeArg);
+                return;
+            }
+
+            final String routeKey = ribSupport.extractRouteKey(routeId);
+            final Uint32 pathId = ribSupport.extractPathId(routeId);
+
+            RouteEntry<C, S> entry;
+            switch (route.getModificationType()) {
+                case DELETE:
+                    entry = this.routeEntries.get(routeKey);
+                    if (entry != null) {
+                        this.totalPathsCounter.decrement();
+                        if (entry.removeRoute(routerId, pathId)) {
+                            this.routeEntries.remove(routeKey);
+                            this.totalPrefixesCounter.decrement();
+                            LOG.trace("Removed route from {}", routerId);
+                        }
+                    }
+                    break;
+                case SUBTREE_MODIFIED:
+                case WRITE:
+                    entry = this.routeEntries.get(routeKey);
+                    if (entry == null) {
+                        entry = createEntry(routeKey);
+                    }
+
+                    final NormalizedNode routeAfter = route.getDataAfter().get();
+                    verify(routeAfter instanceof MapEntryNode, "Unexpected route %s", routeAfter);
+                    entry.addRoute(routerId, pathId, (MapEntryNode) routeAfter);
+                    this.totalPathsCounter.increment();
+                    break;
+                default:
+                    throw new IllegalStateException("Unhandled route modification " + route);
+            }
+
+            final RouteUpdateKey routeUpdateKey = new RouteUpdateKey(routerId, routeKey);
+            LOG.debug("Updated route {} entry {}", routeKey, entry);
+            routes.put(routeUpdateKey, entry);
         }
     }
 
-    private void updateNodes(final DataTreeCandidateNode table, final PeerId peerId, final DOMDataWriteTransaction tx,
-        final Map<RouteUpdateKey, RouteEntry> routes) {
-        for (final DataTreeCandidateNode child : table.getChildNodes()) {
-            LOG.debug("Modification type {}", child.getModificationType());
-            if ((Attributes.QNAME).equals(child.getIdentifier().getNodeType())) {
-                if (child.getDataAfter().isPresent()) {
-                    // putting uptodate attribute in
-                    LOG.trace("Uptodate found for {}", child.getDataAfter());
-                    tx.put(LogicalDatastoreType.OPERATIONAL, this.locRibTarget.node(child.getIdentifier()), child.getDataAfter().get());
-                }
+    private void walkThrough(final DOMDataTreeWriteOperations tx,
+            final Set<Entry<RouteUpdateKey, RouteEntry<C, S>>> toUpdate) {
+        final List<StaleBestPathRoute> staleRoutes = new ArrayList<>();
+        final List<AdvertizedRoute<C, S>> newRoutes = new ArrayList<>();
+        for (final Entry<RouteUpdateKey, RouteEntry<C, S>> e : toUpdate) {
+            LOG.trace("Walking through {}", e);
+            final RouteEntry<C, S> entry = e.getValue();
+
+            if (!entry.selectBest(this.ribSupport, this.ourAs)) {
+                LOG.trace("Best path has not changed, continuing");
                 continue;
             }
-            updateRoutesEntries(child, peerId, routes);
+
+            entry.removeStalePaths(this.ribSupport, e.getKey().getRouteId()).ifPresent(staleRoutes::add);
+            newRoutes.addAll(entry.newBestPaths(this.ribSupport, e.getKey().getRouteId()));
         }
+        updateLocRib(newRoutes, staleRoutes, tx);
+        this.peerTracker.getNonInternalPeers().parallelStream()
+                .filter(toPeer -> toPeer.supportsTable(this.entryDep.getLocalTablesKey()))
+                .forEach(toPeer -> toPeer.refreshRibOut(this.entryDep, staleRoutes, newRoutes));
     }
 
-    private void updateRoutesEntries(final DataTreeCandidateNode child, final PeerId peerId, final Map<RouteUpdateKey, RouteEntry> routes) {
-        final UnsignedInteger routerId = RouterIds.routerIdForPeerId(peerId);
-        final Collection<DataTreeCandidateNode> modifiedRoutes = this.ribSupport.changedRoutes(child);
-        for (final DataTreeCandidateNode route : modifiedRoutes) {
-            final PathArgument routeId = this.ribSupport.createRouteKeyPathArgument(route.getIdentifier());
-            RouteEntry entry = this.routeEntries.get(routeId);
-            final Optional<NormalizedNode<?, ?>> maybeData = route.getDataAfter();
-            final Optional<NormalizedNode<?, ?>> maybeDataBefore = route.getDataBefore();
-            if (maybeData.isPresent()) {
-                if (entry == null) {
-                    entry = createEntry(routeId);
-                }
-                entry.addRoute(routerId, this.ribSupport.extractPathId(maybeData.get()), this.attributesIdentifier, maybeData.get());
-            } else if (entry != null && entry.removeRoute(routerId, this.ribSupport.extractPathId(maybeDataBefore.get()))) {
-                this.routeEntries.remove(routeId);
-                LOG.trace("Removed route from {}", routerId);
+    private void updateLocRib(final List<AdvertizedRoute<C, S>> newRoutes, final List<StaleBestPathRoute> staleRoutes,
+            final DOMDataTreeWriteOperations tx) {
+        final YangInstanceIdentifier locRibTarget = this.entryDep.getLocRibTableTarget();
+
+        for (final StaleBestPathRoute staleContainer : staleRoutes) {
+            for (final NodeIdentifierWithPredicates routeId : staleContainer.getStaleRouteKeyIdentifiers()) {
+                final YangInstanceIdentifier routeTarget = ribSupport.createRouteIdentifier(locRibTarget, routeId);
+                LOG.debug("Delete route from LocRib {}", routeTarget);
+                tx.delete(LogicalDatastoreType.OPERATIONAL, routeTarget);
             }
-            final RouteUpdateKey routeUpdateKey = new RouteUpdateKey(peerId, routeId);
-            LOG.debug("Updated route {} entry {}", routeId, entry);
-            routes.put(routeUpdateKey, entry);
         }
-        updateRouteCounter();
+
+        for (final AdvertizedRoute<C, S> advRoute : newRoutes) {
+            final MapEntryNode route = advRoute.getRoute();
+            final NodeIdentifierWithPredicates iid = advRoute.getAddPathRouteKeyIdentifier();
+            final YangInstanceIdentifier locRibRouteTarget = this.ribSupport.createRouteIdentifier(locRibTarget, iid);
+            LOG.debug("Write LocRib route {}", locRibRouteTarget);
+            if (LOG.isTraceEnabled()) {
+                LOG.trace("Write route to LocRib {}", NormalizedNodes.toStringTree(route));
+            }
+            tx.put(LogicalDatastoreType.OPERATIONAL, locRibRouteTarget, route);
+        }
     }
 
-    /**
-     * Update the statistic of loc-rib route
-     */
-    private void updateRouteCounter() {
-        routeCounter.setCount(this.routeEntries.size());
+    @Override
+    public long getPrefixesCount() {
+        return this.totalPrefixesCounter.longValue();
     }
 
-    private void walkThrough(final DOMDataWriteTransaction tx, final Set<Map.Entry<RouteUpdateKey, RouteEntry>> toUpdate) {
-        for (final Map.Entry<RouteUpdateKey, RouteEntry> e : toUpdate) {
-            LOG.trace("Walking through {}", e);
-            final RouteEntry entry = e.getValue();
+    @Override
+    public long getPathsCount() {
+        return this.totalPathsCounter.longValue();
+    }
 
-            if (!entry.selectBest(this.ourAs)) {
-                LOG.trace("Best path has not changed, continuing");
-                continue;
+    TablesKey getTableKey() {
+        return this.ribSupport.getTablesKey();
+    }
+
+    @Override
+    public synchronized void refreshTable(final TablesKey tk, final PeerId peerId) {
+        final org.opendaylight.protocol.bgp.rib.spi.Peer toPeer = this.peerTracker.getPeer(peerId);
+        if (toPeer != null && toPeer.supportsTable(this.entryDep.getLocalTablesKey())) {
+            LOG.debug("Peer {} table has been created, inserting existent routes", toPeer.getPeerId());
+            final List<ActualBestPathRoutes<C, S>> routesToStore = new ArrayList<>();
+            for (final Entry<String, RouteEntry<C, S>> entry : this.routeEntries.entrySet()) {
+                final List<ActualBestPathRoutes<C, S>> filteredRoute = entry.getValue()
+                        .actualBestPaths(this.ribSupport, new RouteEntryInfoImpl(toPeer, entry.getKey()));
+                routesToStore.addAll(filteredRoute);
             }
-            entry.updateRoute(this.localTablesKey, this.exportPolicyPeerTracker, this.locRibTarget, this.ribSupport, tx, e.getKey().getRouteId());
+            toPeer.reEvaluateAdvertizement(this.entryDep, routesToStore);
         }
     }
 }