Make sure all unique run_ids render a task duration bar (#37717)

apache · Feb 27, 2024 · cd31715 · cd31715
1 parent fc80a24
commit cd31715
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/airflow/www/static/js/dag/details/task/TaskDuration.tsx b/airflow/www/static/js/dag/details/task/TaskDuration.tsx
@@ -178,7 +178,7 @@ const TaskDuration = () => {
     ],
     // @ts-ignore
     dataset: {
-      dimensions: [orderingLabel, "queuedDurationUnit", "runDurationUnit"],
+      dimensions: ["runId", "queuedDurationUnit", "runDurationUnit"],
       source: durations,
     },
     tooltip: {
@@ -192,9 +192,12 @@ const TaskDuration = () => {
       type: "category",
       show: true,
       axisLabel: {
-        formatter: (value: string) =>
+        formatter: (runId: string) => {
+          const dagRun = dagRuns.find((dr) => dr.runId === runId);
+          if (!dagRun || !dagRun[orderingLabel]) return runId;
           // @ts-ignore
-          value ? moment(value).format(defaultFormat) : "",
+          return moment(dagRun[orderingLabel]).format(defaultFormat);
+        },
       },
       name: startCase(orderingLabel),
       nameLocation: "end",