retry connection if pymonetdbd is not ready yet

chriskamphuis · Jun 14, 2019 · 4cb9b41 · 4cb9b41
1 parent 682ec78
commit 4cb9b41
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/src/main/python/search_collection.py b/src/main/python/search_collection.py
@@ -6,6 +6,7 @@
 import random
 import string
 from topic_reader import TopicReader
+import time
 
 class SearchCollection:
 
@@ -52,11 +53,19 @@ def getConnectionCursor(self):
         print("CREATE DATABASE") 
         dbname = 'robust04'
 
-        print("CREATE CONNECTION")
-        connection = pymonetdb.connect(username='monetdb',
-                                       password='monetdb',
-                                       hostname='localhost', 
-                                       database=dbname)
+        connection = None 
+        attempt = 0
+        while connection is None or attempt > 20:
+            print("CREATE CONNECTION")
+            try:
+                connection = pymonetdb.connect(username='monetdb',
+                                               password='monetdb',
+                                               hostname='localhost', 
+                                               database=dbname)
+            except:
+                attempt += 1
+                time.sleep(5) 
+
         cursor = connection.cursor()
         return cursor