Logging which host is being used

Useful when using multiple hosts with same conn_id and one is defective
bootinge · Jun 4, 2015 · b97e478 · b97e478
1 parent 1fe604f
commit b97e478
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/airflow/hooks/base_hook.py b/airflow/hooks/base_hook.py
@@ -1,3 +1,4 @@
+import logging
 import random
 
 from airflow import settings
@@ -28,7 +29,9 @@ def get_connections(self, conn_id):
         return db
 
     def get_connection(self, conn_id):
-        return random.choice(self.get_connections(conn_id))
+        conn = random.choice(self.get_connections(conn_id))
+        logging.info("Using connection to: " + conn.host)
+        return conn
 
     def get_conn(self):
         raise NotImplemented()