add timeout monitord

workflow_builder/argo_builder.go

@@ -20,6 +20,7 @@ import (
 type ArgoBuilder struct {
 	graph    graph.Graph
 	Workflow Workflow
+	Timeout  int
 }
 
 type Workflow struct {
@@ -36,13 +37,16 @@ type Spec struct {
 	Arguments  []Parameter           `yaml:"arguments,omitempty"`
 	Volumes    []VolumeClaimTemplate `yaml:"volumeClaimTemplates,omitempty"`
 	Templates  []Template            `yaml:"templates"`
+	Timeout    int                   `yaml:"activeDeadlineSeconds,omitempty"`
 }
 
 func (b *ArgoBuilder) CreateDAG() (string, error) {
-
 	b.createTemplates()
 	b.createDAGstep()
 	b.createVolumes()
+	if b.Timeout > 0 {
+		b.Workflow.Spec.Timeout = b.Timeout
+	}
 	b.Workflow.Spec.Entrypoint = "dag"
 	b.Workflow.ApiVersion = "argoproj.io/v1alpha1"
 	b.Workflow.Kind = "Workflow"