FixBug日记之我居然被某些工具坑了

Posted 2022-02-28 go大鸡腿

文章目录

• 前言
• 排查现场
• 破案
• • 解决方法

前言

我最近在对接某些平台的接口，然后用的是restTemplate来请求
首先当然是照着文档，然后postman去调一下，都是成功的，很好~然后呢，我用这个工具实现远程调用，然后一直报contentType不支持，但是我已经设置了一样的请求头了

排查现场

1. 普及知识
   我们在做组件特殊化配置的时候，应该避免影响其他人使用。比如说有两个实现类都实现了一个父类，注入的时候肯定会报错

   解决方案
   @Bean 分别写不同的注入方法名，然后@Quailty去指向对应的实现类即可
   @Primary这个注解主要是告诉spring，优先使用哪个类

2. 开始排查
   代码上肯定没有问题的，我就是这么自信~
   如下图，就是form表单格式提交，很简单，为啥一直说不支持呢？文档也是写着要这个header头
   

3. debug模式
   可以看到restTemplate里头有很多转换器
   
   最关键的就是FormHttpMessageConverter

4. FormHttpMessageConverter
   在代码里头会判断字符编码类型，没有就抛出异常
   

5. 看到getFormContentType 方法
   在这个方法里头如果没有设置字符集的话，默认给我们set 字符集
   

破案

对接文档里头，也是奇葩，如果你在contentType里面加上charse=utf-8就报错
resttemplate也是个奇葩，非要给你加个字符集

我就要刚到底

解决方法

思路
有很多转换器，那么我们拿到对应的转换器列表，偷偷替换掉，然后重写方法，还有个棘手的就是如果我set一个MediaType没有字符集，在判断的时候又不通过，其实我们可以用反射大法塞进去！

开干

@Bean
public RestTemplate beiSenTokenRestTemplate() 
    RestTemplate restTemplate = new RestTemplate(new OkHttp3ClientHttpRequestFactory(new OkHttpClient().newBuilder()
            // 建立连接超时时间
            .connectTimeout(5, TimeUnit.SECONDS)
            // 读取数据超时时间
            .readTimeout(60, TimeUnit.SECONDS)
            // 写入数据超时时间
            .writeTimeout(60, TimeUnit.SECONDS)
            // 证书校验
            .sslSocketFactory(SSL.getSSLSocketFactory(), SSL.getX509TrustManager())
            // 主机校验
            .hostnameVerifier(SSL.getHostnameVerifier())
            .build()));
    List<HttpMessageConverter<?>> list = restTemplate.getMessageConverters();
    list.removeIf(item -> item instanceof FormHttpMessageConverter);
    list.add(new FormHttpMessageConverter() 

        @Override
        protected MediaType getFormContentType(MediaType contentType) 
            MediaType type = new MediaType(MediaType.APPLICATION_FORM_URLENCODED);

            if (contentType == null) 
                return new MediaType(MediaType.APPLICATION_FORM_URLENCODED, DEFAULT_CHARSET);
             else if (type.getType().equals(contentType.getType())) 
                setCharset(StandardCharsets.UTF_8);

                try 
                    Field fi = MediaType.class.getSuperclass().getDeclaredField("resolvedCharset");
                    fi.setAccessible(true);
                    fi.set(type, StandardCharsets.UTF_8);
                 catch (NoSuchFieldException | IllegalAccessException e) 
                    return new MediaType(contentType, DEFAULT_CHARSET);
                
                return type;
             else if (contentType.getCharset() == null) 
                return new MediaType(contentType, DEFAULT_CHARSET);
             else 
                return contentType;
            
        
    );
    return restTemplate;

List<HttpMessageConverter<?>> list = restTemplate.getMessageConverters();
    list.removeIf(item -> item instanceof FormHttpMessageConverter);

这块是为了替换掉form转换器，然后塞一个新的进去
然后重写FormHttpMessageConverter getFormContentType方法，如果contentType跟APPLICATION_FORM_URLENCODED一样的时候，我们将
MediaType里面resolvedCharset 给一个默认值

为什么是MediaType.class.getSuperclass()？
因为resolvedCharset 是它父类的值