feat: add job_state as label

goto · Oct 8, 2024 · 0e38e29 · 0e38e29
1 parent 038ef6e
commit 0e38e29
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/modules/dagger/driver.go b/modules/dagger/driver.go
@@ -40,6 +40,7 @@ const (
 	labelURN          = "urn"
 	labelName         = "name"
 	labelNamespace    = "namespace"
+	labelJobState     = "job_state"
 
 	orchestratorLabelValue = "entropy"
 )

diff --git a/modules/dagger/driver_plan.go b/modules/dagger/driver_plan.go
@@ -53,6 +53,7 @@ func (dd *daggerDriver) planCreate(exr module.ExpandedResource, act module.Actio
 	conf.JarURI = dd.conf.JarURI
 	conf.State = StateDeployed
 	conf.JobState = JobStateRunning
+	exr.Resource.Labels[labelJobState] = conf.JobState
 
 	exr.Resource.Spec.Configs = modules.MustJSON(conf)
 
@@ -128,6 +129,11 @@ func (dd *daggerDriver) planChange(exr module.ExpandedResource, act module.Actio
 
 	exr.Resource.Spec.Configs = modules.MustJSON(curConf)
 
+	if act.Labels != nil {
+		exr.Resource.Labels = act.Labels
+	}
+	exr.Resource.Labels[labelJobState] = curConf.JobState
+
 	err = dd.validateHelmReleaseConfigs(exr, *curConf)
 	if err != nil {
 		return nil, err