fix: Add instance ID to workflowtaskresults (#8150)

Signed-off-by: Alex Collins <alex_collins@intuit.com>
argoproj · Mar 18, 2022 · 442096b · 442096b
1 parent 2b87f86
commit 442096b
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 19 deletions.
diff --git a/workflow/common/common.go b/workflow/common/common.go
@@ -106,6 +106,8 @@ const (
 
 	// EnvVarPodName contains the name of the pod (currently unused)
 	EnvVarPodName = "ARGO_POD_NAME"
+	// EnvVarInstanceID is the instance ID
+	EnvVarInstanceID = "ARGO_INSTANCE_ID"
 	// EnvVarWorkflowName is the name of the workflow for which the an agent is responsible for
 	EnvVarWorkflowName = "ARGO_WORKFLOW_NAME"
 	// EnvVarWorkflowUID is the workflow's UID

diff --git a/workflow/controller/workflowpod.go b/workflow/controller/workflowpod.go
@@ -646,6 +646,11 @@ func (woc *wfOperationCtx) createEnvVars() []apiv1.EnvVar {
 			Value: string(woc.wf.UID),
 		},
 	}
+	if v := woc.controller.Config.InstanceID; v != "" {
+		execEnvVars = append(execEnvVars,
+			apiv1.EnvVar{Name: common.EnvVarInstanceID, Value: v},
+		)
+	}
 	if woc.controller.Config.Executor != nil {
 		execEnvVars = append(execEnvVars, woc.controller.Config.Executor.Env...)
 	}

diff --git a/workflow/executor/taskresult.go b/workflow/executor/taskresult.go
@@ -3,6 +3,7 @@ package executor
 import (
 	"context"
 	"encoding/json"
+	"os"
 
 	apierr "k8s.io/apimachinery/pkg/api/errors"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@@ -36,27 +37,32 @@ func (we *WorkflowExecutor) patchTaskResult(ctx context.Context, result wfv1.Nod
 }
 
 func (we *WorkflowExecutor) createTaskResult(ctx context.Context, result wfv1.NodeResult) error {
-	_, err := we.taskResultClient.Create(ctx,
-		&wfv1.WorkflowTaskResult{
-			TypeMeta: metav1.TypeMeta{
+	taskResult := &wfv1.WorkflowTaskResult{
+		TypeMeta: metav1.TypeMeta{
+			APIVersion: workflow.APIVersion,
+			Kind:       workflow.WorkflowTaskResultKind,
+		},
+		ObjectMeta: metav1.ObjectMeta{
+			Name:   we.nodeId,
+			Labels: map[string]string{common.LabelKeyWorkflow: we.workflow},
+		},
+		NodeResult: result,
+	}
+	taskResult.SetOwnerReferences(
+		[]metav1.OwnerReference{
+			{
 				APIVersion: workflow.APIVersion,
-				Kind:       workflow.WorkflowTaskResultKind,
-			},
-			ObjectMeta: metav1.ObjectMeta{
-				Name:   we.nodeId,
-				Labels: map[string]string{common.LabelKeyWorkflow: we.workflow},
-				// make sure deleting the workflow, delete this result
-				OwnerReferences: []metav1.OwnerReference{
-					{
-						APIVersion: workflow.APIVersion,
-						Kind:       workflow.WorkflowKind,
-						Name:       we.workflow,
-						UID:        we.workflowUID,
-					},
-				},
+				Kind:       workflow.WorkflowKind,
+				Name:       we.workflow,
+				UID:        we.workflowUID,
 			},
-			NodeResult: result,
-		},
+		})
+
+	if v := os.Getenv(common.EnvVarInstanceID); v != "" {
+		taskResult.Labels[common.LabelKeyControllerInstanceID] = v
+	}
+	_, err := we.taskResultClient.Create(ctx,
+		taskResult,
 		metav1.CreateOptions{},
 	)
 	return err