diff --git a/dolphinscheduler-task-plugin/dolphinscheduler-task-api/src/main/java/org/apache/dolphinscheduler/plugin/task/api/k8s/impl/K8sTaskExecutor.java b/dolphinscheduler-task-plugin/dolphinscheduler-task-api/src/main/java/org/apache/dolphinscheduler/plugin/task/api/k8s/impl/K8sTaskExecutor.java index 1ce6b12c22..167cae8668 100644 --- a/dolphinscheduler-task-plugin/dolphinscheduler-task-api/src/main/java/org/apache/dolphinscheduler/plugin/task/api/k8s/impl/K8sTaskExecutor.java +++ b/dolphinscheduler-task-plugin/dolphinscheduler-task-api/src/main/java/org/apache/dolphinscheduler/plugin/task/api/k8s/impl/K8sTaskExecutor.java @@ -112,11 +112,17 @@ public class K8sTaskExecutor extends AbstractK8sTaskExecutor { Map limitRes = new HashMap<>(); limitRes.put(MEMORY, new Quantity(String.format("%s%s", limitPodMem, MI))); limitRes.put(CPU, new Quantity(String.valueOf(limitPodCpu))); + Map labelMap = k8STaskMainParameters.getLabelMap(); - labelMap.put(LAYER_LABEL, LAYER_LABEL_VALUE); - labelMap.put(NAME_LABEL, k8sJobName); + Map jobLabelMap = new HashMap<>(); + jobLabelMap.put(LAYER_LABEL, LAYER_LABEL_VALUE); + jobLabelMap.put(NAME_LABEL, k8sJobName); + jobLabelMap.putAll(labelMap); + Map podLabelMap = new HashMap<>(); podLabelMap.put(UNIQUE_LABEL_NAME, taskRequest.getTaskAppId()); + podLabelMap.putAll(labelMap); + EnvVar taskInstanceIdVar = new EnvVar(TASK_INSTANCE_ID, taskInstanceId, null); List envVars = new ArrayList<>(); envVars.add(taskInstanceIdVar); @@ -161,7 +167,7 @@ public class K8sTaskExecutor extends AbstractK8sTaskExecutor { .withApiVersion(API_VERSION) .withNewMetadata() .withName(k8sJobName) - .withLabels(labelMap) + .withLabels(jobLabelMap) .withNamespace(namespaceName) .endMetadata() .withNewSpec()