SpringMVC中如何修改返回值類型后的消息轉(zhuǎn)換器處理方式

這篇文章給大家分享的是有關(guān)SpringMVC中如何修改返回值類型后的消息轉(zhuǎn)換器處理方式的內(nèi)容。小編覺得挺實(shí)用的，因此分享給大家做個(gè)參考，一起跟隨小編過來看看吧。

問題案例

假設(shè)有一個(gè)Controller方法如下

@RequestMapping(value = "test")
    @ResponseBody
    public Object test() {
        Map<String,String> param = new HashMap<>();
        param.put("name","userwyh");
        return param;
    }

然后我們通過實(shí)現(xiàn)ResponseBodyAdvice接口對返回值再輸出之前進(jìn)行了修改，此處我們把它變成了String類型，直接返回hello,world。

@ControllerAdvice
public class MyResponseBodyAdvice implements ResponseBodyAdvice<Object> {
    @Override
    public boolean supports(MethodParameter methodParameter, Class<? extends HttpMessageConverter<?>> aClass) {
        return true;
    }
    @Override
    public Object beforeBodyWrite(Object o, MethodParameter methodParameter, MediaType mediaType, Class<? extends HttpMessageConverter<?>> aClass, ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) {
        return "hello,world";
    }
}

這樣不管Controller中的test方法返回什么值，我們都會把它變成hello，world輸出。想想也沒有什么不對，仔細(xì)確認(rèn)了代碼，它就是這樣做的。

于是，啟動項(xiàng)目，打開瀏覽器，地址欄輸入localhost:8080/test，回車。

我們確實(shí)看到了hello,world字樣，沒有任何問題。

但是仔細(xì)一看的話，你會發(fā)現(xiàn)hello,world前后都多了一個(gè)引號。這顯然不是我們想要的返回值?。。?！

為什么？

這時(shí)候標(biāo)題提到的SpringMVC的消息轉(zhuǎn)換器HttpMessageConverter就該出場了。

HttpMessageConverter源碼剖析可以移步 SpringMVC源碼剖析-消息轉(zhuǎn)換器HttpMessageConverter 進(jìn)行查看。我們這里就不對源碼進(jìn)行詳細(xì)的解讀了。

首先SpringMVC會加載在spring-servlet.xml配置好的消息轉(zhuǎn)換器到messageConverters里。

protected final List<HttpMessageConverter<?>> messageConverters;

debug時(shí)發(fā)現(xiàn)SpringMVC不止加載了我們配置好的消息轉(zhuǎn)換器，它還加載了另外7個(gè)默認(rèn)的消息轉(zhuǎn)換器，即便7個(gè)之中你在配置文件中配置了，它依然會再次加載一次。如圖，0和1是配置的，2-8是默認(rèn)加載的。

上圖中的方法writeWithMessageConverters就是在Controller方法執(zhí)行之后就進(jìn)入的，在抽象類AbstractMessageConverterMethodProcessor的第164行處。這個(gè)方法也正是SpringMVC為當(dāng)前返回值選擇合適的消息轉(zhuǎn)換器，選擇的順序就是messageConverters的轉(zhuǎn)換器順序。

通過閱讀源碼，我們知道，此處對messageConverters進(jìn)行了遍歷，先判斷當(dāng)前的轉(zhuǎn)換器對當(dāng)前返回類型是否能寫canWrite，如果能得話就會調(diào)用beforeBodyWrite方法，然后把beforeBodyWrite的返回值通過write方法進(jìn)行輸出。如果不能的話就選擇下一個(gè)轉(zhuǎn)換器。如果最終沒有一個(gè)合適的，就會拋出一個(gè)異常。

了解問題原因及分析

有了上面對HttpMessageConverter的簡單描述，我們大概可以得到一個(gè)結(jié)論：

因?yàn)樵贑ontroller中的返回值類型是java.util.HashMap，所以在writeWithMessageConverters方法中SpringMVC選定的轉(zhuǎn)換器并不是StringHttpMessageConverter，而是MappingJackson2HttpMessageConverter。

我們可以通過在MyResponseBodyAdvice類beforeBodyWrite方法中打印參數(shù)得以證明確實(shí)當(dāng)前SpringMVC選擇的轉(zhuǎn)換器就是MappingJackson2HttpMessageConverter。

然后我們在beforeBodyWrite執(zhí)行返回了String類型的hello,world。而此時(shí)選定的轉(zhuǎn)換器已經(jīng)是MappingJackson2HttpMessageConverter了，所以通過該轉(zhuǎn)換器進(jìn)行轉(zhuǎn)換輸出。

我們再通過一個(gè)實(shí)例說明MappingJackson2HttpMessageConverter會把String前后新增雙引號。

通過上面的分析，相信大家已經(jīng)大概知道問題的來龍去脈了。

所以第一個(gè)反應(yīng)當(dāng)然就是重寫MappingJackson2HttpMessageConverter的某個(gè)方法咯。

不過在這之前，我們還需要對SpringMVC的源碼進(jìn)行進(jìn)一步分析。

上面提到它會把beforeBodyWrite的返回值通過write方法進(jìn)行輸出，所以我們需要了解這個(gè)write方法。它是一個(gè)接口，由具體的消息轉(zhuǎn)換器進(jìn)行實(shí)現(xiàn)。SpringMVC自己提供了一個(gè)抽象類AbstractGenericHttpMessageConverter進(jìn)行了實(shí)現(xiàn)，但把具體的write任務(wù)交給了抽象方法writeInternal。如下圖

