Pass CustomInfo metadata through ArrayNode ExternalResourceInfo (#591) (

#6116) * Pass Actor metadata through ArrayNode ExternalResourceInfo (#591) * pass CustomInfo through ArrayNode event recorder Signed-off-by: Daniel Rammer <[email protected]> * actually doing it Signed-off-by: Daniel Rammer <[email protected]> * correctly passing custom Signed-off-by: Daniel Rammer <[email protected]> --------- Signed-off-by: Daniel Rammer <[email protected]> * events -> event Signed-off-by: Daniel Rammer <[email protected]> * Run make -C flytepropelelr lint-fix Signed-off-by: Eduardo Apolinario <[email protected]> --------- Signed-off-by: Daniel Rammer <[email protected]> Signed-off-by: Eduardo Apolinario <[email protected]> Co-authored-by: Eduardo Apolinario <[email protected]>
flyteorg · Dec 19, 2024 · b0062e4 · b0062e4
1 parent b981363
commit b0062e4
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/flytepropeller/pkg/controller/nodes/array/event_recorder.go b/flytepropeller/pkg/controller/nodes/array/event_recorder.go
@@ -136,14 +136,21 @@ func (e *externalResourcesEventRecorder) process(ctx context.Context, nCtx inter
 			log.Name = fmt.Sprintf("%s-%d", log.GetName(), index)
 		}
 
-		e.externalResources = append(e.externalResources, &event.ExternalResourceInfo{
+		externalResourceInfo := event.ExternalResourceInfo{
 			ExternalId:   externalResourceID,
 			Index:        uint32(index), // #nosec G115
 			Logs:         taskExecutionEvent.GetLogs(),
 			RetryAttempt: retryAttempt,
 			Phase:        taskExecutionEvent.GetPhase(),
 			CacheStatus:  cacheStatus,
-		})
+			CustomInfo:   taskExecutionEvent.GetCustomInfo(),
+		}
+
+		if taskExecutionEvent.GetMetadata() != nil && len(taskExecutionEvent.GetMetadata().GetExternalResources()) == 1 {
+			externalResourceInfo.CustomInfo = taskExecutionEvent.GetMetadata().GetExternalResources()[0].GetCustomInfo()
+		}
+
+		e.externalResources = append(e.externalResources, &externalResourceInfo)
 	}
 
 	// clear nodeEvents and taskEvents