Feature: Added job namespace to context

yin-zt · Nov 16, 2023 · c7f5c0a · c7f5c0a
1 parent 9b120b5
commit c7f5c0a
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/input/job.go b/input/job.go
@@ -54,6 +54,10 @@ func (ji *Job) ToJob() *tork.Job {
 	j.CreatedAt = n
 	j.Context = tork.JobContext{}
 	j.Context.Inputs = ji.Inputs
+	j.Context.Job = map[string]string{
+		"id":   j.ID,
+		"name": j.Name,
+	}
 	j.TaskCount = len(tasks)
 	j.Output = ji.Output
 	if ji.Defaults != nil {

diff --git a/job.go b/job.go
@@ -58,6 +58,7 @@ type JobSummary struct {
 }
 
 type JobContext struct {
+	Job    map[string]string `json:"job,omitempty"`
 	Inputs map[string]string `json:"inputs,omitempty"`
 	Tasks  map[string]string `json:"tasks,omitempty"`
 }
@@ -107,13 +108,15 @@ func (c JobContext) Clone() JobContext {
 	return JobContext{
 		Inputs: maps.Clone(c.Inputs),
 		Tasks:  maps.Clone(c.Tasks),
+		Job:    maps.Clone(c.Job),
 	}
 }
 
 func (c JobContext) AsMap() map[string]any {
 	return map[string]any{
 		"inputs": c.Inputs,
 		"tasks":  c.Tasks,
+		"job":    c.Job,
 	}
 }
 

diff --git a/job_test.go b/job_test.go
@@ -13,6 +13,10 @@ func TestClone(t *testing.T) {
 			Inputs: map[string]string{
 				"INPUT1": "VAL1",
 			},
+			Job: map[string]string{
+				"id":   "some-id",
+				"name": "my job",
+			},
 		},
 		Tasks: []*tork.Task{
 			{
@@ -33,6 +37,7 @@ func TestClone(t *testing.T) {
 	j2 := j1.Clone()
 
 	assert.Equal(t, j1.Context.Inputs, j2.Context.Inputs)
+	assert.Equal(t, j1.Context.Job, j2.Context.Job)
 	assert.Equal(t, j1.Tasks[0].Env, j2.Tasks[0].Env)
 	assert.Equal(t, j1.Execution[0].Env, j2.Execution[0].Env)