我正在用Java编写一个负载测试应用程序,并且有一个线程池可以对正在测试的服务器执行任务.因此,要创建1000个作业并在5个线程中运行它,我会执行以下操作：

    ExecutorService pool = Executors.newFixedThreadPool(5);
    List<Runnable> jobs = makeJobs(1000);
    for(Runnable job : jobs){
        pool.execute(job);
    }

但是我并不认为这种方法可以很好地扩展,因为我必须提前制作所有“工作”对象并将它们放在内存中直到需要它们为止.

我正在寻找一种方法让池中的线程在每次需要新工作时都会进入某种“JobFactory”类,并且工厂可以根据请求构建Runnables,直到运行所需的工作数.工厂可能会开始返回’null’来向线程发出信号,表示没有更多的工作要做.

我可以手动编写类似这样的东西,但它似乎是一个常见的用例,并且想知道在我可以使用的奇妙但复杂的“java.util.concurrent”包中是否有任何东西？

解决方法:

您可以使用AtomicInteger在线程池的执行线程中完成所有工作,以监视执行的可运行数量

 int numberOfParties = 5;
 AtomicInteger numberOfJobsToExecute = new AtomicInteger(1000);
 ExecutorService pool = Executors.newFixedThreadPool(numberOfParties );
 for(int i =0; i < numberOfParties; i++){
     pool.submit(new Runnable(){
        public void run(){
            while(numberOfJobsToExecute.decrementAndGet() >= 0){
                makeJobs(1).get(0).run();
            }
        }
     });
 }

您还可以将返回的Future存储在List中,并将get()存储在它们中以等待完成(以及其他机制)