[AIRFLOW-2860] DruidHook: time check is wrong (#3745)

apache · Sep 15, 2018 · 71905ca · 71905ca
1 parent e47910f
commit 71905ca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/airflow/hooks/druid_hook.py b/airflow/hooks/druid_hook.py
@@ -81,8 +81,6 @@ def submit_indexing_job(self, json_index_spec):
 
             self.log.info("Job still running for %s seconds...", sec)
 
-            sec = sec + 1
-
             if self.max_ingestion_time and sec > self.max_ingestion_time:
                 # ensure that the job gets killed if the max ingestion time is exceeded
                 requests.post("{0}/{1}/shutdown".format(url, druid_task_id))
@@ -91,6 +89,8 @@ def submit_indexing_job(self, json_index_spec):
 
             time.sleep(self.timeout)
 
+            sec = sec + self.timeout
+
             status = req_status.json()['status']['status']
             if status == 'RUNNING':
                 running = True