我可以访问为线程池线程提交任务（并且正在运行）的线程吗？

Question

我想在MyThreadPoolExecutor 中覆盖ThreadPoolExecutor 中的protected void beforeExecute(Thread t, Runnable r) { }，我想知道哪个线程提交了这个（t）线程要运行的任务？我想将任务提交线程的 threadLocals 复制到要运行此任务的线程（t）（在 threadPool 中）。

Answer 1

我相信你可以在你的 runnable 中做这样的事情来获取它正在执行的线程 id。

private class MyTask implements Runnable {
public void run() {
    long threadId = Thread.currentThread().getId();
    logger.debug("Thread # " + threadId + " is doing this task");
}

如果您想尝试让线程池在 ThreadLocal 中执行更棘手的线程，最好的方法是在可运行类中访问 ThreadLocal。

我也在一个网站上阅读了这篇文章，并认为您可能需要重新考虑使用本地线程的自定义线程池的方法。

如果我们想使用 ExecutorService 并向其提交 Runnable，使用 ThreadLocal 将产生不确定的结果——因为我们不能保证给定 userId 的每个 Runnable 操作每次都会由同一个线程处理它被执行了。

因此，我们的 ThreadLocal 将在不同的 userId 之间共享。这就是为什么我们不应该将 TheadLocal 与 ExecutorService 一起使用。只有当我们完全控制哪个线程将选择执行哪个可运行操作时才应该使用它。

https://www.baeldung.com/java-threadlocal

Answer 2

我成功了

public class MyThreadPoolExecutor extends ThreadPoolExecutor {

    public MyThreadPoolExecutor(int corePoolSize, int maximumPoolSize, long keepAliveTime, TimeUnit unit,
            BlockingQueue<Runnable> workQueue, ThreadFactory threadFactory) {
        super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue, threadFactory);
    }

    @Override
    public Future<?> submit(Runnable task) {
    // get the threadLocal from current thread which is task submitting thread  
        boolean threadLocalValueFromParentThread = ThreadLocalsUtils.getThreadLocalValueForThisThread.get();

        Runnable wrappedTask = () -> {
            try {
                // set value of task submitting thread's value in whichever thread gets this runnable.
                ThreadLocalsUtils.setThreadLocalValueForThisThread(threadLocalValueFromParentThread);
                task.run();
            } finally {
                ThreadLocalsUtils.setThreadLocalValueForThisThread(false);
            }
        };
        return super.submit(wrappedTask);
    }

}