5分鐘攻略Spring-Retry框架實現經典重試場景

2023-12-22 12:01:06

前言

今天分享乾貨,控制了篇幅,5分鐘內就能看完學會。

主題是Spring-Retry框架的應用,做了一個很清晰的案例,程式碼可下載自測。

框架介紹

Spring-Retry框架是Spring自帶的功能,具備間隔重試包含異常排除異常控制重試頻率等特點,是專案開發中很實用的一種框架。

本篇所用框架的版本如下:

技術 版本
Java 17
SpringBoot 3.2
Spring-retry 2.0.4

正文

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>
 *
 * @author 程式設計師濟癲
 * @since 2023-12-21 09:40
 */
@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>
 *
 * @author 程式設計師濟癲
 * @since 2023-12-21 09:43
 */
@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註解引數說明:

  • retryFor:此引數包含的異常會觸發重試機制,多個異常則以陣列形式定義。
  • noRetryFor:此引數包含的異常不會觸發重試機制,多個異常則以陣列形式定義。
  • maxAttempts:重試最大次數,不定義則預設3次。
  • backoff:定義補償機制,delay-延遲時間(s),multiplier-重試時間的倍數(比如設定為2,重試4次的話,補償機制就是分別間隔2s、4s、8s做重試)

@Recover註解說明:用於兜底,當 超出了最大重試次數丟擲了不進行重試的異常 時,直接執行該註解宣告的兜底方法。

PS:順便提一句,如果是 SpringBoot2.x 的版本,這裡@Retryable註解的retryFor引數對應的是includenoRetryFor引數對應的是exclude,可以直接點進去看原始碼便一目瞭然。


5、JunitTest測試

我們編寫一個Junit測試類來測試重試的效果,並列印出結果資訊。

import com.example.springretrydemo.retry.RetryComponent;
import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

@Slf4j
@SpringBootTest
class SpringRetryDemoApplicationTests {

   @Autowired
   private RetryComponent retryComponent;

   @Test
   void sendSmsTest() {
      boolean ret = retryComponent.sendSmsRetry();
      log.info("sendSmsTest result = {}", ret);
   }

}

6、效果

第1次測試時,可以看到,亂數剛好都是1,走的是陣列越界異常。

而這個異常在retryFor中定義了,所以執行了4次,直到結束,最後進入了兜底方法。

同時,可以看到,執行4次的頻率也和預想一樣是2s、4s、8s。

第2次測試時,可以看到,亂數是3,走的是空指標異常。

而這個異常在noRetryFor中定義了,所以接下來直接進入了兜底方法。

第3次測試時,可以看到,第一次亂數是0,走的引數異常,在retryFor中,所以2s後繼續重試。

然後亂數是2,表示業務成功,所以直接返回了true。

這個場景就很像大家經常遇見的補償操作,第一次發生異常失敗,第二次重試後又成功了。

總結

Spring-retry框架還是挺實用的,但不是萬能的。

優點是,簡化了重試邏輯,提供了現成的重試策略,具備一定靈活性。

缺點,也很明顯,生產環境使用有風險,比如在複雜場景下設定的策略有問題,有可能會導致無限重試,這個後果不用說大家也能想象。

所以,使用這個框架,一定要明確好場景再使用,我這裡不推薦複雜場景下使用,因為君子不立於危牆之下

好了,今天的知識點你學會了嗎?

完整程式碼:戳這裡 --> Gitee


喜歡請點贊+關注↑↑↑,持續分享乾貨哦~