AbstractJackson2HttpMessageConverter

Author: nkonev

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

@@ -29,10 +29,13 @@
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.Part;
 
-import com.fasterxml.jackson.databind.ObjectMapper;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 import org.springframework.graphql.server.support.MultipartVariableMapper;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.converter.json.AbstractJackson2HttpMessageConverter;
+import org.springframework.http.converter.json.MappingJackson2HttpMessageConverter;
+import org.springframework.http.converter.json.MappingJacksonInputMessage;
 import org.springframework.web.multipart.MultipartFile;
 import org.springframework.web.multipart.MultipartHttpServletRequest;
 import reactor.core.publisher.Mono;
@@ -76,7 +79,7 @@ public class GraphQlHttpHandler {
 
 	private final WebGraphQlHandler graphQlHandler;
 
-    private final PartReader partReader;
+    private final AbstractJackson2HttpMessageConverter messageConverter;
 
 	/**
 	 * Create a new instance.
@@ -85,14 +88,14 @@ public class GraphQlHttpHandler {
 	public GraphQlHttpHandler(WebGraphQlHandler graphQlHandler) {
 		Assert.notNull(graphQlHandler, "WebGraphQlHandler is required");
 		this.graphQlHandler = graphQlHandler;
-        this.partReader = new JacksonPartReader(new ObjectMapper());
+        this.messageConverter = new MappingJackson2HttpMessageConverter();
 	}
 
-    public GraphQlHttpHandler(WebGraphQlHandler graphQlHandler, PartReader partReader) {
+    public GraphQlHttpHandler(WebGraphQlHandler graphQlHandler, AbstractJackson2HttpMessageConverter messageConverter) {
         Assert.notNull(graphQlHandler, "WebGraphQlHandler is required");
-        Assert.notNull(partReader, "PartConverter is required");
+        Assert.notNull(messageConverter, "PartConverter is required");
         this.graphQlHandler = graphQlHandler;
-        this.partReader = partReader;
+        this.messageConverter = messageConverter;
     }
 
 	/**
@@ -138,7 +141,7 @@ public ServerResponse handleMultipartRequest(ServerRequest serverRequest) throws
 		final Map<String, Object> queryVariables = getFromMapOrEmpty(inputQuery, "variables");
 		final Map<String, Object> extensions = getFromMapOrEmpty(inputQuery, "extensions");
 
-        Map<String, MultipartFile> fileParams = readMultipartFiles(httpServletRequest);
+        Map<String, MultipartFile> fileParams = getMultipartFiles(httpServletRequest);
 
         Map<String, List<String>> fileMappings = Optional.ofNullable(this.<Map<String, List<String>>>deserializePart(
                 httpServletRequest,
@@ -195,7 +198,9 @@ private <T> T deserializePart(HttpServletRequest httpServletRequest, String name
                 return null;
             }
             try(InputStream inputStream = part.getInputStream()) {
-                return partReader.readPart(inputStream, type);
+                return (T) messageConverter.read(
+                        type, null, new MappingJacksonInputMessage(inputStream, new HttpHeaders())
+                );
             } catch (IOException e) {
                 throw new RuntimeException(e);
             }
@@ -212,7 +217,7 @@ private Map<String, Object> getFromMapOrEmpty(Map<String, Object> input, String
 		}
 	}
 
-    private static Map<String, MultipartFile> readMultipartFiles(HttpServletRequest httpServletRequest) {
+    private Map<String, MultipartFile> getMultipartFiles(HttpServletRequest httpServletRequest) {
         Assert.isInstanceOf(MultipartHttpServletRequest.class, httpServletRequest,
                 "Request should be of type MultipartHttpServletRequest");
         MultipartHttpServletRequest multipartHttpServletRequest = (MultipartHttpServletRequest) httpServletRequest;