Merge pull request apache#243 from mistercrunch/sensor_sticky_conn

Insure keeping the same connection through the sensor's lifecycle
jakromm · Jun 5, 2015 · c8891f2 · c8891f2
2 parents c301ff9 + ee7d326
commit c8891f2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/airflow/operators/sensors.py b/airflow/operators/sensors.py
@@ -172,8 +172,9 @@ def poke(self):
         logging.info(
             'Poking for table {self.schema}.{self.table}, '
             'partition {self.partition}'.format(**locals()))
-        self.hook = hooks.HiveMetastoreHook(
-            metastore_conn_id=self.metastore_conn_id)
+        if not hasattr(self, 'hook'):
+            self.hook = hooks.HiveMetastoreHook(
+                metastore_conn_id=self.metastore_conn_id)
         return self.hook.check_for_partition(
             self.schema, self.table, self.partition)