自定义返回数据转换器MessageConverter(请求和响应之间进行转换的策略接口)

通过修改请求头的参数来实现返回值重写

配置类

import com.example.studywork1.pojo.User;
import org.springframework.http.HttpInputMessage;
import org.springframework.http.HttpOutputMessage;
import org.springframework.http.MediaType;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.converter.HttpMessageNotReadableException;
import org.springframework.http.converter.HttpMessageNotWritableException;

import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.StandardCharsets;
import java.util.List;

public class MyMessageConverter implements HttpMessageConverter<User> {

    // 是否能读
    @Override
    public boolean canRead(Class<?> clazz, MediaType mediaType) {
        return false;
    }

    // 是否能写
    @Override
    public boolean canWrite(Class<?> clazz, MediaType mediaType) {
        // 如果是user类型返回true 可以写
        return clazz.isAssignableFrom(User.class);
    }

    // 自定义返回此转换器支持的媒体类型  自定义:application/xxxx
    @Override
    public List<MediaType> getSupportedMediaTypes() {
        return MediaType.parseMediaTypes("application/xxxx");
    }

    // 上面是否能读为false 这个方法不需要操作
    @Override
    public User read(Class<? extends User> clazz, HttpInputMessage inputMessage) throws IOException, HttpMessageNotReadableException {
        return null;
    }

    // 重新写的方法
    @Override
    public void write(User user, MediaType contentType, HttpOutputMessage outputMessage) throws IOException, HttpMessageNotWritableException {
        // 自定义协议数据的写出
        String data = "自定义写的结果------姓名:" + user.getName()+"年龄:"+user.getAge();
        OutputStream body = outputMessage.getBody();
        // 放入body中 输出出去
        body.write(data.getBytes(StandardCharsets.UTF_8));
    }
}
@Component
public class MyConfig {
    @Bean
    public WebMvcConfigurer webMvcConfigurer(){
        return new WebMvcConfigurer() {

            @Override
            public void extendMessageConverters(List<HttpMessageConverter<?>> converters) {
                converters.add(new MyMessageConverter());
            }
        };
    }
}
@Controller
public class MyController {
    @ResponseBody
    @GetMapping("/getUser")
    public User getUser(){
        User user = new User();
        user.setAge(20);
        user.setName("张三");
        return user;
    }
}

输出结果

 通过URL传值来实现返回值重写(自定义内容协商策略)

开启请求参数内容协商模式

spring.mvc.contentnegotiation.favor-parameter=true

引入jar

 <dependency>
      <groupId>com.fasterxml.jackson.dataformat</groupId>
      <artifactId>jackson-dataformat-xml</artifactId>
</dependency>

就可以传入请求参数 获取内容

 最初始format参数集合中只有json和xml两种类型

开始自定义

配置类

import com.example.studywork1.pojo.User;
import org.springframework.http.HttpInputMessage;
import org.springframework.http.HttpOutputMessage;
import org.springframework.http.MediaType;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.converter.HttpMessageNotReadableException;
import org.springframework.http.converter.HttpMessageNotWritableException;

import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.StandardCharsets;
import java.util.List;

public class MyMessageConverter implements HttpMessageConverter<User> {

    // 是否能读
    @Override
    public boolean canRead(Class<?> clazz, MediaType mediaType) {
        return false;
    }

    // 是否能写
    @Override
    public boolean canWrite(Class<?> clazz, MediaType mediaType) {
        // 如果是user类型返回true 可以写
        return clazz.isAssignableFrom(User.class);
    }

    // 自定义返回此转换器支持的媒体类型  自定义:application/xxxx
    @Override
    public List<MediaType> getSupportedMediaTypes() {
        return MediaType.parseMediaTypes("application/xxxx");
    }

    // 上面是否能读为false 这个方法不需要操作
    @Override
    public User read(Class<? extends User> clazz, HttpInputMessage inputMessage) throws IOException, HttpMessageNotReadableException {
        return null;
    }

    // 重新写的方法
    @Override
    public void write(User user, MediaType contentType, HttpOutputMessage outputMessage) throws IOException, HttpMessageNotWritableException {
        // 自定义协议数据的写出
        String data = "自定义写的结果------姓名:" + user.getName()+"年龄:"+user.getAge();
        OutputStream body = outputMessage.getBody();
        // 放入body中 输出出去
        body.write(data.getBytes(StandardCharsets.UTF_8));
    }
}
@Component
public class MyConfig {
    @Bean
    public WebMvcConfigurer webMvcConfigurer(){
        return new WebMvcConfigurer() {
            @Override
            public void configureContentNegotiation(ContentNegotiationConfigurer configurer) {
                Map<String, MediaType> mediaTypes = new HashMap<>();
                mediaTypes.put("json",MediaType.APPLICATION_JSON);
                mediaTypes.put("xml",MediaType.APPLICATION_XML);
// 自定义 xxxx mediaTypes.put(
"xxxx",MediaType.parseMediaType("application/xxxx")); //指定支持解析哪些参数对应的哪些媒体类型 ParameterContentNegotiationStrategy parameterStrategy = new ParameterContentNegotiationStrategy(mediaTypes);
       // 修改内容协商URL中请求参数format值 修改后 请求URL http://localhost:8080/getUser?f=xxxx
          // parameterStrategy.setParameterName("f")
         // 添加基于请求头的内容协商 如果不加 请求头怎么修改都是返回第一个类型 现在第一个类型是json HeaderContentNegotiationStrategy headeStrategy
= new HeaderContentNegotiationStrategy(); configurer.strategies(Arrays.asList(parameterStrategy,headeStrategy)); } @Override public void extendMessageConverters(List<HttpMessageConverter<?>> converters) { converters.add(new MyMessageConverter()); } }; } }
@Controller
public class MyController {

    @ResponseBody
    @GetMapping("/getUser")
    public User getUser(){
        User user = new User();
        user.setAge(20);
        user.setName("张三");
        return user;
    }
}

输出结果

 

posted @ 2022-07-26 13:18  Dabo丶  阅读(398)  评论(0编辑  收藏  举报