springmvc消息转换器

作用域:

read controller进入之前

writeInternal contoller返回结果之前

WebMvcAutoJsonConfiguration:

package com.zy.platform.admin.server.configure;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.ser.std.ToStringSerializer;
import com.fasterxml.jackson.datatype.jsr310.JavaTimeModule;
import org.springframework.boot.autoconfigure.jackson.Jackson2ObjectMapperBuilderCustomizer;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.converter.json.Jackson2ObjectMapperBuilder;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;

import java.util.List;


@Configuration
public class WebMvcAutoJsonConfiguration implements WebMvcConfigurer {

    @Bean
    public WebMvcMessageConverter webMvcMessageConverter() {
        return new WebMvcMessageConverter();
    }

    /**
     * 配置消息转换器
     */
    @Override
    public void configureMessageConverters(List<HttpMessageConverter<?>> converters) {
        converters.add(webMvcMessageConverter());
    }

}

WebMvcMessageConverter:

package com.zy.platform.admin.server.configure;

import com.alibaba.fastjson.JSON;
import com.fasterxml.jackson.databind.JavaType;
import com.zy.platform.admin.server.entity.SysUser;
import com.zy.platform.common.core.utils.Sm4Utils;
import com.zy.platform.common.core.utils.StringUtils;
import com.zy.platform.common.core.web.ServerResponse;
import lombok.RequiredArgsConstructor;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.reflect.TypeUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.HttpInputMessage;
import org.springframework.http.HttpOutputMessage;
import org.springframework.http.converter.HttpMessageNotReadableException;
import org.springframework.http.converter.HttpMessageNotWritableException;
import org.springframework.http.converter.json.MappingJackson2HttpMessageConverter;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Type;
import java.util.Map;

@RequiredArgsConstructor
public class WebMvcMessageConverter extends MappingJackson2HttpMessageConverter {

    Logger logger = LoggerFactory.getLogger(WebMvcMessageConverter.class);


    @Override
    public Object read(Type type, Class<?> contextClass, HttpInputMessage inputMessage) throws IOException, HttpMessageNotReadableException {

        JavaType javaType = this.getJavaType(type, contextClass);

        InputStream is = inputMessage.getBody();

        String inputMessageBody = IOUtils.toString(is, "utf-8");

        if (TypeUtils.isAssignable(type, SysUser.class)) {
            SysUser sysUser = JSON.parseObject(inputMessageBody,SysUser.class);
            sysUser.setPassword(Sm4Utils.decode(sysUser.getPassword()));
            inputMessageBody = JSON.toJSONString(sysUser);
        }

        return this.objectMapper.readValue(new ByteArrayInputStream(inputMessageBody.getBytes()), javaType);

    }

    /**
     * Spring在返回JSON数据时通过此方法将Object转为JSON字符串并写出到Response中，针对于@ResponseBody注解生效
     */
    @Override
    protected void writeInternal(Object object, Type type, HttpOutputMessage outputMessage) throws IOException, HttpMessageNotWritableException {
        //ObjectMapper objectMapper = new ObjectMapper();
        // 用spring带的工具类，比较返给前端的数据类型是否为ResultMessage
        if (TypeUtils.isAssignable(type, ServerResponse.class)) {

            ServerResponse serverResponse = (ServerResponse) object;
            Object data = serverResponse.getData();
            if (data instanceof SysUser) {
                SysUser sysUser = (SysUser) data;
                sysUser.setPassword(StringUtils.isEmpty(sysUser.getPassword())?"":Sm4Utils.encode(sysUser.getPassword()));
                object = sysUser;
            }

        }
        super.writeInternal(object, type, outputMessage);
    }

}