X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=bgp%2Frib-impl%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fprotocol%2Fbgp%2Frib%2Fimpl%2FBGPPeer.java;h=10d3ab5965f94a012f361e6a2b29999d7ffea072;hb=feb60ff47fd4150ce14915506d380704fbf36595;hp=a56a2c05cf7a96d9c1e0393a8eba3ddfe7696dc5;hpb=a01e8a8629311b3480b2085aca40feea539f4ba0;p=bgpcep.git

diff --git a/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/BGPPeer.java b/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/BGPPeer.java
index a56a2c05cf..10d3ab5965 100644
--- a/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/BGPPeer.java
+++ b/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/BGPPeer.java
@@ -9,12 +9,14 @@ package org.opendaylight.protocol.bgp.rib.impl;
 
 import java.util.Set;
 
-import org.opendaylight.protocol.bgp.concepts.BGPTableType;
-import org.opendaylight.protocol.bgp.parser.BGPMessage;
 import org.opendaylight.protocol.bgp.parser.BGPSession;
 import org.opendaylight.protocol.bgp.parser.BGPSessionListener;
 import org.opendaylight.protocol.bgp.parser.BGPTerminationReason;
-import org.opendaylight.protocol.bgp.parser.BGPUpdateMessage;
+import org.opendaylight.protocol.bgp.rib.spi.Peer;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.Update;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130918.BgpTableType;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
+import org.opendaylight.yangtools.yang.binding.Notification;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -26,9 +28,9 @@ import com.google.common.base.Preconditions;
  * Class representing a peer. We have a single instance for each peer, which provides translation from BGP events into
  * RIB actions.
  */
-public final class BGPPeer implements BGPSessionListener {
+public final class BGPPeer implements BGPSessionListener, Peer {
 	private static final Logger logger = LoggerFactory.getLogger(BGPPeer.class);
-	private Set<BGPTableType> tables;
+	private Set<TablesKey> tables;
 	private final String name;
 	private final RIBImpl rib;
 
@@ -38,10 +40,9 @@ public final class BGPPeer implements BGPSessionListener {
 	}
 
 	@Override
-	public void onMessage(final BGPSession session, final BGPMessage message) {
-		if (message instanceof BGPUpdateMessage) {
-			final BGPUpdateMessage m = (BGPUpdateMessage) message;
-			this.rib.updateTables(this, m.getAddedObjects(), m.getRemovedObjects());
+	public void onMessage(final BGPSession session, final Notification message) {
+		if (message instanceof Update) {
+			this.rib.updateTables(this, (Update) message);
 		} else {
 			logger.info("Ignoring unhandled message class " + message.getClass());
 		}
@@ -50,14 +51,20 @@ public final class BGPPeer implements BGPSessionListener {
 	@Override
 	public void onSessionUp(final BGPSession session) {
 		logger.info("Session with peer {} went up with tables: {}", this.name, session.getAdvertisedTableTypes());
+
+		for (final BgpTableType t : session.getAdvertisedTableTypes()) {
+			this.tables.add(new TablesKey(t.getAfi(), t.getSafi()));
+		}
 	}
 
 	@Override
 	public void onSessionDown(final BGPSession session, final Exception e) {
 		// FIXME: support graceful restart
-		for (final BGPTableType t : this.tables) {
-			this.rib.clearTable(this, t);
+		for (final TablesKey key : this.tables) {
+			this.rib.clearTable(this, key);
 		}
+
+		this.tables.clear();
 	}
 
 	@Override
@@ -66,7 +73,7 @@ public final class BGPPeer implements BGPSessionListener {
 	}
 
 	@Override
-	public final String toString() {
+	public String toString() {
 		return addToStringAttributes(Objects.toStringHelper(this)).toString();
 	}