如何在Java中使用多线程进行IO密集型任务

原创淩亂°似流年 2024-10-14 10:33 48阅读 0赞

在Java中，如果你面对的是IO密集型任务（即大量的读写操作），可以考虑使用多线程来提高程序的效率。以下是一个基本的例子：

```java
import java.io.BufferedReader;
import java.io.FileReader;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

public class IOIntensiveTask {

private static final int NUM_THREADS = 5; // 根据实际情况调整

public static void main(String[] args) {

// 定义需要读取的文件
        String filePath = "input.txt"; // 替换为实际文件路径

// 创建线程池
        ExecutorService executorService = Executors.newFixedThreadPool(NUM_THREADS);

// 开始读取并处理文件内容
        for (int i = 0; i < NUM_THREADS; i++) {
            int threadIndex = i;
            executorService.submit(() -> {
                try (BufferedReader br = new BufferedReader(new FileReader(filePath)))) {
                    String line;
                    while ((line = br.readLine()) != null) {
                        // 对每行进行处理（这里只是打印）
                        System.out.println("Thread " + threadIndex + ": " + line);
                    }
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }));
        }

// 关闭线程池
        executorService.shutdown();

// 等待所有任务完成
        while (!executorService.isTerminated())) {
            try {
                Thread.sleep(100); // 每隔一定时间检查一次
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }

System.out.println("All threads finished.");
    }
}
```

这个例子中，我们创建了一个固定大小的线程池，并将IO密集型任务分配给每个线程。当所有线程完成任务后，程序会打印一条消息。