多线程编程八-父子线程怎么共享数据(InheritableThreadLocal的使用)

本文深入探讨InheritableThreadLocal的工作原理,演示如何在子线程中访问父线程的数据,对比ThreadLocal的区别,并解析Java线程构造方法中变量传递的关键代码。
摘要由CSDN通过智能技术生成

前面文章中我们已经通过ThreadLocal来实现线程级别的数据共享。那么如果子线程想访问父线程的数据该怎么做呢,InheritableThreadLocal可以帮助我们实现

1 InheritableThreadLocal的使用

测试代码

private InheritableThreadLocal<Integer> inheritableThreadLocalData = new InheritableThreadLocal<>();

private ThreadLocal<Integer> threadLocalData = new ThreadLocal<>();

@Test
public void testInheritableThreadLocal() throws InterruptedException, ExecutionException {
    inheritableThreadLocalData.set(1);
    threadLocalData.set(1);
    System.out.println("current thread get InheritableThreadLocal data:" + inheritableThreadLocalData.get() +
            " get ThreadLocal data:" + threadLocalData.get());
    new Thread(() -> System.out.println("son thread get InheritableThreadLocal data:" + inheritableThreadLocalData.get() +
            " get ThreadLocal data:" + threadLocalData.get())).start();
}

输出
current thread get InheritableThreadLocal data:1 get ThreadLocal data:1
son thread get InheritableThreadLocal data:1 get ThreadLocal data:null
可以看到当前内是可以访问到ThreadLocal和InheritableThreadLocal的数据的
子线程只能访问InheritableThreadLocal的数据,不能访问ThreadLocal的数据

2 InheritableThreadLocal原理

InheritableThreadLocal继承了ThreadLocal类,get、set方法用的都是父类的方法,只是重写了getMap和createMap

ThreadLocalMap getMap(Thread t) {
   return t.inheritableThreadLocals;
}

void createMap(Thread t, T firstValue) {
    t.inheritableThreadLocals = new ThreadLocalMap(this, firstValue);
}

inheritableThreadLocals也是ThreadLocal.ThreadLocalMap对象,正常来说InheritableThreadLocal和ThreadLocal的功能完全一致的,那么为什么前者可以获取到父线程的值,后者不能呢?
我们看一下Thread的构造方法

public Thread(Runnable target) {
    init(null, target, "Thread-" + nextThreadNum(), 0);
}

直接跟到最下面

private void init(ThreadGroup g, Runnable target, String name,
                  long stackSize, AccessControlContext acc,
                  boolean inheritThreadLocals) {
    if (name == null) {
        throw new NullPointerException("name cannot be null");
    }

    this.name = name;

    Thread parent = currentThread();
    SecurityManager security = System.getSecurityManager();
    if (g == null) {
        /* Determine if it's an applet or not */

        /* If there is a security manager, ask the security manager
           what to do. */
        if (security != null) {
            g = security.getThreadGroup();
        }

        /* If the security doesn't have a strong opinion of the matter
           use the parent thread group. */
        if (g == null) {
            g = parent.getThreadGroup();
        }
    }

    /* checkAccess regardless of whether or not threadgroup is
       explicitly passed in. */
    g.checkAccess();

    /*
     * Do we have the required permissions?
     */
    if (security != null) {
        if (isCCLOverridden(getClass())) {
            security.checkPermission(SUBCLASS_IMPLEMENTATION_PERMISSION);
        }
    }

    g.addUnstarted();

    this.group = g;
    this.daemon = parent.isDaemon();
    this.priority = parent.getPriority();
    if (security == null || isCCLOverridden(parent.getClass()))
        this.contextClassLoader = parent.getContextClassLoader();
    else
        this.contextClassLoader = parent.contextClassLoader;
    this.inheritedAccessControlContext =
            acc != null ? acc : AccessController.getContext();
    this.target = target;
    setPriority(priority);
    if (inheritThreadLocals && parent.inheritableThreadLocals != null)
        this.inheritableThreadLocals =
            ThreadLocal.createInheritedMap(parent.inheritableThreadLocals);
    /* Stash the specified stack size in case the VM cares */
    this.stackSize = stackSize;

    /* Set thread ID */
    tid = nextThreadID();
}

关键就在这段代码

// parent就是父线程,也就是执行new Thread()的线程
if (inheritThreadLocals && parent.inheritableThreadLocals != null)
    this.inheritableThreadLocals =
        ThreadLocal.createInheritedMap(parent.inheritableThreadLocals);


这里会把父线程的inheritableThreadLocals赋值给子线程,从而实现了父子线程变量传递

### InheritableThreadLocal 和 TransmittableThreadLocal 的区别与应用场景 #### 一、基本概念 `InheritableThreadLocal` 是 `ThreadLocal` 的扩展类,允许父线程将其值传递给子线程。而 `TransmittableThreadLocal` 则是一种增强型的 `ThreadLocal` 实现,不仅能够处理父子线程之间的数据共享,还能够在异步调用链中保持上下文一致性。 #### 二、功能对比 1. **继承特性** - `InheritableThreadLocal` 提供了一种机制,使得当新线程被创建时,可以从其父线程复制特定的 `ThreadLocal` 值[^1]。 - 对于 `TransmittableThreadLocal` 而言,它不仅可以像 `InheritableThreadLocal` 那样在父子线程间传输数据,还可以进一步支持跨线程池的任务调度场景下的上下文传播[^4]。 2. **适用范围** - 如果仅需解决简单的父子线程间的变量共享题,则可以考虑使用 `InheritableThreadLocal`[^3]。 - 当涉及到复杂的异步编程模型(如基于线程池的任务分发),或者需要在线程切换过程中保留某些上下文信息时,推荐使用 `TransmittableThreadLocal`[^2]。 #### 三、具体实现方式 以下是两种工具的具体代码示例: ##### (1) 使用 `InheritableThreadLocal` ```java public class ExampleWithInheritableThreadLocal { private static final InheritableThreadLocal<String> inheritableThreadLocal = new InheritableThreadLocal<>(); public static void main(String[] args) throws InterruptedException { inheritableThreadLocal.set("Parent Value"); Thread childThread = new Thread(() -> System.out.println(inheritableThreadLocal.get())); childThread.start(); childThread.join(); // Wait for the thread to finish. } } ``` 上述程序展示了如何利用 `InheritableThreadLocal` 将主线程中的值自动传递至子线程中。 ##### (2) 使用 `TransmittableThreadLocal` 为了演示更复杂的情况——比如涉及线程池执行器的情境下仍能维持一致性的行为,下面给出一段关于 `TransmittableThreadLocal` 的实例: ```java import com.alibaba.ttl.TransmittableThreadLocal; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; public class ExampleWithTTL { private static final TransmittableThreadLocal<String> ttlContext = new TransmittableThreadLocal<>(); public static void main(String[] args) { ExecutorService executor = Executors.newSingleThreadExecutor(); ttlContext.set("Main Context"); Runnable task = () -> System.out.println(ttlContext.get()); executor.execute(task); executor.shutdown(); } } ``` 在此例子中可以看到即使是在多线程环境下运行的任务也能获取到来自主线程设定好的 context 数据。 #### 四、总结说明 综上所述,在选择合适的解决方案之前应先明确实际需求所在之处。如果只是单纯希望让某个特殊属性随着新建出来的子进程一起生效的话那么采用标准库自带的功能即可满足;然而一旦遇到更加棘手的情形例如说要跨越多个不同类型的 worker threads 或者 RPC calls 来追踪请求相关信息的时候就需要借助第三方框架所提供的高级特性来完成目标了。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值