通过aop注解实现自动重试的功能

有时候，我们在程序中不得不对某些方法进行重试，比如说，请求某个接口，如果我们每次，都通过while和count–的写法无疑会让代码，变的很不优雅。因此我们这边用spring的aop设计一下这个功能。

1.我们定义一个Exception，如果我们想重试的方法，可以指定抛出RetryException异常

@EqualsAndHashCode(callSuper = true)
@Data
public class RetryException extends RuntimeException {

    private static final long serialVersionUID = 7886918292771470846L;

    public RetryException(String message) {
        super(message);
    }
}

2.定义我们的retry注解

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public @interface RetryOperation {
    /** 重试次数 */
    int retryCount() default 3;

    /** 重试间隔 */
    int waitSeconds() default 10;
}

3.实现我们的interceptor逻辑

稍微解释一下下面的代码，我们around环绕，可以让我们在method执行之前对他进行拦截，然后只catch我们指定的RetryException,

对整个代码进行重新调用。

@Aspect
@Component
@Slf4j
public class RetryAspect {

    @Around(value = "@annotation(retryOperation)")
    public Object retryOperation(ProceedingJoinPoint joinPoint, RetryOperation retryOperation)
            throws Throwable {

        log.info("指定方法异常重试开始");
        Object response = null;
        int retryCount = retryOperation.retryCount();
        int waitSeconds = retryOperation.waitSeconds();
        boolean successful = false;

        do {
            try {
                response = joinPoint.proceed();
                successful = true;
            } catch (RetryException ex) {
                log.info("Operation failed, retries remaining: {}", retryCount);
                retryCount--;
                if (retryCount < 0) {
                    successful = true;
                    log.error(ex.getMessage());
                }
                if (waitSeconds > 0 && !successful) {
                    log.info("Waiting for {} second(s) before next retry", waitSeconds);
                    Thread.sleep(waitSeconds * 1000L);
                }
            }
        } while (!successful);

        return response;
    }
}

4.测试代码

@RetryOperation(retryCount = 3, waitSeconds = 10)
@GetMapping("/retry")
public String retry() {
    throw new RetryException("aop执行");
}

通过console打印结果，得到了我们需要的效果