fix: add additional data option to dag builder config

Author: f7f376a1fcd0d0e11a10ed1b6577c9

dag/dag.go

@@ -85,7 +85,7 @@ func CreateDagWithConfig(path string, config *DagBuilderConfig) (*Dag, error) {
 		config = DefaultConfig()
 	}
 
-	var additionalData map[string]string = nil
+	var additionalData map[string]string = config.AdditionalData
 	if config.TimestampRoot {
 		currentTime := time.Now().UTC()
 		timeString := currentTime.Format(time.RFC3339)

dag/types.go

@@ -122,16 +122,18 @@ type DagBuilderConfig struct {
 	// 0 = use runtime.NumCPU() (auto-detect based on available cores)
 	// -1 = unlimited workers (not recommended, may overwhelm system)
 	// >0 = use exactly this many workers
-	MaxWorkers    int  // Parallel only, 0=auto-detect
-	TimestampRoot bool // Add timestamp to root
-	Processor     LeafProcessor
+	MaxWorkers     int  // Parallel only, 0=auto-detect
+	TimestampRoot  bool // Add timestamp to root
+	AdditionalData map[string]string
+	Processor      LeafProcessor
 }
 
 func DefaultConfig() *DagBuilderConfig {
 	return &DagBuilderConfig{
 		EnableParallel: false,
 		MaxWorkers:     0,
 		TimestampRoot:  false,
+		AdditionalData: map[string]string{},
 		Processor:      nil,
 	}
 }
@@ -141,6 +143,7 @@ func ParallelConfig() *DagBuilderConfig {
 		EnableParallel: true,
 		MaxWorkers:     0,
 		TimestampRoot:  false,
+		AdditionalData: map[string]string{},
 		Processor:      nil,
 	}
 }
@@ -150,6 +153,7 @@ func ParallelConfigWithWorkers(workers int) *DagBuilderConfig {
 		EnableParallel: true,
 		MaxWorkers:     workers,
 		TimestampRoot:  false,
+		AdditionalData: map[string]string{},
 		Processor:      nil,
 	}
 }