DatabaseManager: add periodic leader advertisements

Change-Id: I6e9244a06191fe0f2dd5eaed7e043e84d704bfcd

 package org.onlab.onos.store.service.impl;
 package org.onlab.onos.store.service.impl;
 
 
+import static java.util.concurrent.Executors.newSingleThreadScheduledExecutor;
+import static org.onlab.util.Tools.namedThreads;
 import static org.slf4j.LoggerFactory.getLogger;
 import static org.slf4j.LoggerFactory.getLogger;
 
 
 import java.io.File;
 import java.io.File;
@@ -11,6 +13,7 @@ import java.util.Map;
 import java.util.Set;
 import java.util.Set;
 import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.ExecutionException;
+import java.util.concurrent.ScheduledExecutorService;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeUnit;
 
 
 import net.kuujo.copycat.Copycat;
 import net.kuujo.copycat.Copycat;
@@ -105,6 +108,10 @@ public class DatabaseManager implements DatabaseService, DatabaseAdminService {
 
 
     private boolean autoAddMember = false;
     private boolean autoAddMember = false;
 
 
+    private ScheduledExecutorService executor;
+
+    private volatile LeaderElectEvent myLeaderEvent = null;
+
     @Activate
     @Activate
     public void activate() throws InterruptedException, ExecutionException {
     public void activate() throws InterruptedException, ExecutionException {
 
 
@@ -176,6 +183,11 @@ public class DatabaseManager implements DatabaseService, DatabaseAdminService {
         // of the Raft cluster.
         // of the Raft cluster.
         if (copycat != null) {
         if (copycat != null) {
             copycat.start().get();
             copycat.start().get();
+
+            executor =
+                    newSingleThreadScheduledExecutor(namedThreads("db-heartbeat-%d"));
+            executor.scheduleWithFixedDelay(new LeaderAdvertiser(), 5, 2, TimeUnit.SECONDS);
+
         }
         }
 
 
         client.waitForLeader();
         client.waitForLeader();
@@ -189,6 +201,9 @@ public class DatabaseManager implements DatabaseService, DatabaseAdminService {
 
 
     @Deactivate
     @Deactivate
     public void deactivate() {
     public void deactivate() {
+        if (executor != null) {
+            executor.shutdownNow();
+        }
         clusterService.removeListener(clusterEventListener);
         clusterService.removeListener(clusterEventListener);
         // TODO: ClusterCommunicationService must support more than one
         // TODO: ClusterCommunicationService must support more than one
         // handler per message subject.
         // handler per message subject.
@@ -231,9 +246,9 @@ public class DatabaseManager implements DatabaseService, DatabaseAdminService {
                     throw e;
                     throw e;
                 } else {
                 } else {
                     log.debug("Failed to listTables. Will retry...", e);
                     log.debug("Failed to listTables. Will retry...", e);
-                    retries++;
                 }
                 }
             }
             }
+            retries++;
         } while (true);
         } while (true);
     }
     }
 
 
@@ -395,17 +410,47 @@ public class DatabaseManager implements DatabaseService, DatabaseAdminService {
         }
         }
     }
     }
 
 
+    private final class LeaderAdvertiser implements Runnable {
+
+        @Override
+        public void run() {
+            try {
+                LeaderElectEvent event = myLeaderEvent;
+                if (event != null) {
+                    log.trace("Broadcasting RAFT_LEADER_ELECTION_EVENT: {}", event);
+                    // This node just became the leader.
+                    clusterCommunicator.broadcastIncludeSelf(
+                            new ClusterMessage(
+                                    clusterService.getLocalNode().id(),
+                                    RAFT_LEADER_ELECTION_EVENT,
+                                    ClusterMessagingProtocol.SERIALIZER.encode(event)));
+                }
+            } catch (Exception e) {
+                log.debug("LeaderAdvertiser failed with exception", e);
+            }
+        }
+
+    }
+
     private final class RaftLeaderElectionMonitor implements EventHandler<LeaderElectEvent> {
     private final class RaftLeaderElectionMonitor implements EventHandler<LeaderElectEvent> {
         @Override
         @Override
         public void handle(LeaderElectEvent event) {
         public void handle(LeaderElectEvent event) {
             try {
             try {
+                log.debug("Received LeaderElectEvent: {}", event);
                 if (clusterConfig.getLocalMember() != null && event.leader().equals(clusterConfig.getLocalMember())) {
                 if (clusterConfig.getLocalMember() != null && event.leader().equals(clusterConfig.getLocalMember())) {
+                    log.debug("Broadcasting RAFT_LEADER_ELECTION_EVENT");
+                    myLeaderEvent = event;
                     // This node just became the leader.
                     // This node just became the leader.
                     clusterCommunicator.broadcastIncludeSelf(
                     clusterCommunicator.broadcastIncludeSelf(
                             new ClusterMessage(
                             new ClusterMessage(
                                     clusterService.getLocalNode().id(),
                                     clusterService.getLocalNode().id(),
                                     RAFT_LEADER_ELECTION_EVENT,
                                     RAFT_LEADER_ELECTION_EVENT,
                                     ClusterMessagingProtocol.SERIALIZER.encode(event)));
                                     ClusterMessagingProtocol.SERIALIZER.encode(event)));
+                } else {
+                    if (myLeaderEvent != null) {
+                        log.debug("This node is no longer the Leader");
+                    }
+                    myLeaderEvent = null;
                 }
                 }
             } catch (IOException e) {
             } catch (IOException e) {
                 log.error("Failed to broadcast raft leadership change event", e);
                 log.error("Failed to broadcast raft leadership change event", e);