Java Concurrency - ThreadLocal, 本地线程变量

共享数据是多线程应用最常见的问题之一，但有时我们需要为每个线程保存一份独立的变量。Java API 提供了 ThreadLocal 来解决这个问题。

一个 ThreadLocal 作用的例子：

import java.util.Date;

public class Main {

    public static void main(String[] args) {

        Runnable task = new Runnable() {

            private ThreadLocal<Date> dateVar = new ThreadLocal<Date>();

            public void run() {

                dateVar.set(new Date());

                System.out.printf("%s, GET dateVar: %s\n", Thread.currentThread().getName(), dateVar.get());

                try {

                    Thread.sleep(5000L);

                } catch (InterruptedException e) {

                    e.printStackTrace();

                }

                System.out.printf("%s, FINAL dateVar: %s\n", Thread.currentThread().getName(), dateVar.get());

            }

        };

        for (int i = 0; i < 3; i++) {

            String threadName = "Thread" + (i + 1);

            Thread thread = new Thread(task, threadName);

            thread.start();

            try {

                Thread.sleep(1000L);

            } catch (InterruptedException e) {

                e.printStackTrace();

            }

        }

    }

}

观察运行结果：

Thread1, GET dateVar: Fri Oct 14 22:06:33 CST 2016
Thread2, GET dateVar: Fri Oct 14 22:06:34 CST 2016
Thread3, GET dateVar: Fri Oct 14 22:06:35 CST 2016
Thread1, FINAL dateVar: Fri Oct 14 22:06:33 CST 2016
Thread2, FINAL dateVar: Fri Oct 14 22:06:34 CST 2016
Thread3, FINAL dateVar: Fri Oct 14 22:06:35 CST 2016

可以看到每个线程都共用一个 Task 实例，线程之间间隔 1 秒启动。线程执行 run 方法的时候首先将 dateVar 的值设置为系统当前时间并打印 dateVar 值，然后线程会休眠 5 秒，最后再打印 dateVar 的值。注意到 run 方法设置 dateVar 值与最后打印 dateVar 值间隔 5 秒，而下一个线程启动时只间隔 1 秒，在当前线程打印 dateVar 之前，下个线程甚至是下下个线程已经重置 dateVar 的值，但是每个线程最后打印 dateVar 值的时候仍然是显示该线程最初设置的值。可见，每个线程中的 dateVar 并不会被其他线程所干扰。

再看下没有使用 ThreadLocal 的情况。

import java.util.Date;

public class Main2 {

    public static void main(String[] args) {

        Runnable task = new Runnable() {

            private Date dateVar;

            public void run() {

                dateVar = new Date();

                System.out.printf("%s, GET dateVar: %s\n", Thread.currentThread().getName(), dateVar);

                try {

                    Thread.sleep(5000L);

                } catch (InterruptedException e) {

                    e.printStackTrace();

                }

                System.out.printf("%s, FINAL dateVar: %s\n", Thread.currentThread().getName(), dateVar);

            }

        };

        for (int i = 0; i < 3; i++) {

            String threadName = "Thread" + (i + 1);

            Thread thread = new Thread(task, threadName);

            thread.start();

            try {

                Thread.sleep(1000L);

            } catch (InterruptedException e) {

                e.printStackTrace();

            }

        }

    }

}

观察运行结果：

Thread1, GET dateVar: Fri Oct 14 22:17:38 CST 2016
Thread2, GET dateVar: Fri Oct 14 22:17:39 CST 2016
Thread3, GET dateVar: Fri Oct 14 22:17:40 CST 2016
Thread1, FINAL dateVar: Fri Oct 14 22:17:40 CST 2016
Thread2, FINAL dateVar: Fri Oct 14 22:17:40 CST 2016
Thread3, FINAL dateVar: Fri Oct 14 22:17:40 CST 2016

可以看到，当直接使用 Date 类型时，线程最终打印 dateVar 的值与最初设置的值不一致。这个是因为所有的线程共享一个 Task 实例，所以 dateVar 是共享数据，在多个线程竞争时，造成数据不一致。

ThreadLocal 的初始化

可以通过覆盖 initialValue 方法初始化 ThreadLocal 的值。

ThreadLocal<Date> dateVar = new ThreadLocal<Date>() {

    @Override

    protected Date initialValue() {

        return new Date();

    }

};

InheritableThreadLocal

如果在线程 A 创建了子线程 B，那么线程 A 和线程 B 都是各自维护一份 ThreadLocal 值，线程 A 的 ThreadLocal 值不会传递给子线程 B。Java API 提供了 InheritableThreadLocal 类，它是 ThreadLocal 的子类。如果使用 InheritableThreadLocal，那么在线程 A 创建子线程 B，线程 A 和线程 B 仍然都是各自维护一份 InheritableThreadLocal 值，但是现场 A 的 InheritableThreadLocal 值则会传递给子线程 B。可以重写 childValue 方法修改从父现场继承的 InheritableThreadLocal 值。

public static ThreadLocal<Date> dateVar = new InheritableThreadLocal<Date>() {

    protected Date initialValue() {

        return new Date();

    };

    protected Date childValue(Date parentValue) {

        return new Date(parentValue.getTime() + 1000L);

    };

};

巴特西

Java Concurrency - ThreadLocal, 本地线程变量

ThreadLocal 的初始化

InheritableThreadLocal

最新文章

热门文章