RestTemplate-源码阅读

news2024/11/13 9:38:12

参考资料：https://github.com/mercyblitz/java-training-camp/blob/main/stage-1/docs/04.%20%E7%AC%AC%E5%9B%9B%E8%8A%82%EF%BC%9AREST%20API%20%E5%AE%A2%E6%88%B7%E7%AB%AF%E8%AE%BE%E8%AE%A1.md

RestTemplate 源码阅读

GET POST EXCHANGE

在这里插入图片描述

我们可以看到其实没有什么本质的区别，只是在创建CallBack的时候会有一点区别。另外还有exchange:

三者有何区别？

请求类型：GET 和 POST 分别针对特定的 HTTP 方法，而 EXCHANGE 方法则可以用于任何 HTTP 方法。
请求体：GET 方法不支持请求体，而 POST 和 EXCHANGE 方法支持请求体。
灵活性：EXCHANGE 方法提供了最大的灵活性，允许你自定义请求的构建过程，包括设置请求头、请求体等。
返回类型：这三个方法都返回 ResponseEntity，包含了响应的状态码、响应头和响应体

请求执行流程以GET为例

1 创建回调函数
在这里插入图片描述
我们可以看到有两种回调函数，都是RestTemplate的内部类。而且HttpEntityRequestCallback继承了AcceptHeaderRequestCallback。可以看到HttpEntityRequestCallback多了可以携带Body的HttpEntity。然后里面的核心方法是doWithRequest：

设置请求头：例如设置 Content-Type、Authorization 等。
设置请求体：例如设置 POST 请求的请求体。
执行其他预处理逻辑：例如日志记录、性能监控等。

2 核心API用来进行数据提取

HttpMessageConverterExtractor<T> responseExtractor =
				new HttpMessageConverterExtractor<>(responseType, getMessageConverters(), logger);

也许你会看到exchange中ResponseEntityResponseExtractor：
在这里插入图片描述
但其实内部也是委托给HttpMessageConverterExtractor。我们可以看到有这样几种Converter。

在这里插入图片描述
至于如何选择Converters是通过下面这个方法，canWrite 是发送请求的时候些Body，canRead是调用结束以后读取Body。

在这里插入图片描述

然后就是执行调用了：

调用执行完毕以后处理返回结果，可以看到如果有错误就会使用ResponseErrorHandler 进行错误处理：

// 如果没有设置错误处理器则有默认的
private ResponseErrorHandler errorHandler = new DefaultResponseErrorHandler();

如果没有错误就是提取结果了，同样的还是HttpMessageConverter。

底层实现

我们可以看到可以指定ClientHttpRequestFactory

public RestTemplate restTemplate(ClientHttpRequestFactory factory) {
   RestTemplate restTemplate = new RestTemplate(factory);
   return restTemplate;
}
@Bean
public ClientHttpRequestFactory simpleClientHttpRequestFactory() {
   SimpleClientHttpRequestFactory factory = new SimpleClientHttpRequestFactory();
   factory.setReadTimeout(5000);
   factory.setConnectTimeout(15000);
   return factory;
}

有下面三种实现：

JDK HttpURLConnection - SimpleClientHttpRequestFactory（默认）
Apache HttpComponents HttpClient - HttpComponentsClientHttpRequestFactory
OkHttp3 - OkHttp3ClientHttpRequestFactory

拓展 InterceptingClientHttpRequestFactory

这个工工厂提供了我们自定义拦截器进行处理的能力，这里可以进行一些自定义的拓展。就比如我们可以在调用之前自定义一个拦截器进行参数校验，又或者进行服务与请求与重试。
这个类其实就是ClientHttpRequestFactory 的一个包装类，怎么实现拦截的呢？我们来看看，其实一眼就能看到，核心的逻辑都是委托给ClientHttpRequestFactory来实现的。
在这里插入图片描述