Handle nil context fields, not just a nil context

Found while testing crossplane-contrib/function-patch-and-transform#17 Signed-off-by: Nic Cope <[email protected]>
crossplane · Oct 16, 2023 · ffe3c15 · ffe3c15
1 parent 26c3c27
commit ffe3c15
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/response/response.go b/response/response.go
@@ -47,7 +47,7 @@ func To(req *v1beta1.RunFunctionRequest, ttl time.Duration) *v1beta1.RunFunction
 
 // SetContextKey sets context to the supplied key.
 func SetContextKey(rsp *v1beta1.RunFunctionResponse, key string, v *structpb.Value) {
-	if rsp.Context == nil {
+	if rsp.GetContext().GetFields() == nil {
 		rsp.Context = &structpb.Struct{Fields: make(map[string]*structpb.Value)}
 	}
 	rsp.Context.Fields[key] = v