Avoid unnecessary async handling in GraphQlHttpHandler

In the webmvc GraphQlHttpHandler, return the server response directly if the
WebGraphQlHandler returns an already completed Mono. This avoids some extra
overhead incurred with async processing.

Author: kilink

spring-graphql/src/main/java/org/springframework/graphql/server/webmvc/GraphQlHttpHandler.java

@@ -17,6 +17,7 @@
 package org.springframework.graphql.server.webmvc;
 
 import java.io.IOException;
+import java.time.Duration;
 import java.util.Arrays;
 import java.util.List;
 import java.util.Map;
@@ -106,7 +107,12 @@ public ServerResponse handleRequest(ServerRequest serverRequest) throws ServletE
 					builder.headers(headers -> headers.putAll(response.getResponseHeaders()));
 					builder.contentType(selectResponseMediaType(serverRequest));
 					return builder.body(response.toMap());
-				});
+				})
+				.cache();
+
+		try {
+			return responseMono.block(Duration.ZERO);
+		} catch (IllegalStateException ignored) {}
 
 		return ServerResponse.async(responseMono);
 	}
@@ -138,5 +144,4 @@ private static MediaType selectResponseMediaType(ServerRequest serverRequest) {
 		}
 		return MediaType.APPLICATION_JSON;
 	}
-
 }

spring-graphql/src/test/java/org/springframework/graphql/server/webmvc/GraphQlHttpHandlerTests.java

@@ -122,7 +122,10 @@ private MockHttpServletResponse handleRequest(
 			MockHttpServletRequest servletRequest, GraphQlHttpHandler handler) throws ServletException, IOException {
 
 		ServerRequest request = ServerRequest.create(servletRequest, MESSAGE_READERS);
-		ServerResponse response = ((AsyncServerResponse) handler.handleRequest(request)).block();
+		ServerResponse response = handler.handleRequest(request);
+		if (response instanceof AsyncServerResponse asyncServerResponse) {
+			asyncServerResponse.block();
+		}
 
 		MockHttpServletResponse servletResponse = new MockHttpServletResponse();
 		response.writeTo(servletRequest, servletResponse, new DefaultContext());