Spring-Retry 框架实战经典重试场景

原创

程序员蜗牛 2024-07-31 22:16:58 博主文章分类：springboot ©著作权

文章标签 spring java 随机数 文章分类 Java 后端开发 yyds干货盘点

©著作权归作者所有：来自51CTO博客作者程序员蜗牛的原创作品，请联系作者获取转载授权，否则将追究法律责任

Spring-Retry框架是Spring自带的功能，具备间隔重试、包含异常、排除异常、控制重试频率等特点，是项目开发中很实用的一种框架。

1、引入依赖

坑点：需要引入AOP，否则会抛异常。

<!-- Spring-Retry -->
<dependency>
    <groupId>org.springframework.retry</groupId>
    <artifactId>spring-retry</artifactId>
</dependency>
<!-- Spring-AOP -->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-aop</artifactId>
</dependency>

2、启动类注解

坑点：很容易一时疏忽忘记启动类开启@EnableRetry，大家别忘了哦。

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.retry.annotation.EnableRetry;

@SpringBootApplication
@EnableRetry
public class SpringRetryDemoApplication {

   public static void main(String[] args) {
      SpringApplication.run(SpringRetryDemoApplication.class, args);
   }

}

3、模拟发短信

我们模拟一个发短信功能，根据随机数分别作为成功、失败、抛出各种异常的入口。
这里抛出几种异常的目的，是为了后面演示出重试注解参数产生的效果。

import cn.hutool.core.util.RandomUtil;
import lombok.extern.slf4j.Slf4j;

/**
 * <p>
 * 短信服务工具类
 * </p>
 *
 */
@Slf4j
public class SmsUtil {

   /**
    * 发送短信
    */
   public static boolean sendSms() {

      // 使用随机数模拟重试场景
      int num = RandomUtil.randomInt(4);
      log.info("[SmsUtil][sendSms]>>>> random num = {}", num);

      return switch (num) {
         case 0 ->
               // 模拟发生参数异常
               throw new IllegalArgumentException("参数有误！");
         case 1 ->
               // 模拟发生数组越界异常
               throw new ArrayIndexOutOfBoundsException("数组越界！");
         case 2 ->
               // 模拟成功
               true;
         case 3 ->
               // 模拟发生空指针界异常
               throw new NullPointerException();
         default ->
               // 未成功则返回false
               false;
      };

   }
}

4、Retry应用

我们单独写一个用于重试调用的组件类，用于业务类调用。

import com.example.springretrydemo.util.SmsUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.retry.annotation.Backoff;
import org.springframework.retry.annotation.Recover;
import org.springframework.retry.annotation.Retryable;
import org.springframework.stereotype.Component;

import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;

/**
 * <p>
 * 重试组件
 * </p>
 *
 */
@Slf4j
@Component
public class RetryComponent {

   /**
    * 重试机制发送短信
    */
   @Retryable(
         retryFor = {IllegalArgumentException.class, ArrayIndexOutOfBoundsException.class},
         noRetryFor = {NullPointerException.class},
         maxAttempts = 4,
         backoff = @Backoff(delay = 2000L, multiplier = 2)
   )
   public boolean sendSmsRetry() {

      log.info("[RetryComponent][sendSmsRetry]>>>> 当前时间：{}", getNowTime());
      return SmsUtil.sendSms();
   }

   /**
    * 兜底方法，规则：
    *        1、超出了最大重试次数；
    *        2、抛出了不进行重试的异常；
    */
   @Recover
   public boolean recover() {
      log.info("[RetryComponent][recover]>>>> 短信发送次数过多，请稍后重试！");
      return false;
   }

   /**
    * 获取当前时间
    */
   private String getNowTime() {

      return LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"));
   }
}

@Retryable注解参数说明：

Spring-Retry 框架实战经典重试场景_spring