Merge pull request #50 from negz/contextual

Handle nil context fields, not just a nil context
crossplane · Oct 16, 2023 · 0a4b30c · 0a4b30c
2 parents 26c3c27 + ffe3c15
commit 0a4b30c
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