java 超时中断_JavaExecutorService在超时后中断任务

我正在寻找可以提供超时的ExecutorService实现。如果提交到ExecutorService的任务花费的时间超过了超时时间，则这些任务将被中断。实现这样的野兽并不是一个困难的任务，但是我想知道是否有人知道现有的实现。

这是我根据以下一些讨论得出的。任何意见？

import java.util.List;

import java.util.concurrent.*;

public class TimeoutThreadPoolExecutor extends ThreadPoolExecutor {

private final long timeout;

private final TimeUnit timeoutUnit;

private final ScheduledExecutorService timeoutExecutor = Executors.newSingleThreadScheduledExecutor();

private final ConcurrentMap runningTasks = new ConcurrentHashMap();

public TimeoutThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue workQueue, long timeout, TimeUnit timeoutUnit) {

super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue);

this.timeout = timeout;

this.timeoutUnit = timeoutUnit;

}

public TimeoutThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue workQueue, ThreadFactory threadFactory, long timeout, TimeUnit timeoutUnit) {

super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue, threadFactory);

this.timeout = timeout;

this.timeoutUnit = timeoutUnit;

}

public TimeoutThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue workQueue, RejectedExecutionHandler handler, long timeout, TimeUnit timeoutUnit) {

super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue, handler);

this.timeout = timeout;

this.timeoutUnit = timeoutUnit;

}

public TimeoutThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit, BlockingQueue workQueue, ThreadFactory threadFactory, RejectedExecutionHandler handler, long timeout, TimeUnit timeoutUnit) {

super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue, threadFactory, handler);

this.timeout = timeout;

this.timeoutUnit = timeoutUnit;

}

@Override

public void shutdown() {

timeoutExecutor.shutdown();

super.shutdown();

}

@Override

public List shutdownNow() {

timeoutExecutor.shutdownNow();

return super.shutdownNow();

}

@Override

protected void beforeExecute(Thread t, Runnable r) {

if(timeout > 0) {

final ScheduledFuture> scheduled = timeoutExecutor.schedule(new TimeoutTask(t), timeout, timeoutUnit);

runningTasks.put(r, scheduled);

}

}

@Override

protected void afterExecute(Runnable r, Throwable t) {

ScheduledFuture timeoutTask = runningTasks.remove(r);

if(timeoutTask != null) {

timeoutTask.cancel(false);

}

}

class TimeoutTask implements Runnable {

private final Thread thread;

public TimeoutTask(Thread thread) {

this.thread = thread;

}

@Override

public void run() {

thread.interrupt();

}

}

}