还在手写重试，不妨试试Spring Retry（二）

最新推荐文章于 2024-02-28 20:01:06 发布

青Cheng序员石头

最新推荐文章于 2024-02-28 20:01:06 发布

阅读量2.1k

点赞数

本文链接：https://blog.csdn.net/DynmicResource/article/details/121727947

版权

theme: scrolls-light

🤞 个人主页：@青Cheng序员石头
🤞 粉丝福利：加粉丝群一对一问题解答，获取免费的丰富简历模板、提高学习资料等，做好新时代的卷卷王！

紧接着上一篇《还在手写重试，不妨试试Spring Retry(一)》，点击前往。

四、RetryTemplate

RetryTemplate是使用注解形式重试的一种替代。

4.1 RetryOperations

使用 RetryOperations 接口的 Spring retry providesRetryOperations 策略。

public interface RetryOperations { <T> T execute(RetryCallback < T > retryCallback) throws Exception; // other execute methods <T> T execute(RetryCallback < T > retryCallback, RecoveryCallback < T > recoveryCallback, RetryState retryState) throws Exception; }

RetryCallback允许插入在失败时需要重试的业务逻辑。

public interface RetryCallback <T> { T doWithRetry(RetryContext context) throws Throwable; }

4.2 RetryTemplate

RetryTemplate提供了RetryOperations的一种具体实现。它被认为是从中创建bean的良好做法。

@Bean @ConditionalOnMissingBean public RetryTemplate retryTemplate(){ final SimpleRetryPolicy simpleRetryPolicy = new SimpleRetryPolicy(); simpleRetryPolicy.setMaxAttempts(4); final FixedBackOffPolicy fixedBackOffPolicy = new FixedBackOffPolicy(); fixedBackOffPolicy.setBackOffPeriod(1000L); return RetryTemplate.builder() .customPolicy(simpleRetryPolicy) .customBackoff(fixedBackOffPolicy) .retryOn(CustomRetryException.class) .build(); }

进行单元测试。

@Autowired pivate RetryTemplate retryTemplate; @Test void retryWithoutAnnotation(){ try { String message = retryTemplate.execute(x -> retryService.retryWithoutAnnotation()); log.info("message = "+message); } catch (CustomRetryException e) { log.error("Error while executing test {}",e.getMessage()); } }

如下图的输出结果所示，也是重试了四次，但是没有recover的策略。

4.2 RecoveryCallback

execute时，可以选择输入RecoveryCallback回调，确定重试结束后，仍然出现异常的recovery行为。方法签名如下。

public final <T, E extends Throwable> T execute(RetryCallback<T, E> retryCallback, RecoveryCallback<T> recoveryCallback) throws E { return this.doExecute(retryCallback, recoveryCallback, (RetryState)null); }

所以，我们先自定义RecoveryCallback。

@Slf4j public class CustomRecoveryCallback implements RecoveryCallback<String> { @Override public String recover(RetryContext retryContext) throws Exception { log.info("Default Retry service test,total retry {}",retryContext.getRetryCount()); return "Error Class :: " + retryContext.getLastThrowable().getClass().getName(); } }

然后进行单元测试。

@Autowired private CustomRecoveryCallback customRecoveryCallback; @Test void retryWithoutAnnotation(){ try { String message = retryTemplate.execute(x -> retryService.retryWithoutAnnotation(), customRecoveryCallback); log.info("message = "+message); } catch (CustomRetryException e) { log.error("Error while executing test {}",e.getMessage()); } }

如下图的输出结果所示,重试完成后，执行了我们自定义的recover。

五、RetryListenerSupport

如果我们想在重试整个生命周期中，按照不同的阶段设置一些事件监听处理机制，那怎么办呢？设置自定义的RetryListenerSupport能帮助到我们。我们继承RetryListenerSupport，并重新Override close 、onError、open方法，这三个方法分别表示

所有重试结束时
每一次重试发生异常时
重试正式开始前。

@Slf4j public class DefaultListenerSupport extends RetryListenerSupport { @Override public <T, E extends Throwable> void close(RetryContext context, RetryCallback<T, E> callback, Throwable throwable) { log.info("DefaultListenerSupport close"); super.close(context, callback, throwable); } @Override public <T, E extends Throwable> void onError(RetryContext context, RetryCallback<T, E> callback, Throwable throwable) { log.info("DefaultListenerSupport onError"); super.onError(context, callback, throwable); } @Override public <T, E extends Throwable> boolean open(RetryContext context, RetryCallback<T, E> callback) { log.info("DefaultListenerSupport open"); return super.open(context, callback); } }

并且在构造RetryTemaplate时候，设置withListener字段。

``` @Bean @ConditionalOnMissingBean public RetryListenerSupport retryListenerSupport(){ return new DefaultListenerSupport(); } @Bean @ConditionalOnMissingBean public RetryTemplate retryTemplate(RetryListenerSupport retryListenerSupport){ ...

return RetryTemplate.builder() .customPolicy(simpleRetryPolicy) .customBackoff(fixedBackOffPolicy) .withListener(retryListenerSupport) .retryOn(CustomRetryException.class) .build(); } ```

运行单元测试，输出结果如下。