Spring boot ClientHttpRequestInterceptor 在 401 上重新发送

使用Spring Boot的ClientHttpRequestInterceptor在401状态码下重新发送请求

java
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpRequest;
import org.springframework.http.HttpStatus;
import org.springframework.http.client.ClientHttpRequestExecution;
import org.springframework.http.client.ClientHttpRequestInterceptor;
import org.springframework.http.client.ClientHttpResponse;
import org.springframework.util.StreamUtils;
import java.io.IOException;
import java.nio.charset.Charset;
public class RetryInterceptor implements ClientHttpRequestInterceptor {
    @Override
    public ClientHttpResponse intercept(HttpRequest request, byte[] body, ClientHttpRequestExecution execution) throws IOException {
        ClientHttpResponse response = execution.execute(request, body);
        if (response.getStatusCode() == HttpStatus.UNAUTHORIZED) {
            HttpHeaders headers = response.getHeaders();
            // 获取新的身份验证凭证
            String newToken = getNewToken();
            headers.set("Authorization", newToken);
            // 重新发送请求
            response = execution.execute(request, body);
        }
        return response;
    }
    private String getNewToken() {
        // 实现获取新的身份验证凭证的逻辑
        return "newToken";
    }
}

java
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.annotation.Bean;
import org.springframework.http.client.ClientHttpRequestInterceptor;
import org.springframework.web.client.RestTemplate;
import java.util.Collections;
@SpringBootApplication
public class Application {
    public static void main(String[] args) {
        SpringApplication.run(Application.class, args);
    }
    @Bean
    public RestTemplate restTemplate() {
        RestTemplate restTemplate = new RestTemplate();
        restTemplate.setInterceptors(Collections.singletonList(retryInterceptor()));
        return restTemplate;
    }
    @Bean
    public ClientHttpRequestInterceptor retryInterceptor() {
        return new RetryInterceptor();
    }
}

java
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.client.RestTemplate;
@RestController
public class MyController {
    @Autowired
    private RestTemplate restTemplate;
    @GetMapping("/api/data")
    public ResponseEntity getData() {
        String url = "https://api.example.com/data";
        ResponseEntity response = restTemplate.getForEntity(url, String.class);
        return response;
    }
}