自定义注解+切面AOP实现当前服务超时控制

1 自定义注解

import java.lang.annotation.*;

@Inherited
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Timeout {
    long value() default 1000;
}

2 切面(AOP)

import com.gosun.pcms.common.ReturnDTO;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;

/**
 * @author yiqiang
 * @date 2020/8/12 13:38
 */
@Aspect
@Slf4j
@Component
public class TimeoutAspect {

    @Pointcut(value = "@annotation(com.gosun.pcms.aop.Timeout)")
    public void pointCut() {
    }


    @Around(value = "pointCut()")
    public Object timeOut(ProceedingJoinPoint joinPoint) throws Exception {
        log.info("正在进入切面");
        Method method = ((MethodSignature) joinPoint.getSignature()).getMethod();
        Timeout timeOut = method.getDeclaredAnnotation(Timeout.class);
        int i = 0;
        ExecutorService executor = Executors.newCachedThreadPool();
        Future future = executor.submit(() -> {
            log.info("正在执行");
            try {
                return joinPoint.proceed();
            } catch (Throwable throwable) {
                throwable.printStackTrace();
                return ReturnDTO.fail("操作失败");
            }
        });

        long t = System.currentTimeMillis();
        while (true) {
            Thread.sleep(1000);
            log.info("第{}次执行", i++);
            if (System.currentTimeMillis() - t >= timeOut.value()) {
                return ReturnDTO.fail("超时,TimeOut:"+timeOut.value());
            }
            if (future.isDone()) {
                return future.get();
            }
        }

    }

}