时间戳

首页 > 解决方案 > 线程的 ThreadLocals 清理

java - 线程的 ThreadLocals 清理

问题描述

在这里，我找到了以下代码如何在 Java 中清理 Thread 的 ThreadLocals：

private void cleanThreadLocals() {
    try {
        // Get a reference to the thread locals table of the current thread
        Thread thread = Thread.currentThread();
        Field threadLocalsField = Thread.class.getDeclaredField("threadLocals");
        threadLocalsField.setAccessible(true);
        Object threadLocalTable = threadLocalsField.get(thread);

        // Get a reference to the array holding the thread local variables inside the
        // ThreadLocalMap of the current thread
        Class threadLocalMapClass = Class.forName("java.lang.ThreadLocal$ThreadLocalMap");
        Field tableField = threadLocalMapClass.getDeclaredField("table");
        tableField.setAccessible(true);
        Object table = tableField.get(threadLocalTable);

        // The key to the ThreadLocalMap is a WeakReference object. The referent field of this object
        // is a reference to the actual ThreadLocal variable
        Field referentField = Reference.class.getDeclaredField("referent");
        referentField.setAccessible(true);

        for (int i=0; i < Array.getLength(table); i++) {
            // Each entry in the table array of ThreadLocalMap is an Entry object
            // representing the thread local reference and its value
            Object entry = Array.get(table, i);
            if (entry != null) {
                // Get a reference to the thread local object and remove it from the table
                ThreadLocal threadLocal = (ThreadLocal)referentField.get(entry);
                threadLocal.remove();
            }
        }
    } catch(Exception e) {
        // We will tolerate an exception here and just log it
        throw new IllegalStateException(e);
    }
}

这很复杂。遵循更简单的代码怎么样，清理 ThreadLocals 就足够了吗？谢谢你。

private void cleanThreadLocals(Thread thread) {
    try {
        // Get a reference to the thread locals table of the current thread
        Thread thread = Thread.currentThread();
        Field threadLocalsField = Thread.class.getDeclaredField("threadLocals");
        threadLocalsField.setAccessible(true);
        threadLocalsField.set(thread, null);
    } catch (Exception e) {
        throw new RuntimeException(e);
    }
}

标签： javamultithreadingreflectionconcurrencythread-local

解决方案

这两个片段都清理了 ThreadLocals，但它们有不同的方法。

较长的代码片段首先从 ThreadLocalMap 获取所有线程 ThreadLocals 并在它们上调用 remove() 方法。您只能将它用于当前线程，因为 remove() 方法只是从当前线程中删除（）值，无法指定它应该使用哪个线程。通过一些修改，您可以将它用于任何线程，您必须直接在 ThreadLocalMap 上调用 remove()。

较短的代码段从 Thread 中删除了整个 ThreadLocalMap 实例。它是延迟初始化的，因此如果需要，它将再次创建。您可以将此方法用于任何线程。

我测试了是否所有实例都从 JVM 中删除。该测试依赖于 GC 行为，您可能需要在某些环境中调整 GC 行为，但在我的环境 win10/OracleJava8 上它开箱即用。

测试：

@Test
public void howToCleanThreadLocalValues() throws ReflectiveOperationException {
    Thread thread = Thread.currentThread();

    // Set thread local value for current thread
    WeakReference<ThreadLocal> threadLocal = new WeakReference<>(new ThreadLocal<>());
    threadLocal.get().set("foo");

    // Get ThreadLocalMap
    Field threadLocalsField = Thread.class.getDeclaredField("threadLocals");
    threadLocalsField.setAccessible(true);
    WeakReference<Object> threadLocalMap = new WeakReference<>(threadLocalsField.get(thread));
    Assert.assertNotNull(threadLocalMap.get());
    Assert.assertNotNull(threadLocal.get());

    // Set ThreadLocalMap to null, GC do the rest
    threadLocalsField.set(Thread.currentThread(), null);
    System.gc();
    Assert.assertNull(threadLocalMap.get());
    Assert.assertNull(threadLocal.get());
}

推荐阅读