接下來就是看MappingJackson2HttpMessageConverter的代碼了。該類的父類AbstractJackson2HttpMessageConverter確實(shí)繼承了AbstractGenericHttpMessageConverter并實(shí)現(xiàn)了writeInternal方法。

所以，方法很簡單，我們只需要把jackson的writeInternal重寫一下就可以了。

解決方法

1、創(chuàng)建一個(gè)MappingJackson2HttpMessageConverterFactory類

package com.userwyh.spring.controller;
import org.slf4j.Logger;
import org.springframework.http.HttpOutputMessage;
import org.springframework.http.MediaType;
import org.springframework.http.converter.HttpMessageNotWritableException;
import org.springframework.http.converter.json.MappingJackson2HttpMessageConverter;
import org.springframework.util.StreamUtils;
import java.io.IOException;
import java.lang.reflect.Type;
import java.nio.charset.Charset;
import static org.slf4j.LoggerFactory.getLogger;
/**
 * Created by userwyh on 2017/3/4.
 */
public class MappingJackson2HttpMessageConverterFactory {
    private static final Logger logger = getLogger(MappingJackson2HttpMessageConverterFactory.class);
    public MappingJackson2HttpMessageConverter init() {
        return new MappingJackson2HttpMessageConverter(){
            /**
             * 重寫Jackson消息轉(zhuǎn)換器的writeInternal方法
             * SpringMVC選定了具體的消息轉(zhuǎn)換類型后,會調(diào)用具體類型的write方法,將Java對象轉(zhuǎn)換后寫入返回內(nèi)容
             */
            @Override
            protected void writeInternal(Object object, Type type, HttpOutputMessage outputMessage) throws IOException, HttpMessageNotWritableException {
                if (object instanceof String){
                    logger.info("在MyResponseBodyAdvice進(jìn)行轉(zhuǎn)換時(shí)返回值變成String了,不能用原來選定消息轉(zhuǎn)換器進(jìn)行轉(zhuǎn)換,直接使用StringHttpMessageConverter轉(zhuǎn)換");
                    //StringHttpMessageConverter中就是用以下代碼寫的
                    Charset charset = this.getContentTypeCharset(outputMessage.getHeaders().getContentType());
                    StreamUtils.copy((String)object, charset, outputMessage.getBody());
                }else{
                    logger.info("返回值不是String類型,還是使用之前選擇的轉(zhuǎn)換器進(jìn)行消息轉(zhuǎn)換");
                    super.writeInternal(object, type, outputMessage);
                }
            }
            private Charset getContentTypeCharset(MediaType contentType) {
                return contentType != null && contentType.getCharset() != null?contentType.getCharset():this.getDefaultCharset();
            }
        };
    }
}

2、稍微修改一下spring的配置文件

<mvc:annotation-driven>
        <mvc:message-converters>
            <bean class="org.springframework.http.converter.ByteArrayHttpMessageConverter">
                <property name="supportedMediaTypes">
                    <list>
                        <value>image/jpeg</value>
                        <value>image/png</value>
                        <value>image/gif</value>
                    </list>
                </property>
            </bean>
            <bean  factory-bean="mappingJackson2HttpMessageConverterFactory" factory-method="init"
                   class="org.springframework.http.converter.json.MappingJackson2HttpMessageConverter" >
            </bean>
        </mvc:message-converters>
    </mvc:annotation-driven>
    <bean id="mappingJackson2HttpMessageConverterFactory" class = "com.userwyh.spring.controller.MappingJackson2HttpMessageConverterFactory" />

此時(shí)SpringMVC啟動時(shí)，messageConverters的順序就是ByteArrayHttpMessageConverter，mappingJackson2HttpMessageConverterFactory，然后另外7個(gè)默認(rèn)的，共9個(gè)。如第一張截圖所示即為配置后的效果。

3、啟動項(xiàng)目，打開瀏覽器，地址欄輸入localhost:8080/test，回車。雙引號沒有了，正是我們想要的結(jié)果。

再看一下日志：

三月 05, 2017 11:01:51 下午 com.userwyh.spring.controller.MappingJackson2HttpMessageConverterFactory$1 writeInternal 信息: 在MyResponseBodyAdvice進(jìn)行轉(zhuǎn)換時(shí)返回值變成String了,不能用原來選定消息轉(zhuǎn)換器進(jìn)行轉(zhuǎn)換,直接使用StringHttpMessageConverter轉(zhuǎn)換

結(jié)語

其實(shí)你可以直接在Controller中直接返回String類型的？

其實(shí)你可以針對在MyResponseBodyAdvice 中確認(rèn)要返回不同類型的，直接在Controller中判斷下就行了啊，比如以下這樣就可以了，為什么要這么麻煩呢？

@RequestMapping(value = "test")
    @ResponseBody
    public Object test() {
        Map<String,String> param = new HashMap<>();
        param.put("name","userwyh");
        if(condition){
            return "hello,world";
        }
        return param;
    }

可能，因?yàn)橄矚g折騰，既然可以在MyResponseBodyAdvice 進(jìn)行統(tǒng)一的返回值轉(zhuǎn)換，我就斷定可以找到方法解決這個(gè)問題的，也確實(shí)解決了。

感謝各位的閱讀！關(guān)于“SpringMVC中如何修改返回值類型后的消息轉(zhuǎn)換器處理方式”這篇文章就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，讓大家可以學(xué)到更多知識，如果覺得文章不錯(cuò)，可以把它分享出去讓更多的人看到吧！