springboot单独为某个接口设置超时时间

Question

写了个导入接口，但是导入时间比较长，想单独为这个接口设置一个超时时间，或者直接没有超时，其他接口和原来一样。有什么办法吗

Answer 1

引用 皆我百晓生 小程序回复内容作答：
有几种办法可以实现为某个接口设置超时时间：

1. 使用@Async注解：可以将导入接口的方法标记为异步，然后在调用该方法时使用CompletableFuture设置超时时间。示例代码如下：

@RestController
public class ImportController {
    
    @Autowired
    private ImportService importService;
    
    @Async
    @PostMapping("/import")
    public CompletableFuture<Object> importData() {
        return CompletableFuture.supplyAsync(() -> importService.importData())
                .orTimeout(10, TimeUnit.SECONDS) // 设置超时时间为10秒
                .exceptionally(ex -> {
                    // 处理超时异常
                    return null;
                });
    }
}

1. 使用RestTemplate：可以在导入接口的实现中使用RestTemplate来发起HTTP请求，并设置超时时间。示例代码如下：

@RestController
public class ImportController {
    
    @PostMapping("/import")
    public Object importData() {
        RestTemplate restTemplate = new RestTemplate();
        SimpleClientHttpRequestFactory requestFactory = new SimpleClientHttpRequestFactory();
        requestFactory.setConnectTimeout(10000); // 设置连接超时时间为10秒
        requestFactory.setReadTimeout(10000); // 设置读取超时时间为10秒
        restTemplate.setRequestFactory(requestFactory);
        
        // 发起HTTP请求
        String result = restTemplate.postForObject("http://localhost:8080/data", null, String.class);
        return result;
    }
}

1. 使用定制的HttpClient：可以创建一个定制的HttpClient，并为导入接口设置超时时间。示例代码如下：

@Configuration
public class HttpClientConfig {

    @Bean
    public CloseableHttpClient httpClient() {
        RequestConfig requestConfig = RequestConfig.custom()
                .setConnectTimeout(10 * 1000) // 设置连接超时时间为10秒
                .setSocketTimeout(10 * 1000) // 设置读取超时时间为10秒
                .build();
        
        return HttpClientBuilder.create().setDefaultRequestConfig(requestConfig).build();
    }
}

@RestController
public class ImportController {
    
    @Autowired
    private CloseableHttpClient httpClient;
    
    @PostMapping("/import")
    public Object importData() {
        HttpPost httpPost = new HttpPost("http://localhost:8080/data");
        httpPost.setEntity(new ByteArrayEntity(new byte[0]));
        
        try {
            CloseableHttpResponse response = httpClient.execute(httpPost);
            
            // 处理返回结果
            return EntityUtils.toString(response.getEntity());
        } catch (Exception e) {
            // 处理异常
            return null;
        }
    }
}

以上是几种常用的方法，你可以根据自己的需求选择其中一种来实现为某个接口设置超时时间。