feat: 新增可计数的拒绝策略.

This commit is contained in:
2025-03-05 14:50:36 +08:00
parent f7e4400e37
commit af9c1c9622
6 changed files with 83 additions and 9 deletions

View File

@@ -0,0 +1,40 @@
package com.serliunx.statemanagement.support;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.atomic.AtomicLong;
/**
* 附带计数的拒绝策略默认实现(丢弃任务并计数)
*
* @author <a href="mailto:serliunx@yeah.net">SerLiunx</a>
* @version 1.0.0
* @since 2025/3/5
*/
public class DefaultCountableRejectedExecutionHandler implements CountableRejectedExecutionHandler {
/**
* 计数器
*/
private final AtomicLong counter = new AtomicLong(0);
/**
* 最后一次被拒绝的任务
*/
private volatile Runnable last = null;
@Override
public long getCount() {
return counter.get();
}
@Override
public Runnable getLastRejectedTask() {
return last;
}
@Override
public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
last = r;
counter.incrementAndGet();
}
}