add AsyncRetryExecutor
This commit is contained in:
148
src/main/java/com/imyeyu/java/thread/AsyncRetryExecutor.java
Normal file
148
src/main/java/com/imyeyu/java/thread/AsyncRetryExecutor.java
Normal file
@ -0,0 +1,148 @@
|
|||||||
|
package com.imyeyu.java.thread;
|
||||||
|
|
||||||
|
import com.imyeyu.java.bean.Callback;
|
||||||
|
import com.imyeyu.java.bean.CallbackArg;
|
||||||
|
|
||||||
|
import java.util.concurrent.TimeUnit;
|
||||||
|
|
||||||
|
/**
|
||||||
|
*
|
||||||
|
* @author 夜雨
|
||||||
|
* @since 2025-11-06 17:45
|
||||||
|
*/
|
||||||
|
public class AsyncRetryExecutor {
|
||||||
|
|
||||||
|
/** 默认重试次数,-1 表无限 */
|
||||||
|
private static final int DEFAULT_MAX_RETRY = -1;
|
||||||
|
|
||||||
|
/** 默认重试间隔(毫秒) */
|
||||||
|
private static final long DEFAULT_RETRY_INTERVAL = -1;
|
||||||
|
|
||||||
|
/** 线程名前缀 */
|
||||||
|
private static final String DEFAULT_THREAD_NAME_PREFIX = "AsyncRetryThread-";
|
||||||
|
|
||||||
|
/** true 为守护进程 */
|
||||||
|
private static final boolean DEFAULT_DAEMON = false;
|
||||||
|
|
||||||
|
private final Thread thread;
|
||||||
|
private volatile boolean isSuccess = false;
|
||||||
|
private volatile boolean isRunning = false;
|
||||||
|
|
||||||
|
private AsyncRetryExecutor(Builder builder) {
|
||||||
|
thread = new Thread(() -> {
|
||||||
|
int retryCount = 0;
|
||||||
|
while (isRunning && !isSuccess && (builder.maxRetry < 0 || retryCount <= builder.maxRetry)) {
|
||||||
|
try {
|
||||||
|
builder.callback.handler();
|
||||||
|
isSuccess = true;
|
||||||
|
} catch (Exception e) {
|
||||||
|
retryCount++;
|
||||||
|
if (!isRunning) {
|
||||||
|
// 中断
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if (0 < builder.maxRetry && builder.maxRetry < retryCount) {
|
||||||
|
// 超过重试次数
|
||||||
|
builder.onRetryExhausted.handler(e);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
// 重试
|
||||||
|
try {
|
||||||
|
TimeUnit.MILLISECONDS.sleep(builder.retryInterval);
|
||||||
|
} catch (InterruptedException ex) {
|
||||||
|
Thread.currentThread().interrupt();
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}, builder.threadNamePrefix + Thread.currentThread().threadId());
|
||||||
|
thread.setDaemon(builder.daemon);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 构造器
|
||||||
|
*
|
||||||
|
* @author 夜雨
|
||||||
|
* @since 2025-11-06 23:37
|
||||||
|
*/
|
||||||
|
public static class Builder {
|
||||||
|
|
||||||
|
// 必需参数
|
||||||
|
private final Callback callback;
|
||||||
|
|
||||||
|
// 可选参数(带默认值)
|
||||||
|
private int maxRetry = DEFAULT_MAX_RETRY;
|
||||||
|
private long retryInterval = DEFAULT_RETRY_INTERVAL;
|
||||||
|
private String threadNamePrefix = DEFAULT_THREAD_NAME_PREFIX;
|
||||||
|
private boolean daemon = DEFAULT_DAEMON;
|
||||||
|
private CallbackArg<Exception> onRetryExhausted;
|
||||||
|
|
||||||
|
public Builder(Callback callback) {
|
||||||
|
this.callback = callback;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder maxRetry(int maxRetry) {
|
||||||
|
this.maxRetry = maxRetry;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder retryInterval(long retryInterval) {
|
||||||
|
this.retryInterval = retryInterval;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder threadNamePrefix(String prefix) {
|
||||||
|
this.threadNamePrefix = prefix;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder daemon(boolean daemon) {
|
||||||
|
this.daemon = daemon;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Builder onRetryExhausted(CallbackArg<Exception> onRetryExhausted) {
|
||||||
|
this.onRetryExhausted = onRetryExhausted;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public AsyncRetryExecutor build() {
|
||||||
|
return new AsyncRetryExecutor(this);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public static AsyncRetryExecutor create(Callback callback) {
|
||||||
|
return new Builder(callback).build();
|
||||||
|
}
|
||||||
|
|
||||||
|
public static AsyncRetryExecutor create(Callback callback, int maxRetry) {
|
||||||
|
return new Builder(callback).maxRetry(maxRetry).build();
|
||||||
|
}
|
||||||
|
|
||||||
|
public static AsyncRetryExecutor create(Callback callback, long interval) {
|
||||||
|
return new Builder(callback).retryInterval(interval).build();
|
||||||
|
}
|
||||||
|
|
||||||
|
public void start() {
|
||||||
|
if (!isRunning) {
|
||||||
|
isRunning = true;
|
||||||
|
isSuccess = false;
|
||||||
|
thread.start();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void interrupt() {
|
||||||
|
isRunning = false;
|
||||||
|
if (thread.isAlive() && !thread.isInterrupted()) {
|
||||||
|
thread.interrupt();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public boolean isSuccess() {
|
||||||
|
return isSuccess;
|
||||||
|
}
|
||||||
|
|
||||||
|
public boolean isRunning() {
|
||||||
|
return isRunning;
|
||||||
|
}
|
||||||
|
}
|
||||||
35
src/test/java/test/TestThread.java
Normal file
35
src/test/java/test/TestThread.java
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
package test;
|
||||||
|
|
||||||
|
import com.imyeyu.java.thread.AsyncRetryExecutor;
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author 夜雨
|
||||||
|
* @since 2025-11-06 23:59
|
||||||
|
*/
|
||||||
|
public class TestThread {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testAsyncRetryExecutor() throws InterruptedException {
|
||||||
|
Object lock = new Object();
|
||||||
|
AsyncRetryExecutor executor = AsyncRetryExecutor.create(() -> {
|
||||||
|
double value = Math.random();
|
||||||
|
if (.9 < value) {
|
||||||
|
System.out.println("success " + value);
|
||||||
|
synchronized (lock) {
|
||||||
|
lock.notifyAll();
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
System.out.println("fail " + value);
|
||||||
|
throw new RuntimeException("fail");
|
||||||
|
}
|
||||||
|
});
|
||||||
|
executor.start();
|
||||||
|
|
||||||
|
synchronized (lock) {
|
||||||
|
lock.wait();
|
||||||
|
}
|
||||||
|
|
||||||
|
assert executor.isSuccess();
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user