詳解如何在SpringBoot中優(yōu)雅地重試調(diào)用第三方API

更新時間：2022年12月16日 13:57:31 作者：JAVA旭陽

作為后端程序員，我們的日常工作就是調(diào)用一些第三方服務，將數(shù)據(jù)存入數(shù)據(jù)庫，返回信息給前端。本文為大家介紹了如何在SpringBoot中優(yōu)雅地重試調(diào)用第三方API，需要的可以參考一下

前言

作為后端程序員，我們的日常工作就是調(diào)用一些第三方服務，將數(shù)據(jù)存入數(shù)據(jù)庫，返回信息給前端。但你不能保證所有的事情一直都很順利。像有些第三方API，偶爾會出現(xiàn)超時。此時，我們要重試幾次，這取決于你的重試策略。

下面舉一個我在日常開發(fā)中多次看到的例子：

public interface OutSource {
    List<Integer> getResult() throws TimeOutException;
}

@Service
public class OutSourceImpl implements OutSource {

    static Random random = new Random();
    @Override
    public List<Integer> getResult() {
        //mock failure
        if (random.nextInt(2) == 1)
            throw new TimeOutException();
        return List.of(1, 2, 3);
    }
}


@Slf4j
@Service
public class ManuallyRetryService {

    @Autowired
    private OutSource outSource;

    public List<Integer> getOutSourceResult(String data, int retryTimes) {
        log.info("trigger time:{}", retryTimes);

        if (retryTimes > 3) {
            return List.of();
        }

        try {
            List<Integer> lst = outSource.getResult();
            if (!CollectionUtils.isEmpty(lst)) {
                return lst;
            }

            log.error("getOutSourceResult error, data:{}", data);
        } catch (TimeOutException e) {
            log.error("getOutSourceResult timeout", e);
        }
        // 遞歸調(diào)用
        return getOutSourceResult(data, retryTimes + 1);
    }

}

@Slf4j
@RestController
public class RetryTestController {

    @Autowired
    private ManuallyRetryService manuallyRetryService;
    
    @GetMapping("manually")
    public String manuallyRetry() {
        List<Integer> result = manuallyRetryService.getOutSourceResult("haha", 0);
        if (!CollectionUtils.isEmpty(result)) {
            return "ok";
        }
        return "fail";
    }
}

看看上面這段代碼，我認為它可以正常工作，當retryTimes達到4時，無論如何我們都會得到最終結果。但是你覺得寫的好嗎？優(yōu)雅嗎？下面我來介紹Spring中的一個組件：spring-retry，我們不妨來試一試。

Spring-Retry介紹使用

spring-retry是Spring中的提供的一個重試框架，提供了注解的方式，在不入侵原有業(yè)務邏輯代碼的方式下，優(yōu)雅的實現(xiàn)重處理功能。

安裝依賴

如果你的是gradle應用，引入下面的依賴

implementation 'org.springframework.boot:spring-boot-starter-aop''org.springframework.boot:spring-boot-starter-aop'
implementation 'org.springframework.retry:spring-retry'

如果你的項目使用的是maven項目，引入下面的依賴

<dependency>
    <groupId>org.springframework.retry</groupId>
    <artifactId>spring-retry</artifactId>
</dependency>

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-aop</artifactId>
</dependency>

啟用重試功能

添加@EnableRetry注解在入口的類上從而啟用功能。

@SpringBootApplication
//看過來
@EnableRetry
public class TestSpringApplication {

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

}

應用

我們以前面的為例，看看怎么使用，如下面的代碼：

public interface OutSource {
    List<Integer> getResult() throws TimeOutException;
}

@Service
public class OutSourceImpl implements OutSource {

    static Random random = new Random();
    @Override
    public List<Integer> getResult() {
        //mock failure will throw an exception every time
        throw new TimeOutException();
    }
}

@Slf4j
@Service
public class RetryableService {

    @Autowired
    private OutSource outSource;

    // 看這里
    @Retryable(value = {TimeOutException.class}, maxAttempts = 3)
    public List<Integer> getOutSourceResult(String data) {
        log.info("trigger timestamp:{}", System.currentTimeMillis() / 1000);

        List<Integer> lst = outSource.getResult();
        if (!CollectionUtils.isEmpty(lst)) {
            return lst;
        }
        log.error("getOutSourceResult error, data:{}", data);

        return null;
    }

}


@Slf4j
@RestController
public class RetryTestController {

    @Autowired
    private RetryableService retryableService;

    @GetMapping("retryable")
    public String manuallyRetry2() {
        try {
            List<Integer> result = retryableService.getOutSourceResult("aaaa");
            if (!CollectionUtils.isEmpty(result)) {
                return "ok";
            }
        } catch (Exception e) {
            log.error("retryable final exception", e);
        }
        return "fail";
    }

}

關鍵在于Service層中的實現(xiàn)類中添加了 @Retryable注解，實現(xiàn)了重試, 指定value是TimeOutException異常會進行重試，最大重試maxAttempts3次。

驗證

這一次，當我們訪問http://localhost:8080/retryable時，我們將看到瀏覽器上的結果失敗。然后在你的終端上看到：

INFO 66776 --- [nio-9997-exec-1] c.m.testspring.service.RetryableService : trigger timestamp:1668236840
INFO 66776 --- [nio-9997-exec-1] c.m.testspring.service.RetryableService : trigger timestamp:1668236841
INFO 66776 --- [nio-9997-exec-1] c.m.testspring.service.RetryableService : trigger timestamp:1668236842
ERROR 66776 --- [nio-9997-exec-1] c.m.t.controller.RetryTestController : retryable final exception