SecurityDataFetcherExceptionResolver.setAuthenticationTrustResolver

Author: rwinch

samples/webflux-security/src/main/java/io/spring/sample/graphql/SecurityDataFetcherExceptionResolver.java

@@ -6,6 +6,7 @@
 import graphql.GraphQLError;
 import graphql.GraphqlErrorBuilder;
 import graphql.schema.DataFetchingEnvironment;
+import org.springframework.util.Assert;
 import reactor.core.publisher.Mono;
 
 import org.springframework.graphql.execution.DataFetcherExceptionResolver;
@@ -40,6 +41,11 @@ public Mono<List<GraphQLError>> resolveException(Throwable exception, DataFetchi
 		return Mono.empty();
 	}
 
+	public void setAuthenticationTrustResolver(AuthenticationTrustResolver authenticationTrustResolver) {
+		Assert.notNull(authenticationTrustResolver, "authenticationTrustResolver cannot be null");
+		this.authenticationTrustResolver = authenticationTrustResolver;
+	}
+
 	private Mono<List<GraphQLError>> unauthorized(DataFetchingEnvironment environment) {
 		return Mono.fromCallable(() -> Arrays.asList(
 				GraphqlErrorBuilder.newError(environment)