一个高性能、支持百万级多任务重试框架：Fast-Retry

2024-06-02码农

点击「 IT码徒」，关注，置顶公众号

每日技术干货，第一时间送达！

前言

假设你的系统里有100万个用户，然后你要轮询重试的获取每个用户的身份信息, 如果你还在使用SpringRetry和GuavaRetry 之类的这种单任务的同步重试框架，那你可能到猴年马月也处理不完，即使加再多的机器和线程也是杯水车薪，而Fast-Retry正是为这种场景而生。

Fast-Retry

一个高性能的多任务重试框架，支持百万级任务的异步重试、以及支持编程式和注解声明式等多种使用方式、也支持自定义结果重试逻辑。

What is this?

与主流的Spring-Retry、Guava-Retry等单任务同步重试框架不同，Fast-Retry是一个支持异步重试框架，支持异步任务的重试、超时等待、回调。

Spring-Retry、Guava-Retry均无法支持大批量任务的重试，即使加入线程池也无法解决，因为实际每个重试任务都是单独的同步逻辑，然后会会占用过多线程资源导致大量任务在等待处理，随着任务数的增加，系统吞吐量大大降低，性能指数级降低，而Fast-Retry在异步重试下的性能是前者的指数倍。

下图是三者的性能对比

测试线程池: 8个固定线程

单个任务逻辑: 轮询5次，隔2秒重试一次，总耗时10秒

未测预计公式：当我们使用线程池的时候，一般线程池中总任务处理耗时 = 任务数/并发度 x 单个任务重试耗时

可以看到即使是处理100万个任务，Fast-Retry的性能也比Spring-Retry和Guava-Retry处理在50个任务时的性能还要快的多的多属实降维打击，这么快的秘密在于除了是异步，重要的是当别人在重试间隔里休息的时候，Fast-Retry还在不停忙命的工作着。

即使抛开性能不谈， SpringRetry 使用繁琐，不支持根据结果的进行重试， GuavaRetry 虽然支持，但是又没有提供注解声明式的使用。

快速开始

引入依赖

<dependency> <groupId>io.github.burukeyou</groupId> <artifactId>fast-retry-all</artifactId> <version>0.2.0</version> </dependency>

有以下三种方式去构建我们的重试任务

使用重试队列

RetryTask 就是可以配置我们重试任务的一些逻辑，比如怎么重试，怎么获取重试结果，隔多久后重试，在什么情况下重试。它可以帮助我们更加自由的去构建重试任务的逻辑。但如果只是简单使用，强烈建议使用 FastRetryBuilder 或者 @FastRetry 注解

RetryQueue 就是一个执行和调度我们重试任务的核心角色，其在使用上与线程池的API方法基本一致

ExecutorService executorService = Executors.newFixedThreadPool(8); RetryQueue queue = new FastRetryQueue(executorService); RetryTask<String> task = new RetryTask<String>() { int result = 0 ; // 下一次重试的间隔 @Override publiclongwaitRetryTime(){ return2000; } // 执行重试，每次重试回调此方法 @Override publicbooleanretry(){ return ++result < 5; } // 获取重试结果 @Override public String getResult(){ return result + ""; } }; CompletableFuture<String> future = queue.submit(task); log.info("任务结束结果:{}",future.get());

使用FastRetryBuilder

底层还是使用的 RetryQueue 去处理，只是帮我们简化了构建 RetryTask 的逻辑

RetryResultPolicy<String> resultPolicy = result -> result.equals("444"); FastRetryer<String> retryer = FastRetryBuilder.<String>builder() .attemptMaxTimes(3) .waitRetryTime(3, TimeUnit.SECONDS) .retryIfException(true) .retryIfExceptionOfType(TimeoutException. class) .exceptionRecover(true) .resultPolicy(resultPolicy) .build(); CompletableFuture<String> future = retryer.submit(() -> { log.info("重试"); //throw new Exception("test"); //int i = 1/0; if (0 < 10){ thrownew TimeoutException("test"); } return"444"; }); String o = future.get(); log.info("结果{}", o);

使用@FastRetry注解

底层还是使用的 RetryQueue 去处理，只是帮我们简化了构建 RetryTask 的逻辑，并且与Spring进行整合能对Spring的bean标记了 FastRetry 注解的方法进行代理，提供了重试任务注解声明式的使用方式

依赖Spring环境，所以需要在Spring配置类加上 @EnableFastRetry 注解启用配置，这个 @FastRetry 注解的使用才会生效

如果将结果类型使用 CompletableFuture 包装，自动进行异步轮询返回，否则同步阻塞等待重试结果。(推荐)

下面定义等价于 RetryQueue.execute 方法

// 如果发生异常，每隔两秒重试一次 @FastRetry(retryWait = @RetryWait(delay = 2)) public String retryTask(){ return"success"; }

下面定义等价于 RetryQueue.submit 方法，支持异步轮询

@FastRetry(retryWait = @RetryWait(delay = 2)) public CompletableFuture<String> retryTask(){ return CompletableFuture.completedFuture("success"); }

自定义重试注解

如果不喜欢或者需要更加通用化的贴近业务的重试注解，提供一些默认的参数和处理逻辑，可以自行定义一个重试注解并标记上 @FastRetry 并指定factory，然后实现 AnnotationRetryTaskFactory 接口实现自己的构建重试任务的逻辑即可。@FastRetry默认实现就是： FastRetryAnnotationRetryTaskFactory