Merge remote-tracking branch 'flyteplugins/expose-node-id' into monor…

…epo-move/flyteplugins/expose-node-id
flyteorg · Oct 3, 2023 · 6c24344 · 6c24344
2 parents d9586b0 + dff82ed
commit 6c24344
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/flyteplugins/go/tasks/pluginmachinery/flytek8s/k8s_resource_adds.go b/flyteplugins/go/tasks/pluginmachinery/flytek8s/k8s_resource_adds.go
@@ -31,6 +31,15 @@ func GetContextEnvVars(ownerCtx context.Context) []v1.EnvVar {
 			},
 		)
 	}
+
+	if nodeID := contextutils.Value(ownerCtx, contextutils.NodeIDKey); nodeID != "" {
+		envVars = append(envVars,
+			v1.EnvVar{
+				Name:  "FLYTE_INTERNAL_NODE_ID",
+				Value: nodeID,
+			},
+		)
+	}
 	return envVars
 }