BGPCEP-578: Extended peer-group support

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

diff --git a/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/RouteUpdateKey.java b/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/RouteUpdateKey.java
index c3693bda79d2d90d1f4718b38211350dbf504422..ff16233ce83d7bd106a51e6ee05ea3b16c79aff9 100644 (file)
--- a/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/RouteUpdateKey.java
+++ b/bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/RouteUpdateKey.java
@@ -9,27 +9,27 @@ package org.opendaylight.protocol.bgp.rib.impl;
 
 import static java.util.Objects.requireNonNull;
 
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.PeerId;
-import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
+import com.google.common.primitives.UnsignedInteger;
+import org.opendaylight.yangtools.yang.binding.Identifier;
 
 /**
  * Combined key formed as a concatenation of source peer and route identifiers.
  * This is used to internally track updates which need to be processed.
  */
 final class RouteUpdateKey {
-    private final PeerId peerId;
-    private final PathArgument routeId;
+    private final UnsignedInteger peerId;
+    private final Identifier routeId;
 
-    RouteUpdateKey(final PeerId peerId, final PathArgument routeId) {
+    RouteUpdateKey(final UnsignedInteger peerId, final Identifier routeKey) {
         this.peerId = requireNonNull(peerId);
-        this.routeId = requireNonNull(routeId);
+        this.routeId = requireNonNull(routeKey);
     }
 
-    PeerId getPeerId() {
+    UnsignedInteger getPeerId() {
         return this.peerId;
     }
 
-    PathArgument getRouteId() {
+    Identifier getRouteId() {
         return this.routeId;
     }