From 25cda8d2bd900424bd4aa7da461e0c9f18f2b04f Mon Sep 17 00:00:00 2001
From: mprahl <mprahl@redhat.com>
Date: Wed, 23 Jan 2019 15:12:47 -0500
Subject: [PATCH] Shutdown fedmsg-hub when SQLAlchemy can no longer resolve DNS
 records

---
 module_build_service/scheduler/consumer.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/module_build_service/scheduler/consumer.py b/module_build_service/scheduler/consumer.py
index 3a54bfb5..e33621e8 100644
--- a/module_build_service/scheduler/consumer.py
+++ b/module_build_service/scheduler/consumer.py
@@ -39,6 +39,7 @@ import koji
 import fedmsg.consumers
 import moksha.hub
 import six
+import sqlalchemy.exc
 
 from module_build_service.utils import module_build_state_from_msg
 import module_build_service.messaging
@@ -160,6 +161,13 @@ class MBSConsumer(fedmsg.consumers.FedmsgConsumer):
         try:
             with models.make_session(conf) as session:
                 self.process_message(session, msg)
+        except sqlalchemy.exc.OperationalError as error:
+            if 'could not translate host name' in str(error):
+                log.exception(
+                    "SQLAlchemy can't resolve DNS records. Scheduling fedmsg-hub to shutdown.")
+                self.shutdown()
+            else:
+                raise
         except Exception:
             log.exception('Failed while handling {0!r}'.format(msg))