feat: add job labels to support job attribution (#115) (#126)

goto · Aug 24, 2023 · d08e19e · d08e19e
1 parent 2cac6c9
commit d08e19e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/core/scheduler/service/executor_input_compiler.go b/core/scheduler/service/executor_input_compiler.go
@@ -2,6 +2,7 @@ package service
 
 import (
 	"context"
+	"fmt"
 	"strings"
 	"time"
 
@@ -36,6 +37,8 @@ const (
 	configDend          = "DEND"
 	configExecutionTime = "EXECUTION_TIME"
 	configDestination   = "JOB_DESTINATION"
+
+	JobAttributionLabelsKey = "JOB_LABELS"
 )
 
 type TenantService interface {
@@ -92,6 +95,13 @@ func (i InputCompiler) Compile(ctx context.Context, job *scheduler.Job, config s
 		return nil, err
 	}
 
+	jobAttributionLabels := fmt.Sprintf("project=%s,namespace=%s,job=%s", job.Tenant.ProjectName(), job.Tenant.NamespaceName(), job.Name)
+	if jobLables, ok := confs[JobAttributionLabelsKey]; ok {
+		confs[JobAttributionLabelsKey] = jobLables + "," + jobAttributionLabels
+	} else {
+		confs[JobAttributionLabelsKey] = jobAttributionLabels
+	}
+
 	if config.Executor.Type == scheduler.ExecutorTask {
 		return &scheduler.ExecutorInput{
 			Configs: utils.MergeMaps(confs, systemDefinedVars),

diff --git a/core/scheduler/service/executor_input_compiler_test.go b/core/scheduler/service/executor_input_compiler_test.go
@@ -228,6 +228,7 @@ func TestExecutorCompiler(t *testing.T) {
 						"EXECUTION_TIME":       executedAt.Format(time.RFC3339),
 						"JOB_DESTINATION":      job.Destination,
 						"some.config.compiled": "val.compiled",
+						"JOB_LABELS":           "project=proj1,namespace=ns1,job=job1",
 					},
 					Secrets: map[string]string{"secret.config.compiled": "a.secret.val.compiled"},
 					Files:   compiledFile,