Fixes for DVR

[groupbasedpolicy.git] / renderers / vpp / src / main / java / org / opendaylight / groupbasedpolicy / renderer / vpp / lisp / info / container / states / VrfState.java

diff --git a/renderers/vpp/src/main/java/org/opendaylight/groupbasedpolicy/renderer/vpp/lisp/info/container/states/VrfState.java b/renderers/vpp/src/main/java/org/opendaylight/groupbasedpolicy/renderer/vpp/lisp/info/container/states/VrfState.java
index c3d34663d98df49da462eb9bb26d19e58127b34f..21cd1c5598349ac1e824c14eb5d1898c844e395f 100644 (file)
--- a/renderers/vpp/src/main/java/org/opendaylight/groupbasedpolicy/renderer/vpp/lisp/info/container/states/VrfState.java
+++ b/renderers/vpp/src/main/java/org/opendaylight/groupbasedpolicy/renderer/vpp/lisp/info/container/states/VrfState.java
@@ -10,9 +10,6 @@ package org.opendaylight.groupbasedpolicy.renderer.vpp.lisp.info.container.state
 import org.apache.commons.lang3.mutable.MutableLong;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
 
-/**
- * Created by Shakib Ahmed on 7/17/17.
- */
 public class VrfState {
     private SubnetHolder subnetHolder;
     private String protocolName;
@@ -42,6 +39,17 @@ public class VrfState {
     }
 
     public void removePortIpFromVrf(String portSubnetUuid, Ipv4Address portIp) {
-        subnetHolder.getSubnetState(portSubnetUuid).removeIp(portIp);
+        if (subnetHolder.subnetStateContains(portSubnetUuid)) {
+            subnetHolder.getSubnetState(portSubnetUuid).removeIp(portIp);
+        }
+    }
+
+    public int subnetCount() {
+        return subnetHolder.subnetHolderCount();
+    }
+
+    @Override public String toString() {
+        return "VrfState{" + "subnetHolder=" + subnetHolder + ", protocolName='" + protocolName + '\''
+            + ", nextRouteId=" + nextRouteId + '}';
     }
 }