From 2feab6e9c60a80415d87cf25e8b19298b1adea47 Mon Sep 17 00:00:00 2001
From: Petar Efnushev <petar.efnushev@gmail.com>
Date: Fri, 26 Aug 2016 10:30:53 +0200
Subject: [PATCH] Add prerun checks for postgres & solr database

---
 rootfs/setup/prerun.py     | 49 ++++++++++++++++++++++++++++++++++++++
 rootfs/setup/start_ckan.sh |  7 +++++-
 2 files changed, 55 insertions(+), 1 deletion(-)

diff --git a/rootfs/setup/prerun.py b/rootfs/setup/prerun.py
index 7b1eb16..842763a 100644
--- a/rootfs/setup/prerun.py
+++ b/rootfs/setup/prerun.py
@@ -1,10 +1,57 @@
 import os
 import sys
 import subprocess
+import psycopg2
+import urllib2
 
 
 ckan_ini = os.environ.get('CKAN_INI', '/srv/app/production.ini')
 
+RETRY = 5
+
+def check_db_connection(retry=None):
+    
+    if retry is None:
+        retry = RETRY
+    elif retry == 0:
+        print '[prerun] Giving up after 5 tries...'
+        sys.exit(1)
+    
+    conn_str = os.environ.get('CKAN_SQLALCHEMY_URL', '')
+    try:
+        connection = psycopg2.connect(conn_str)
+        
+    except psycopg2.Error as e:
+        print str(e)
+        print '[prerun] Unable to connect to the database...try again in a while.'
+        import time
+        time.sleep(10)
+        check_db_connection(retry = retry - 1)
+    else:
+        connection.close()
+        
+def check_solr_connection(retry=None):
+    
+    if retry is None:
+        retry = RETRY
+    elif retry == 0:
+        print '[prerun] Giving up after 5 tries...'
+        sys.exit(1)
+    
+    url = os.environ.get('CKAN_SOLR_URL', '')
+    search_url = '{url}/select/?q=*&wt=json'.format(url=url)
+    
+    try:
+        connection = urllib2.urlopen(search_url)
+    except urllib2.URLError as e:
+        print str(e)
+        print '[prerun] Unable to connect to solr...try again in a while.'
+        import time
+        time.sleep(10)
+        check_solr_connection(retry = retry - 1)
+    else:
+        eval(connection.read())
+
 
 def init_db():
 
@@ -68,5 +115,7 @@ if __name__ == '__main__':
     if maintenance:
         print '[prerun] Maintenance mode, skipping setup...'
     else:
+        check_db_connection()
+        check_solr_connection()
         init_db()
         create_sysadmin()
diff --git a/rootfs/setup/start_ckan.sh b/rootfs/setup/start_ckan.sh
index 209d664..dd8a8da 100755
--- a/rootfs/setup/start_ckan.sh
+++ b/rootfs/setup/start_ckan.sh
@@ -1,3 +1,8 @@
 #!/bin/bash
 python prerun.py
-gunicorn --log-file=- --paste production.ini
+if [ $? -eq 0 ]
+then
+  gunicorn --log-file=- --paste production.ini
+else
+  echo "[prerun] failed...not starting CKAN."
+fi