Feign调用可重试的最佳方案分享

前言

在我们公司里，不同的服务之间通过Feign进行远程调用，但是，我们在尝试使调用可重试时遇到了一个小问题，Feign框架本身可以配置的自己的重试机制，但是它是一刀切的方式，所有的调用都是同样的机制，没有办法像我们希望的那样在每个方法的基础上配置。不过我在项目中探索除了一种新的写法，通过spring-retry框架集合Feign去实现重试机制，可以为每个调用实现不同的重试机制，那究竟是如何做到的呢，继续往下看呀。

自定义注解@FeignRetry

为了解决上面提到的问题，让Feign调用的每个接口单独配置不同的重试机制。我们使用了面向切面编程并编写了一个自定义注解：@FeignRetry。此注释的工作方式类似于@Retryable的包装器，并与其共享相同的规范以避免混淆。

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface FeignRetry {

Backoff backoff() default @Backoff();
   int maxAttempt() default 3;
   Class<? extends Throwable>[] include() default {};
}

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
public @interface Backoff {
   long delay() default 1000L;;
   long maxDelay() default 0L;
   double multiplier() default 0.0D;;
}

FeignRetryAspect切面处理@FeignRetry注解。

Slf4j
@Aspect
@Component
public class FeignRetryAspect {

@Around("@annotation(FeignRetry)")
   public Object retry(ProceedingJoinPoint joinPoint) throws Throwable {
       Method method = getCurrentMethod(joinPoint);
       FeignRetry feignRetry = method.getAnnotation(FeignRetry.class);

RetryTemplate retryTemplate = new RetryTemplate();
       retryTemplate.setBackOffPolicy(prepareBackOffPolicy(feignRetry));
       retryTemplate.setRetryPolicy(prepareSimpleRetryPolicy(feignRetry));

// 重试
       return retryTemplate.execute(arg0 -> {
           int retryCount = arg0.getRetryCount();
           log.info("Sending request method: {}, max attempt: {}, delay: {}, retryCount: {}",
                   method.getName(),
                   feignRetry.maxAttempt(),
                   feignRetry.backoff().delay(),
                   retryCount
           );
           return joinPoint.proceed(joinPoint.getArgs());
       });
   }

private BackOffPolicy prepareBackOffPolicy(FeignRetry feignRetry) {
       if (feignRetry.backoff().multiplier() != 0) {
           ExponentialBackOffPolicy backOffPolicy = new ExponentialBackOffPolicy();
           backOffPolicy.setInitialInterval(feignRetry.backoff().delay());
           backOffPolicy.setMaxInterval(feignRetry.backoff().maxDelay());
           backOffPolicy.setMultiplier(feignRetry.backoff().multiplier());
           return backOffPolicy;
       } else {
           FixedBackOffPolicy fixedBackOffPolicy = new FixedBackOffPolicy();
           fixedBackOffPolicy.setBackOffPeriod(feignRetry.backoff().delay());
           return fixedBackOffPolicy;
       }
   }

private SimpleRetryPolicy prepareSimpleRetryPolicy(FeignRetry feignRetry) {
       Map<Class<? extends Throwable>, Boolean> policyMap = new HashMap<>();
       policyMap.put(RetryableException.class, true);  // Connection refused or time out
       policyMap.put(ClientException.class, true);     // Load balance does not available (cause of RunTimeException)
       if (feignRetry.include().length != 0) {
           for (Class<? extends Throwable> t : feignRetry.include()) {
               policyMap.put(t, true);
           }
       }
       return new SimpleRetryPolicy(feignRetry.maxAttempt(), policyMap, true);
   }

private Method getCurrentMethod(JoinPoint joinPoint) {
       MethodSignature signature = (MethodSignature) joinPoint.getSignature();
       return signature.getMethod();
   }
}

捕获FeignRetry注解的方法，将配置传递给Spring RetryTemplate，根据配置调用服务。

@FeignRetry 的使用

用法很简单，只需将注释放在我们希望重试机制处于活动状态的 Feign Client 方法上即可。自定义切面的用法类似于Spring自带的@Retryable注解。

@GetMapping
@FeignRetry(maxAttempt = 3, backoff = @Backoff(delay = 500L))
ResponseEntity<String> retrieve1();

@GetMapping
@FeignRetry(maxAttempt = 6, backoff = @Backoff(delay = 500L, maxDelay = 20000L, multiplier = 4))
ResponseEntity<String> retrieve2();

另外还需要在应用程序类中使用 @EnableRetry 注释来启动重试，比如可以加载SpringBoot的启动类中。

来源：https://juejin.cn/post/7187252990821007397