From fd7ddd3de00578b6df44f665d70459b360802dcf Mon Sep 17 00:00:00 2001
From: John Bodley <4567245+john-bodley@users.noreply.github.com>
Date: Sat, 16 Dec 2017 11:59:41 +1300
Subject: [PATCH] [health] Adding DB check to /health (#4062)

---
 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'