如何在Java中设置线程限制

2022-09-03 04:46:03

假设我有1000个文件要读取,并且由于某些限制,我想并行读取最多5个文件。而且,一旦其中一个完成,我想要一个新的开始。

我有一个主函数,它有文件列表,每当一个线程完成时,我都会尝试更改计数器。但它不起作用!

有什么建议吗?

以下是主函数循环

for (final File filename : folder.listFiles()) {

    Object lock1 = new Object();
    new myThread(filename, lock1).start();
    counter++;
    while (counter > 5);
}

答案 1

像这样生成线程不是要走的路。使用 和 将池指定为 5。将所有文件放在类似于或其他线程安全集合之类的东西中,所有正在执行的文件都可以随意使用它。ExecutorServiceBlockingQueuepoll()

public class ThreadReader {

    public static void main(String[] args) {
        File f = null;//folder
        final BlockingQueue<File> queue = new ArrayBlockingQueue<File>(1000);
        for(File kid : f.listFiles()){
            queue.add(kid);
        }

        ExecutorService pool = Executors.newFixedThreadPool(5);

        for(int i = 1; i <= 5; i++){
            Runnable r = new Runnable(){
                public void run() {
                    File workFile = null;
                    while((workFile = queue.poll()) != null){
                        //work on the file.
                    }
                }
            };
            pool.execute(r);
        }
    }
}

答案 2

您可以将执行器服务用作线程池和队列。

ExecutorService pool = Executors.newFixedThreadPool(5);
File f = new File(args[0]);

for (final File kid : f.listFiles()) {
    pool.execute(new Runnable() {
        @Override
        public void run() {
            process(kid);
        }
    });
}
pool.shutdown();
// wait for them to finish for up to one minute.
pool.awaitTermination(1, TimeUnit.MINUTES);