Nil pointer fix

Signed-off-by: Abhijit Mukherjee <[email protected]>
kanisterio · Mar 21, 2024 · 907854b · 907854b
1 parent e72b3eb
commit 907854b
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/pkg/function/kube_task.go b/pkg/function/kube_task.go
@@ -66,14 +66,15 @@ func kubeTask(ctx context.Context, cli kubernetes.Interface, namespace, image st
 		PodOverride:  podOverride,
 	}
 	// Mark labels to pods with prefix `jobPrefix`. Add the jobID as reference to the origin for the pod.
-	fields := field.FromContext(ctx)
-	for _, f := range fields.Fields() {
-		if strings.HasPrefix(f.Key(), consts.LabelPrefix) {
-			jobID := f.Value().(string)
-			if options.Labels == nil {
-				options.Labels = make(map[string]string)
+	if fields := field.FromContext(ctx); fields != nil {
+		for _, f := range fields.Fields() {
+			if strings.HasPrefix(f.Key(), consts.LabelPrefix) {
+				jobID := f.Value().(string)
+				if options.Labels == nil {
+					options.Labels = make(map[string]string)
+				}
+				options.Labels[consts.LabelPrefix+"JobID"] = jobID
 			}
-			options.Labels[consts.LabelPrefix+"JobID"] = jobID
 		}
 	}