Java将对象与Executor中的每个线程相关联

Question

我有一堆运行程序，我想通过线程池来执行。然而，每个运行程序也会将一些结果写入一些文件。所以现在，运行程序的界面很简单：

class MyRunnable implements Runnable {
    ...
    MyRunnable(BufferedWriter writer, Task t) {
       this.writer = writer;
       this.task = t;
    }

    public void run() {
       ...
       this.writer.write(SOME_DATA);
    }
}

然而，我想要的是将一个BufferedWriter（换句话说，一个输出文件）与Executor池中的每个线程相关联。但是，我正在调用。使用执行器服务执行功能，如下所示：

BufferedWriter[] writers = initialize to 20 BufferedWriters;
ExecutorService executor = Executors.newFixedThreadPool(20);
for (Task t : tasks) {
    MyRunnable runnable = new MyRunnable(WHAT SHOULD GO HERE?, t)
    executor.execute(runnable);
}

我不知道执行器将分配哪个线程来运行给定的任务，所以我不知道应该向runnable提供哪个BufferedWriter。如何确保ExecutorService管理的每个线程都与一个对象（在本例中为BufferedWriter）关联？

Answer 1

...我想要的是将一个BufferedWriter（换句话说，一个输出文件）与执行器池中的每个线程相关联...

@djechlin关于ThreadLocal的回答很好，但问题是，当线程完成运行最后一个任务时，您无法访问BufferedWriter来close（）它们。

另一个答案可以在这里看到：

具有持久工作实例的线程池

在这篇文章中，我建议创建你自己的任务的BlockingQueue，每个线程分叉一个任务，让这些线程从你的队列中获取任务。所以线程运行方法类似于：

private final BlockingQueue<MyRunnable> queue = new ArrayBlockingQueue<>();
// if you want to shutdown your threads with a boolean
private volatile boolean shutdown;
...

// threads running in the `ExecutorService` will be doing this run() method
public void run() {
    // this allows them to maintain state, in this case your writer
    BufferedWriter writer = ...;
    while (!shutdown && !Thread.currentThread.isInterrupted()) {
        // they get their tasks from your own queue
        MyRunnable runnable = queue.take();
        // if you are using a poison pill but you'll have to add X of them
        if (runnable == STOP_OBJECT) {
            break;
        }
        runnable.run();
    }
    writer.close();
}

在这里，告诉线程何时完成有点棘手。您可以向队列中添加“毒药丸”对象，但必须向队列中添加与正在运行的线程数量相同的对象。

Answer 2

有一个名为ThreadLocal的类。

例如。

ThreadLocal<Type> t = new ThreadLocal<>() {
    @Override protected Type initialValue() {
        return new Type(Thread.currentThread.getName());
}

每当新线程尝试访问t时，这将延迟初始化类型。

我上一次使用这个类只是在一个类中计算出某台机器最擅长运行多少线程。（答案通常是“内核的数量”，但我想确保虚拟内核驱动了这个数量，而不是物理内核）。所以我写了一个任务，其中每个线程只发送了一个AtomicInteger计数器。但是所有线程都在争夺一个计数器，这在处理线程争用时产生了大量开销，所以我创建了一个threadlocal计数器，这样线程就可以在不受其他线程干扰的情况下发送自己的计数器的垃圾邮件。

它的用例有些模糊，因为大多数好的多线程设计都会避免这种情况，但当然也有这样的时候。