Local GraphQLContext doesn't propagate values when spring actuator is active

[DoodlesOnMyFood]

On my controllers my code propagates values via local context, so something like this.

@SchemaMapping(typeName = "Foo", field = "bar")
public List<Bar> getBars(DataFetchingEnvironment env) {
var localContext = (GraphQLContext) Objects.requireNonNull(env.getLocalContext());
...
localContext.put("BAR_ID", service.getValue(bar.getId()));
...
}

@SchemaMapping(typeName = "Bar", field = "name")
public String getBarName(DataFetchingEnvironment env) {
var localContext = (GraphQLContext) Objects.requireNonNull(env.getLocalContext());
...
var barId = (String) localContext.get("BAR_ID");
...
}

Which worked fine for my use-case.

But I noticed that my code breaks when I activate spring-actuator.

To elaborate, the local context would no longer have my values, and only contain 'micrometer.observation' values.

var barId = (String) localContext.get("BAR_ID"); // null

I read the documentations regarding contexts and observability but failed to find a fix.

I was wondering if this was intentional and I need some additional configuration to make my code work.

I haven't tested with global context values.

[bclozel]

Which Spring GraphQL version are you using? This looks like a duplicate of #761

[DoodlesOnMyFood]

I'm on version 1.3.3

[bclozel]

Something looks wrong with your code snippet. It's assuming that a local context exists and adds to it, whereas I believe that a new local context is supposed to be returned with the data fetcher result for the next data fetchers to consider.

I have created a sample application and I'm not seeing any failure.

Here's a RuntimeWiringConfigurer that contributes a data fecher with a new localcontext:

package org.example.graphqlcontext;

import graphql.GraphQLContext;
import graphql.execution.DataFetcherResult;
import graphql.schema.DataFetcher;
import graphql.schema.DataFetchingEnvironment;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.graphql.execution.RuntimeWiringConfigurer;

@Configuration
public class GraphQlConfiguration {

	@Bean
	public RuntimeWiringConfigurer runtimeWiringConfigurer() {
		return runtimeWiring -> {
			runtimeWiring.type("Query", typeBuilder ->
					typeBuilder.dataFetcher("bookById", environment -> {
						GraphQLContext localContext = GraphQLContext.newContext()
								.put("project", "spring-graphql")
								.build();
						return DataFetcherResult.newResult()
								.data(new Book("test book", 42L))
								.localContext(localContext)
								.build();
					}));
		};
	}
}

As you can see, the schema mapping handler expects and finds a value in the local context:

package org.example.graphqlcontext;

import graphql.GraphQLContext;
import graphql.schema.DataFetchingEnvironment;

import org.springframework.graphql.data.method.annotation.SchemaMapping;
import org.springframework.stereotype.Controller;
import org.springframework.util.Assert;

@Controller
public class TestController {

	@SchemaMapping
	public Author author(Book book, DataFetchingEnvironment environment) {
		GraphQLContext localContext = (GraphQLContext) environment.getLocalContext();
		String project = (String) localContext.get("project");
		Assert.isTrue(project == "spring-graphql", "missing value from local context");
		return new Author(book.authorId(), "Test author");
	}

}

This is for the following schema:

type Query {
    bookById(id: ID!): Book
}

type Book {
    title: String
    author: Author
}

type Author {
    name: String
}

I have tested this with and without actuator on the classpath.

Please share a minimal sample application that reproduces the problem. Maybe your assumptions about the GraphQL local context are wrong in the first place?

[DoodlesOnMyFood]

In my setup, I create and inject a context in a filter. Maybe this is an anti-pattern?

Anyways, I'll try to reproduce my issue on a smaller demo during the weekend. Thanks for the feedback.