Springboot中的restTemplate携带请求头请求体访问第三方接口

restTemplate配置类

import org.springframework.boot.web.client.RestTemplateBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.client.RestTemplate;

@Configuration
public class RestTemplateConfig {
    @Bean
    public RestTemplate restTemplate(RestTemplateBuilder builder) {
        return builder.build();
    }
}

控制层

import com.example.jutongdascrmdemo2.Domain.Test2;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;


import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.client.RestTemplate;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.Enumeration;


@RequestMapping("/u")
@RestController
public class UserController {

    @Autowired
    private RestTemplate restTemplate;
    @Autowired
    private RestTemplate restTemplate2;

    @ApiOperation("普通body请求")
    @PostMapping("/add")
    public Test2 add(HttpServletRequest request, HttpServletResponse response, @RequestBody Test2 test2) throws IOException {

        //获取请求头
        Enumeration<String> headerNames = request.getHeaderNames();

        ResponseEntity<String> forEntity = restTemplate.getForEntity("https://www.baidu.com/", String.class);
        Test2 test3 = new Test2();
        test3.setAge("999");
        test3.setName("999");

        return test3;
    }

    @ApiOperation("带token")
    @PostMapping("/add2")
    public Test2 add2(HttpServletRequest request, HttpServletResponse response) {
        Test2 test2 = new Test2();
        test2.setAge("111");
        test2.setName("111");
        HttpHeaders requestHeaders = new HttpHeaders();
        requestHeaders.set("Token", "1122339955555.....");
        HttpEntity requestEntity = new HttpEntity(test2, requestHeaders);

        ResponseEntity<Test2> exchange = restTemplate.postForEntity(
                "http://127.0.0.1:8080/u/add",
                requestEntity,
                Test2.class);
        Test2 body = exchange.getBody();

        return body;
    }
}

以上最终实现了要求,但过于繁琐

参考

使用RestTemplate发请求时自动拼接header信息（一学就会系列）_前端拼接header_Plusroax的博客-CSDN博客

最终出现了如下代码

自定义拦截器TokenClientHttpRequestInterceptor

import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpRequest;
import org.springframework.http.client.ClientHttpRequestExecution;
import org.springframework.http.client.ClientHttpRequestInterceptor;
import org.springframework.http.client.ClientHttpResponse;
import org.springframework.stereotype.Component;
import java.io.IOException;


@Component
public class TokenClientHttpRequestInterceptor implements ClientHttpRequestInterceptor {

    @Override
    public ClientHttpResponse intercept(HttpRequest httpRequest, byte[] body, ClientHttpRequestExecution execution) throws IOException {
        //拿到请求头
        HttpHeaders headers = httpRequest.getHeaders();
        //你想做的操作，比如在请求头上加上Authorization信息
        String token = "your token";
        headers.add("Authorization",token);
trackRequest(httpRequest,body);
        //一定要放行
ClientHttpResponse execute = execution.execute(httpRequest,body);
trackResponse(execute);
        return execute;

    }



    private void trackResponse(ClientHttpResponse httpResponse)throws IOException {
        System.out.println("============================response begin==========================================");
        System.out.println("Status code  : {}"+httpResponse.getStatusCode());
        System.out.println("Status text  : {}"+httpResponse.getStatusText());
        System.out.println("Headers      : {}"+httpResponse.getHeaders());
        System.out.println("=======================response end=================================================");
    }

    private void trackRequest(HttpRequest request, byte[] body)throws UnsupportedEncodingException {
        System.out.println("======= request begin ========");
        System.out.println("uri : {}"+ request.getURI());
        System.out.println("method : {}"+ request.getMethod());
        System.out.println("headers : {}"+ request.getHeaders());
        System.out.println("request body : {}"+new String(body, "UTF-8"));
        System.out.println("======= request end ========");
    }

}

配置RestTemplateConfig，把自定义拦截器加入到RestTemplate对象中

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.client.RestTemplate;
import java.util.Collections;

@Configuration
public class RestTemplateConfig2 {
    @Autowired
    TokenClientHttpRequestInterceptor tokenClientHttpRequestInterceptor;
    //通过工厂创建restTemplate，并且把自定义拦截器添加到到restTemplate中
    @Bean("restTemplate2")
    public RestTemplate restTemplate() {
        RestTemplate restTemplate = new RestTemplate();
        // 把自定义的拦截器添加到RestTemplate，可添加多个,传入一个List<ClientHttpRequestInterceptor>类型的集合
        restTemplate.setInterceptors(Collections.singletonList(tokenClientHttpRequestInterceptor));
        return restTemplate;
    }
}

 控制层使用

    @ApiOperation("带token4")
    @PostMapping("/add4")
    public Test2 add4(HttpServletRequest request, HttpServletResponse response) {
        Test2 test2 = new Test2();
        test2.setAge("111");
        test2.setName("111");

        ResponseEntity<Test2> exchange = restTemplate2.postForEntity(
                "http://127.0.0.1:8080/u/add",
                test2,
                Test2.class);
        Test2 body = exchange.getBody();

        return body;
    }