added mastership to lldpdiscovery

Change-Id: I94bcf04eff134a2bb63325468f26406108608c92

@@ -16,5 +16,15 @@
     <packaging>bundle</packaging>
     <packaging>bundle</packaging>
 
 
     <description>ONOS host tracking provider</description>
     <description>ONOS host tracking provider</description>
+    <dependencies>
+        <dependency>
+            <groupId>org.onlab.onos</groupId>
+            <artifactId>onos-api</artifactId>
+            <classifier>tests</classifier>
+            <scope>test</scope>
+        </dependency>
+    </dependencies>
+
+
 
 
 </project>
 </project>

@@ -5,6 +5,7 @@ import org.apache.felix.scr.annotations.Component;
 import org.apache.felix.scr.annotations.Deactivate;
 import org.apache.felix.scr.annotations.Deactivate;
 import org.apache.felix.scr.annotations.Reference;
 import org.apache.felix.scr.annotations.Reference;
 import org.apache.felix.scr.annotations.ReferenceCardinality;
 import org.apache.felix.scr.annotations.ReferenceCardinality;
+import org.onlab.onos.mastership.MastershipService;
 import org.onlab.onos.net.ConnectPoint;
 import org.onlab.onos.net.ConnectPoint;
 import org.onlab.onos.net.Device;
 import org.onlab.onos.net.Device;
 import org.onlab.onos.net.DeviceId;
 import org.onlab.onos.net.DeviceId;
@@ -46,6 +47,9 @@ public class LLDPLinkProvider extends AbstractProvider implements LinkProvider {
     @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
     @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
     protected PacketService packetSevice;
     protected PacketService packetSevice;
 
 
+    @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
+    protected MastershipService masterService;
+
     private LinkProviderService providerService;
     private LinkProviderService providerService;
 
 
     private final boolean useBDDP = true;
     private final boolean useBDDP = true;
@@ -95,7 +99,7 @@ public class LLDPLinkProvider extends AbstractProvider implements LinkProvider {
             switch (event.type()) {
             switch (event.type()) {
                 case DEVICE_ADDED:
                 case DEVICE_ADDED:
                     discoverers.put(device.id(),
                     discoverers.put(device.id(),
-                                    new LinkDiscovery(device, packetSevice,
+                                    new LinkDiscovery(device, packetSevice, masterService,
                                                       providerService, useBDDP));
                                                       providerService, useBDDP));
                     break;
                     break;
                 case PORT_ADDED:
                 case PORT_ADDED:

@@ -30,10 +30,12 @@ import java.util.concurrent.atomic.AtomicInteger;
 
 
 import org.jboss.netty.util.Timeout;
 import org.jboss.netty.util.Timeout;
 import org.jboss.netty.util.TimerTask;
 import org.jboss.netty.util.TimerTask;
+import org.onlab.onos.mastership.MastershipService;
 import org.onlab.onos.net.ConnectPoint;
 import org.onlab.onos.net.ConnectPoint;
 import org.onlab.onos.net.Device;
 import org.onlab.onos.net.Device;
 import org.onlab.onos.net.DeviceId;
 import org.onlab.onos.net.DeviceId;
 import org.onlab.onos.net.Link.Type;
 import org.onlab.onos.net.Link.Type;
+import org.onlab.onos.net.MastershipRole;
 import org.onlab.onos.net.Port;
 import org.onlab.onos.net.Port;
 import org.onlab.onos.net.PortNumber;
 import org.onlab.onos.net.PortNumber;
 import org.onlab.onos.net.flow.DefaultTrafficTreatment;
 import org.onlab.onos.net.flow.DefaultTrafficTreatment;
@@ -79,22 +81,24 @@ public class LinkDiscovery implements TimerTask {
     private final boolean useBDDP;
     private final boolean useBDDP;
     private final LinkProviderService linkProvider;
     private final LinkProviderService linkProvider;
     private final PacketService pktService;
     private final PacketService pktService;
+    private final MastershipService mastershipService;
     private Timeout timeout;
     private Timeout timeout;
 
 
     /**
     /**
      * Instantiates discovery manager for the given physical switch. Creates a
      * Instantiates discovery manager for the given physical switch. Creates a
      * generic LLDP packet that will be customized for the port it is sent out on.
      * generic LLDP packet that will be customized for the port it is sent out on.
      * Starts the the timer for the discovery process.
      * Starts the the timer for the discovery process.
-     *
      *  @param device the physical switch
      *  @param device the physical switch
+     * @param masterService
      * @param useBDDP flag to also use BDDP for discovery
      * @param useBDDP flag to also use BDDP for discovery
      */
      */
     public LinkDiscovery(Device device, PacketService pktService,
     public LinkDiscovery(Device device, PacketService pktService,
-             LinkProviderService providerService, Boolean... useBDDP) {
+                         MastershipService masterService, LinkProviderService providerService, Boolean... useBDDP) {
         this.device = device;
         this.device = device;
         this.probeRate = 3000;
         this.probeRate = 3000;
         this.linkProvider = providerService;
         this.linkProvider = providerService;
         this.pktService = pktService;
         this.pktService = pktService;
+        this.mastershipService = masterService;
         this.slowPorts = Collections.synchronizedSet(new HashSet<Long>());
         this.slowPorts = Collections.synchronizedSet(new HashSet<Long>());
         this.fastPorts = Collections.synchronizedSet(new HashSet<Long>());
         this.fastPorts = Collections.synchronizedSet(new HashSet<Long>());
         this.portProbeCount = new HashMap<>();
         this.portProbeCount = new HashMap<>();
@@ -337,6 +341,8 @@ public class LinkDiscovery implements TimerTask {
     }
     }
 
 
     private void sendProbes(Long portNumber) {
     private void sendProbes(Long portNumber) {
+       if (mastershipService.getLocalRole(this.device.id()) ==
+               MastershipRole.MASTER) {
            OutboundPacket pkt = this.createOutBoundLLDP(portNumber);
            OutboundPacket pkt = this.createOutBoundLLDP(portNumber);
            pktService.emit(pkt);
            pktService.emit(pkt);
            if (useBDDP) {
            if (useBDDP) {
@@ -344,5 +350,6 @@ public class LinkDiscovery implements TimerTask {
                pktService.emit(bpkt);
                pktService.emit(bpkt);
            }
            }
        }
        }
+    }
 
 
 }
 }