From ac201955fa7b51debbb699626f521523ef4658bc Mon Sep 17 00:00:00 2001
From: Madan Jampani <madan@onlab.us>
Date: Tue, 18 Nov 2014 10:06:01 -0800
Subject: [PATCH] DistributedLockManger creates a locks table if one does not
 exist

Change-Id: Ifa766ad441f677a4071b68d8f6caa564cf320869
---
 .../service/impl/DistributedLockManager.java  | 21 +++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/core/store/dist/src/main/java/org/onlab/onos/store/service/impl/DistributedLockManager.java b/core/store/dist/src/main/java/org/onlab/onos/store/service/impl/DistributedLockManager.java
index a8941aaacb..ea0f7d5634 100644
--- a/core/store/dist/src/main/java/org/onlab/onos/store/service/impl/DistributedLockManager.java
+++ b/core/store/dist/src/main/java/org/onlab/onos/store/service/impl/DistributedLockManager.java
@@ -5,6 +5,7 @@ import static org.slf4j.LoggerFactory.getLogger;
 
 import java.util.Iterator;
 import java.util.List;
+import java.util.Set;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
@@ -20,6 +21,8 @@ import org.onlab.onos.cluster.ClusterService;
 import org.onlab.onos.store.cluster.messaging.ClusterCommunicationService;
 import org.onlab.onos.store.cluster.messaging.ClusterMessage;
 import org.onlab.onos.store.cluster.messaging.ClusterMessageHandler;
+import org.onlab.onos.store.service.DatabaseAdminService;
+import org.onlab.onos.store.service.DatabaseException;
 import org.onlab.onos.store.service.DatabaseService;
 import org.onlab.onos.store.service.Lock;
 import org.onlab.onos.store.service.LockEventListener;
@@ -41,12 +44,17 @@ public class DistributedLockManager implements LockService {
 
     public static final String ONOS_LOCK_TABLE_NAME = "onos-locks";
 
+    public static final int DEAD_LOCK_TIMEOUT_MS = 5000;
+
     private final ListMultimap<String, LockRequest> locksToAcquire =
                 Multimaps.synchronizedListMultimap(LinkedListMultimap.<String, LockRequest>create());
 
     @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
     private ClusterCommunicationService clusterCommunicator;
 
+    @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
+    private DatabaseAdminService databaseAdminService;
+
     @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY)
     private DatabaseService databaseService;
 
@@ -55,9 +63,22 @@ public class DistributedLockManager implements LockService {
 
     @Activate
     public void activate() {
+        try {
+            Set<String> tableNames = databaseAdminService.listTables();
+            if (!tableNames.contains(ONOS_LOCK_TABLE_NAME)) {
+                if (databaseAdminService.createTable(ONOS_LOCK_TABLE_NAME, DEAD_LOCK_TIMEOUT_MS)) {
+                    log.info("Created {} table.", ONOS_LOCK_TABLE_NAME);
+                }
+            }
+        } catch (DatabaseException e) {
+            log.error("DistributedLockManager#activate failed.", e);
+            throw e;
+        }
+
         clusterCommunicator.addSubscriber(
                 DatabaseStateMachine.DATABASE_UPDATE_EVENTS,
                 new LockEventMessageListener());
+
         log.info("Started.");
 
     }