使用ThreadPoolExecutor的活动任务数

Question

我正在使用ThreadPoolExecutor在我的Java应用程序中执行任务。我有一个要求，我希望在执行程序队列中的任意时间点获取队列中活动任务的数量。我抬头看看ThreadPoolExecutor的javadoc，发现了两个相关的方法：getTaskCount()和getCompletedTaskCount()。

根据文档，我可以分别从上述两种方法中获取计划任务的数量和完成的任务。但是我无法找到解决方案，以便在任何时间点获取队列中活动任务的数量。我可以这样做：

getTaskCount() = getCompletedTaskCount() + failed tasks + active tasks 

但是失败的任务数量不能直接用于达到预期的计算。

我在这里错过了什么吗？

Answer 1

我不认为你需要知道失败的计数与你试图使用的计算。

long submitted = executor.getTaskCount(); 
long completed = executor.getCompletedTaskCount(); 
long notCompleted = submitted - completed; // approximate 

会（近似）足够。

---

或者，你可以使用getQueue()与size()：

int queued = executor.getQueue().size(); 
int active = executor.getActiveCount(); 
int notCompleted = queued + active; // approximate 

这个答案假定你正在寻找一个 “尚未完成” 计数。你的问题自相矛盾，所以我不完全确定你在问什么。回复我对你的问题的评论，如果这是不正确的，我会相应地更新这个答案。

Answer 2

您是否尝试过使用beforeExecute和afterExecute方法？这些在执行任务之前和之后调用。 after执行方法甚至提供了一个throwable作为第二个参数，所以你知道任务何时失败。

您可以添加一个钩子，以便beforeExecute递增活动任务的值，afterExecute递减它。当然，这些方法在它们各自的字段上被调用，所以你必须在一个相互锁定的对象上同步结果。

要使用这些方法，只需重写您选择的ThreadPoolExecutor对象并在其中添加钩子。

例如，下面的代码应该有希望的工作：

public class MyExecutor extends ThreadPoolExecutor { 
     //Lock object used for synchronization 
     private final Object lockObject = new Object(); 
     //Contains the active task count 
     private int activeTaskCount = 0; 
     //Failed task count 
     private int failedTaskCount = 0; 
     private int succeededTaskCount = 0; 

     public MyExecutor() { 
      //call super here with your parameters; 
     } 

     public int getActiveTaskCount(){ 
      synchronized(lockObject){ 
       return activeTaskCount; 
      } 
     } 

     public int getFailedTaskCount(){ 
      synchronized(lockObject){ 
       return failedTaskCount ; 
      } 
     } 

     public int getSucceededTaskCount(){ 
      synchronized(lockObject){ 
       return succeededTaskCount ; 
      } 
     } 

     protected void beforeExecute(Thread t, 
          Runnable r){ 
      super.beforeExecute(t,r); 
      synchronized(lockObject){ 
       activeTaskCount++; 
      } 
     } 

     protected void afterExecute(Runnable r,Throwable t){ 
      super.afterExecute(r,t); 
      synchronized(lockObject){ 
       activeTaskCount--; 
       if(t!=null){ 
        failedTaskCount++; 
       }else{ 
        succeededTaskCount++; 
       } 
      } 
     } 

}