From d37d6178d308bf339f646c2e2fb33e1c32ff58f1 Mon Sep 17 00:00:00 2001
From: John Bodley <john.bodley@airbnb.com>
Date: Thu, 14 Dec 2017 22:13:23 +1300
Subject: [PATCH] [health] Adding DB check to /health

---
 superset/views/core.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/superset/views/core.py b/superset/views/core.py
index 00254b4ca27ba..88014194e5895 100755
--- a/superset/views/core.py
+++ b/superset/views/core.py
@@ -28,6 +28,7 @@
 import sqlalchemy as sqla
 from sqlalchemy import create_engine
 from sqlalchemy.engine.url import make_url
+from sqlalchemy.exc import OperationalError
 from unidecode import unidecode
 from werkzeug.routing import BaseConverter
 from werkzeug.utils import secure_filename
@@ -654,6 +655,11 @@ class LogModelView(SupersetModelView):
 
 @app.route('/health')
 def health():
+    try:
+        db.session.execute('SELECT 1')
+    except OperationalError:
+        return Response('BAD', status=500)
+
     return 'OK'