From 411effb0005707204fc491793608f5ff467de113 Mon Sep 17 00:00:00 2001
From: pb <pierre.bayle@irt-stexupery.com>
Date: Wed, 28 May 2025 10:53:15 +0200
Subject: [PATCH] Added the label to YAML to excute the pods on another node

---
 workflow_builder/argo_builder.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/workflow_builder/argo_builder.go b/workflow_builder/argo_builder.go
index 03b9d38..e062497 100644
--- a/workflow_builder/argo_builder.go
+++ b/workflow_builder/argo_builder.go
@@ -59,6 +59,9 @@ type Spec struct {
 	Volumes    			[]VolumeClaimTemplate 	`yaml:"volumeClaimTemplates,omitempty"`
 	Templates  			[]Template            	`yaml:"templates"`
 	Timeout    			int                   	`yaml:"activeDeadlineSeconds,omitempty"`
+	NodeSelector		struct{
+							NodeRole string `yaml:"node-role"`
+						} `yaml:"nodeSelector"`
 }
 
 // TODO: found on a processing instance linked to storage
@@ -73,6 +76,7 @@ func (b *ArgoBuilder) CreateDAG(namespace string, write bool) ( int, []string, [
 	if b.Timeout > 0 {
 		b.Workflow.Spec.Timeout = b.Timeout
 	}
+	b.Workflow.Spec.NodeSelector.NodeRole = "worker"
 	b.Workflow.Spec.ServiceAccountName = "sa-"+namespace
 	b.Workflow.Spec.Entrypoint = "dag"
 	b.Workflow.ApiVersion = "argoproj.io/v1alpha1"