applications/topology-lldp-discovery/src/main/java/org/opendaylight/openflowplugin/applications/topology/lldp/LLDPLinkAger.java

   1 /*
   2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.openflowplugin.applications.topology.lldp;
   9
  10 import java.util.Date;
  11 import java.util.Map;
  12 import java.util.Map.Entry;
  13 import java.util.Timer;
  14 import java.util.TimerTask;
  15 import java.util.concurrent.ConcurrentHashMap;
  16 import org.opendaylight.controller.sal.binding.api.NotificationProviderService;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.topology.discovery.rev130819.LinkDiscovered;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.topology.discovery.rev130819.LinkRemovedBuilder;
  19
  20
  21 public class LLDPLinkAger implements AutoCloseable {
  22     private final long linkExpirationTime;
  23     private final Map<LinkDiscovered, Date> linkToDate;
  24     private final Timer timer;
  25     private final NotificationProviderService notificationService;
  26
  27     /**
  28      * default ctor - start timer
  29      */
  30     public LLDPLinkAger(final long lldpInterval, final long linkExpirationTime,
  31             final NotificationProviderService notificationService) {
  32         this.linkExpirationTime = linkExpirationTime;
  33         this.notificationService = notificationService;
  34         linkToDate = new ConcurrentHashMap<>();
  35         timer = new Timer();
  36         timer.schedule(new LLDPAgingTask(), 0, lldpInterval);
  37     }
  38
  39     public void put(LinkDiscovered link) {
  40         Date expires = new Date();
  41         expires.setTime(expires.getTime() + linkExpirationTime);
  42         linkToDate.put(link, expires);
  43     }
  44
  45     @Override
  46     public void close() {
  47         timer.cancel();
  48         linkToDate.clear();
  49     }
  50
  51     private class LLDPAgingTask extends TimerTask {
  52
  53         @Override
  54         public void run() {
  55             for (Entry<LinkDiscovered,Date> entry : linkToDate.entrySet()) {
  56                 LinkDiscovered link = entry.getKey();
  57                 Date expires = entry.getValue();
  58                 Date now = new Date();
  59                 if(now.after(expires)) {
  60                     if (notificationService != null) {
  61                         LinkRemovedBuilder lrb = new LinkRemovedBuilder(link);
  62                         notificationService.publish(lrb.build());
  63                         linkToDate.remove(link);
  64                     }
  65                 }
  66             }
  67
  68         }
  69
  70     }
  71 }
  72