怎么對Spring的Resttemplate調用進行攔截

這篇文章主要介紹“怎么對Spring的Resttemplate調用進行攔截”，在日常操作中，相信很多人在怎么對Spring的Resttemplate調用進行攔截問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”怎么對Spring的Resttemplate調用進行攔截”的疑惑有所幫助！接下來，請跟著小編一起來學習吧！

Spring RestTemplate經常被用作客戶端向Restful API發送各種請求，也許你也碰到過這種需求，很多請求都需要用到相似或者相同的Http Header。如果在每次請求之前都把Header填入HttpEntity/RequestEntity，這樣的代碼會顯得十分冗余。

Spring提供了ClientHttpRequestInterceptor接口，可以對請求進行攔截，并在其被發送至服務端之前修改請求或是增強相應的信息。

首先，我們自定義一個攔截器 UserContextInterceptor

public class UserContextInterceptor implements ClientHttpRequestInterceptor {
    private static final Logger logger = LoggerFactory.getLogger(UserContextInterceptor.class);
    @Override
    public ClientHttpResponse intercept(
            HttpRequest request, byte[] body, ClientHttpRequestExecution execution)
            throws IOException {

        HttpHeaders headers = request.getHeaders();
        headers.add("test", "我是攔截器添加的請求頭");
        return execution.execute(request, body);
    }
}

這個攔截器的作用很簡單，就是往請求頭里面添加一個信息。

接下來要想對RestTemplate起作用，還需要一些配置：我們需要定義一個 restTemplate bean，然后，把上面我們自定義的攔截器給添加進去。

@LoadBalanced
    @Bean
    public RestTemplate getRestTemplate(){
        RestTemplate template = new RestTemplate();
        List interceptors = template.getInterceptors();
        if (interceptors==null){
            template.setInterceptors(Collections.singletonList(new UserContextInterceptor()));
        }
        else{
            interceptors.add(new UserContextInterceptor());
            template.setInterceptors(interceptors);
        }

        return template;
    }

有了這個bean定義，每當使用 @Autowrited注解將 RestTemplate 注入一個類，就會使請求頭中添加一個test的信息。

到此，關于“怎么對Spring的Resttemplate調用進行攔截”的學習就結束了，希望能夠解決大家的疑惑。理論與實踐的搭配能更好的幫助大家學習，快去試試吧！若想繼續學習更多相關知識，請繼續關注億速云網站，小編會繼續努力為大家帶來更多實用的文章！