CORD-537 Added removing vSG

- Remove vSG from ONOS host list with allowed-ip-pairs update event
- Remove vSG from ONOS host list when vSG VM removed

Change-Id: Ic93fceec67d3faada795805d4c4a56ab5187d7e2

@@ -291,17 +291,9 @@ public class CordVtn extends AbstractProvider implements CordVtnService, HostPro
 
 
     @Override
     @Override
     public void removeServiceVm(ConnectPoint connectPoint) {
     public void removeServiceVm(ConnectPoint connectPoint) {
-        Host host = hostService.getConnectedHosts(connectPoint)
+        hostService.getConnectedHosts(connectPoint)
                 .stream()
                 .stream()
-                .findFirst()
-                .orElse(null);
-
-        if (host == null) {
-            log.debug("No host is connected on {}", connectPoint.toString());
-            return;
-        }
-
-        hostProvider.hostVanished(host.id());
+                .forEach(host -> hostProvider.hostVanished(host.id()));
     }
     }
 
 
     @Override
     @Override
@@ -322,6 +314,14 @@ public class CordVtn extends AbstractProvider implements CordVtnService, HostPro
                         entry.getValue(),
                         entry.getValue(),
                         serviceVlan));
                         serviceVlan));
 
 
+        hostService.getConnectedHosts(vSgVm.location()).stream()
+                .filter(host -> !host.mac().equals(vSgVm.mac()))
+                .filter(host -> !vSgs.values().contains(host.mac()))
+                .forEach(host -> {
+                    log.info("Removed vSG {}", host.toString());
+                    hostProvider.hostVanished(host.id());
+                });
+
         ruleInstaller.populateSubscriberGatewayRules(vSgVm, vSgs.keySet());
         ruleInstaller.populateSubscriberGatewayRules(vSgVm, vSgs.keySet());
     }
     }
 
 
@@ -333,11 +333,12 @@ public class CordVtn extends AbstractProvider implements CordVtnService, HostPro
      * @param vSgMac vSG mac address
      * @param vSgMac vSG mac address
      * @param serviceVlan service vlan
      * @param serviceVlan service vlan
      */
      */
-    public void addVirtualSubscriberGateway(Host vSgHost, IpAddress vSgIp, MacAddress vSgMac, String serviceVlan) {
+    private void addVirtualSubscriberGateway(Host vSgHost, IpAddress vSgIp, MacAddress vSgMac,
+                                             String serviceVlan) {
         HostId hostId = HostId.hostId(vSgMac);
         HostId hostId = HostId.hostId(vSgMac);
         Host host = hostService.getHost(hostId);
         Host host = hostService.getHost(hostId);
         if (host != null) {
         if (host != null) {
-            log.debug("vSG with {} already exists", vSgMac.toString());
+            log.trace("vSG with {} already exists", vSgMac.toString());
             return;
             return;
         }
         }
 
 
@@ -524,7 +525,7 @@ public class CordVtn extends AbstractProvider implements CordVtnService, HostPro
     private void serviceVmAdded(Host host) {
     private void serviceVmAdded(Host host) {
         String vNetId = host.annotations().value(SERVICE_ID);
         String vNetId = host.annotations().value(SERVICE_ID);
         if (vNetId == null) {
         if (vNetId == null) {
-            // ignore this host, it not a VM we injected or a vSG
+            // ignore this host, it is not the service VM, or it's a vSG
             return;
             return;
         }
         }
 
 
@@ -583,12 +584,16 @@ public class CordVtn extends AbstractProvider implements CordVtnService, HostPro
      * @param host host
      * @param host host
      */
      */
     private void serviceVmRemoved(Host host) {
     private void serviceVmRemoved(Host host) {
-        if (host.annotations().value(OPENSTACK_VM_ID) == null) {
-            // this host was not injected from CordVtn, just return
+        String vNetId = host.annotations().value(SERVICE_ID);
+        if (vNetId == null) {
+            // ignore it, it's not the service VM or it's a vSG
+            String serviceVlan = host.annotations().value(S_TAG);
+            if (serviceVlan != null) {
+                log.info("vSG {} removed", host.id());
+            }
             return;
             return;
         }
         }
 
 
-        String vNetId = host.annotations().value(SERVICE_ID);
         OpenstackNetwork vNet = openstackService.network(vNetId);
         OpenstackNetwork vNet = openstackService.network(vNetId);
         if (vNet == null) {
         if (vNet == null) {
             log.warn("Failed to get OpenStack network {} for VM {}({}).",
             log.warn("Failed to get OpenStack network {} for VM {}({}).",

@@ -233,7 +233,7 @@ public class CordVtnArpProxy {
                 .orElse(null);
                 .orElse(null);
 
 
         if (host != null) {
         if (host != null) {
-            log.debug("Found MAC from host service for {}", targetIp.toString());
+            log.trace("Found MAC from host service for {}", targetIp.toString());
             return host.mac();
             return host.mac();
         } else {
         } else {
             return MacAddress.NONE;
             return MacAddress.NONE;