getContext alternative now that it's deprecated

[sapinheiro]

I'm new to graphql and have been using DataFetchingEnvironment#getContext in my project. For example:

class MyResolver implements GraphQLQueryResolver {
    public String helloWorld(DataFetchingEnvironment env) {
        // this works!!
        CustomGraphQLContext context = env.getContext();
        return "Hello world";
    } 
}

I've built the context elsewhere using the GraphQLServletContextBuilder which seems to work out of the box for setting the context.

The problem is that getContext is deprecated. It says to alternatively use getGraphQLContext, but it's currently null for me. How do I get it populated with the proper value? If I need to override the ExecutionInput, where do I do that? There doesn't seem to be any clear documentation on this and everything I've seen uses getContext