当我们在同一个线程中,想要共享变量的话,是可以直接使用ThreadLocal的,但是如果在父子线程之间,共享变量,ThreadLocal就不行了。

如以下代码,会抛出NPE:

public static ThreadLocal<Integer> sharedData = new ThreadLocal<>();


public static void main(String[] args) {
    sharedData.set(0);
    MyThread thread = new MyThread();
    thread.start();
    sharedData.set(sharedData.get() + 1);
    System.out.println("sharedData in main thread: " + sharedData.get());

}

static class MyThread extends Thread {

    @Override
    public void run() {
        System.out.println("sharedData in child thread: " + sharedData.get());
        sharedData.set(sharedData.get() + 1);
        System.out.println("sharedData in child thread after increment: " + sharedData.get());
    }
}

因为ThreadLocal 变量是为每个线程提供了独立的副本,因此不同线程之间只能访问它们自己的副本。

那么,想要实现数据共享,主要有两个办法,第一个是自己传递,第二个是借助InheritableThreadLocal

自己传递

我们可以在子线程中创建一个成员变量,这样在主线程创建子线程的时候,可以给成员变量赋值,这样实现数据共享。

import java.util.concurrent.ConcurrentHashMap;

public class SharedDataExample {
   public static void main(String[] args) {
      ConcurrentHashMap<String, String> sharedData = new ConcurrentHashMap<String, String>();
      MyThread thread = new MyThread(sharedData);
      thread.start();
      sharedData.put("key", "value");
      System.out.println("sharedData in main thread: " + sharedData.get("key"));
   }
}

class MyThread extends Thread {
   ConcurrentHashMap<String, String> sharedData;
   public MyThread(ConcurrentHashMap<String, String> data) {
      this.sharedData = data;
   }
   public void run() {
      sharedData.put("key", "new value");
      System.out.println("sharedData in child thread: " + sharedData.get("key"));
   }
}

但是一定要考虑线程安全的问题。

InheritableThreadLocal

与 ThreadLocal 不同,InheritableThreadLocal 可以在子线程中继承父线程中的值。在创建子线程时,子线程将复制父线程中的 InheritableThreadLocal 变量。

我们把开头的示例中ThreadLocal改成InheritableThreadLocal就可以了:

public static InheritableThreadLocal<Integer> sharedData = new InheritableThreadLocal<>();


    public static void main(String[] args) {
        sharedData.set(0);
        MyThread thread = new MyThread();
        thread.start();
        sharedData.set(sharedData.get() + 1);
        System.out.println("sharedData in main thread: " + sharedData.get());

    }

    static class MyThread extends Thread {

        @Override
        public void run() {
            System.out.println("sharedData in child thread: " + sharedData.get());
            sharedData.set(sharedData.get() + 1);
            System.out.println("sharedData in child thread after increment: " + sharedData.get());
        }
    }

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