Java线程池RejectedExecutionHandler传入的runnable对象是什么

Question

我有这样的thread pool with a RejectedExecutionHandler 和一个 Runnable 任务。

sExecutorService = new ThreadPoolExecutor( ...    new MyRejectionHandler() );

interface MyTask extends Runnable { ... }

class MyTaskImpl implements MyTask { ...}     

我执行这样的任务

   sExecutorService.submit(myTask);        

在被拒绝的执行案例中，我希望获得被拒绝的 Runnable（即 MyTask）并在其中设置一些字段，将其标记为被拒绝。但我无法将其投射到 MyTask。 那么传递给rejectExecution 的runnable 到底是什么？我提交的似乎不是 MyTask 。以及如何在 RejectedExecutionHandler 中获取被拒绝的任务。

public class MyRejectionHandler implements RejectedExecutionHandler{  
   public void rejectedExecution(Runnable runnable, ThreadPoolExecutor executor)  {
    MyTask myTask = (MyTask) runnable; // exception
    myTask.doSomething();
   }
java.lang.ClassCastException: java.util.concurrent.FutureTask cannot be cast to MyTask       

Answer 1

问题是当您使用提交方法 TreadPoolExecutor（实际上是 AbstractExecutorService）提交任务时将其包装到 FutureTask。之后，您收到 FutureTask 而不是您的 Runnable。你可以调用执行不提交：

sExecutorService.execute(yourTask);

我认为没有办法从 FutureTask 获取您的任务。您只能为它调用运行。因此，如果您想调用 submit 并且需要调用 run - 只是不要转换为 MyTask：

FutureTask myTask = (FutureTask) runnable;
myTask.run();
Object result = myTask.get();

如果你想访问你的 MyTask 对象，可以创建 MyFutureTask 扩展 FutureTask，这将允许获取你的对象：

public MyFutureTask<V> extends FutureTask<V> {

    private Runnable myTask;

    public MyFutureTask(Runnable runnable, V result) {
        super(runnable, result);
        this.myTask = runnable;
    }

    public Runnable getMyTask() {
        return myTask;
    }        
}

您还需要扩展 ThreadPoolExecutor 并重新定义负责 Runnable 到 FutureTask 包装的 newTaskFor 方法：

public class MyThreadPoolExecutor extends ThreadPoolExecutor {

    @Override
    protected <T> RunnableFuture<T> newTaskFor(Runnable runnable, T value) {
        return new MyFutureTask(task, value);            
    }
}

之后，您可以将 MyThreadPoolExecutor 用作 ThreadPoolExecutor 并拒绝任务处理：

MyFutureTask myFutureTask = (MyFutureTask) runnable;
MyTask myTask = (MyTask) myFutureTask.getMyTask();
// Now you can do what you want with you task:
myTask.doSomthing();