danccooper commented on a change in pull request #6377: [AIRFLOW-5589] monitor 
pods by labels instead of names
URL: https://github.com/apache/airflow/pull/6377#discussion_r342144520
 
 

 ##########
 File path: airflow/contrib/operators/kubernetes_pod_operator.py
 ##########
 @@ -112,55 +113,60 @@ class KubernetesPodOperator(BaseOperator):  # pylint: 
disable=too-many-instance-
     """
     template_fields = ('cmds', 'arguments', 'env_vars', 'config_file')
 
+    @staticmethod
+    def create_labels_for_pod(context):
+        """
+        Generate labels for the pod s.t. we can track it in case of Operator 
crash
+
+        :param context:
+        :return:
+        """
+        labels = {
+            'dag_id': context['dag'].dag_id,
+            'task_id': context['task'].task_id,
+            'exec_date': context['ts'],
+            'try_number': context['ti'].try_number,
 
 Review comment:
   Hey @ashb @dimberman how do feel about removing the try_number changes and 
allowing the re-attach to a previous pod?  This is the logic I've been running 
for a while now.  The situation I can think of where you'd not want this is if 
the pod hangs for some reason (sub process failure or similar) and then a kill 
on timeout fails.  Quite unlikely... but to consider.

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
[email protected]


With regards,
Apache Git Services

Reply via email to