RestTemplate設置請求頭詳解

一、什麼是RestTemplate？

RestTemplate是spring framework中的一個HTTP請求庫，用於發送HTTP請求並處理HTTP響應。它可以發送GET，POST，PUT，DELETE等HTTP請求，並提供了多種設置請求頭的方法。

二、為什麼要設置請求頭？

在HTTP請求的過程中，請求頭是非常重要的。請求頭中包含了很多關鍵信息，如User-Agent，Content-Type等。這些信息可以幫助伺服器正確識別請求的來源，並決定如何處理請求。

在實際應用中，我們常常需要設置請求頭來滿足對接介面的要求，例如設置Authorization頭以驗證用戶身份。

三、如何設置請求頭？

1. 使用HttpHeaders

HttpHeaders是spring framework中一個專門用於設置HTTP請求和響應頭的類，在RestTemplate中的應用非常廣泛。

RestTemplate restTemplate = new RestTemplate();
HttpHeaders headers = new HttpHeaders();
headers.add("Authorization", "Bearer " + token);
headers.add("Content-Type", "application/json");

HttpEntity requestEntity = new HttpEntity(body, headers);
ResponseEntity<String> responseEntity = restTemplate.exchange(url, HttpMethod.POST, requestEntity, String.class);

在上述代碼中，我們使用HttpHeaders設置了兩個請求頭：Authorization和Content-Type。使用HttpEntity將請求體和請求頭封裝在一起，並傳入exchange方法中。

2. 使用HttpComponentsClientHttpRequestFactory

RestTemplate默認使用的HTTP請求實現是JDK自帶的，如果需要更多的配置選項，可以使用Apache HttpComponents作為RestTemplate的HTTP請求實現，並且可以通過HttpComponentsClientHttpRequestFactory設置請求頭。

RestTemplate restTemplate = new RestTemplate(new HttpComponentsClientHttpRequestFactory());
HttpHeaders headers = new HttpHeaders();
headers.add("Authorization", "Bearer " + token);
headers.add("Content-Type", "application/json");

HttpEntity requestEntity = new HttpEntity(body, headers);
ResponseEntity<String> responseEntity = restTemplate.exchange(url, HttpMethod.POST, requestEntity, String.class);

在上述代碼中，我們創建了一個HttpComponentsClientHttpRequestFactory實例，並將其傳入RestTemplate的構造函數中。然後使用HttpHeaders設置請求頭，並將請求體和請求頭封裝在一起作為參數傳入exchange方法。

3. 使用ClientHttpRequestInterceptor

ClientHttpRequestInterceptor是一種可以攔截所有RestTemplate發送的請求並對其進行修改的方式。通過實現ClientHttpRequestInterceptor介面，並在RestTemplate中註冊該攔截器，可以非常靈活地設置請求頭。

RestTemplate restTemplate = new RestTemplate();
restTemplate.setInterceptors(Collections.singletonList(new ClientHttpRequestInterceptor() {
    @Override
    public ClientHttpResponse intercept(HttpRequest request, byte[] body, ClientHttpRequestExecution execution) throws IOException {
        HttpHeaders headers = request.getHeaders();
        headers.add("Authorization", "Bearer " + token);
        headers.add("Content-Type", "application/json");
        return execution.execute(request, body);
    }
}));

ResponseEntity<String> responseEntity = restTemplate.exchange(url, HttpMethod.POST, new HttpEntity<>(body), String.class);

在上述代碼中，我們創建了一個ClientHttpRequestInterceptor介面的實例，並在RestTemplate中註冊該攔截器。攔截器對請求進行了修改，設置了Authorization和Content-Type頭。

四、總結

在HTTP請求中，請求頭是非常重要的。通過設置請求頭，我們可以發送帶有關鍵信息的HTTP請求，並滿足介面對請求頭的要求。RestTemplate提供了多種設置請求頭的方式，開發者可以根據自己的需求靈活選擇。