springboot如何针对单个接口设置超时时间?

Question

springboot如何针对单个接口设置超时时间?

比如a接口设置超时时间为5s，b接口设置超时时间为10s

关注者

11

被浏览

87,561

4 个回答

浮生若梦

先看两个接口

这两个接口springboot已经帮我们封装好了，我们不需要去手动使用

TaskScheduler //任务调度者
TaskExecutor //任务执行者

具体步骤：

在启动类上添加这个注解：

@EnableScheduling//开启定时功能的注解

给需要定时任务的方法添加@Scheduled(cron="0 * * * * *")注解

@Service
public class ScheduledService {
    //cron
    //秒 分 时 日 月 周几（0-7） 年
    //每到0秒执行一次
    @Scheduled(cron="0 * * * * *")
    30 15 10 * * ? 每天10点15分30秒执行一次

程序员阿紫 · Accepted Answer

首先我们得明白一件事，接口超时了，只是等待结果响应的线程提前响应给前端。但实际上运行业务逻辑的线程还在跑。

所以我们只需要自定义这个机制，比如利用切面加并发编程中Future，就能达到这个效果。然后再根据配置，让a接口的超时时间为5秒，B接口的超时时间为10秒。就可以了。

案例：

a和b接口

@RestController
@RequestMapping("/foo")
public class FooController {
    @GetMapping("/a")
    public String a() throws InterruptedException {
        TimeUnit.SECONDS.sleep(2);
        return "ok";
    @GetMapping("/b")
    public String b() throws InterruptedException {
        TimeUnit.SECONDS.sleep(2);
        return "ok";
}

切面

@Component
@Aspect
public class ResponseTimeAspect {
    @Autowired
    private ThreadPoolTaskExecutor threadPoolTaskExecutor;
    @Around("@annotation(getMapping)")
    public Object around(ProceedingJoinPoint joinPoint, GetMapping getMapping) {
        final Future<Object> future = threadPoolTaskExecutor.submit(() -> {
            try {
                return joinPoint.proceed();
            } catch (Throwable throwable) {
                throw new RuntimeException(throwable);
        try {
            HttpServletRequest request =
                    ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest();
            final String uri = request.getRequestURI();
            if("/foo/a".endsWith(uri)){
                // 获取响应，超过1秒抛出超时异常
                return future.get(1, TimeUnit.SECONDS);
            }else if ("/foo/b".endsWith(uri)){
                return future.get(3, TimeUnit.SECONDS);
            return future.get(5, TimeUnit.SECONDS);
        } catch (InterruptedException e) {
            return "被中断了";
        } catch (ExecutionException e) {
            return "执行任务发生异常";
        } catch (TimeoutException e) {
            return "超时";
}

以上判断接口的逻辑可改为配置，catch的异常可以改成任意你想要的。

注意点：由于是单开线程进行调用业务代码，如果业务代码中有从线程上下文获取信息是会出问题，比如在Filter中往ThreadLocal中放入了User，在Servcie想从ThreadLocal获取User，会获取不到。

解决方案：在调用 joinPoint.proceed 前进行线程上下文传递。

当然，如果有更好的方式可以弃用这个方案，不过我看了下tomcat的源码，没找到很好的切入点。