ThreadLocal 那点事儿

ThreadLocal,直译为“线程本地”或“本地线程”,如果你真的这么认为,那就错了!其实,它就是一个容器,用于存放线程的局部变量,我认为应该叫做 ThreadLocalVariable(线程局部变量)才对,真不理解为什么当初 Sun 公司的工程师这样命名。

早在 JDK 1.2 的时代,java.lang.ThreadLocal 就诞生了,它是为了解决多线程并发问题而设计的,只不过设计得有些难用,所以至今没有得到广泛使用。其实它还是挺有用的,不相信的话,我们一起来看看这个例子吧。

一个序列号生成器的程序,可能同时会有多个线程并发访问它,要保证每个线程得到的序列号都是自增的,而不能相互干扰。

先定义一个接口:

1 public interface Sequence {
2  
3     int getNumber();
4 }

每次调用 getNumber() 方法可获取一个序列号,下次再调用时,序列号会自增。

再做一个线程类:

01 public class ClientThread extends Thread {
02  
03     private Sequence sequence;
04  
05     public ClientThread(Sequence sequence) {
06         this.sequence = sequence;
07     }
08  
09     @Override
10     public void run() {
11         for (int i = 0; i < 3; i++) {
12             System.out.println(Thread.currentThread().getName() + " => " + sequence.getNumber());
13         }
14     }
15 }

在线程中连续输出三次线程名与其对应的序列号。

我们先不用 ThreadLocal,来做一个实现类吧。

01 public class SequenceA implements Sequence {
02  
03     private static int number = 0;
04  
05     public int getNumber() {
06         number = number + 1;
07         return number;
08     }
09  
10     public static void main(String[] args) {
11         Sequence sequence = new SequenceA();
12  
13         ClientThread thread1 = new ClientThread(sequence);
14         ClientThread thread2 = new ClientThread(sequence);
15         ClientThread thread3 = new ClientThread(sequence);
16  
17         thread1.start();
18         thread2.start();
19         thread3.start();
20     }
21 }

序列号初始值是0,在 main() 方法中模拟了三个线程,运行后结果如下:

Thread-0 => 1
Thread-0 => 2
Thread-0 => 3
Thread-2 => 4
Thread-2 => 5
Thread-2 => 6
Thread-1 => 7
Thread-1 => 8
Thread-1 => 9

由于线程启动顺序是随机的,所以并不是0、1、2这样的顺序,这个好理解。为什么当 Thread-0 输出了1、2、3之后,而 Thread-2 却输出了4、5、6呢?线程之间竟然共享了 static 变量!这就是所谓的“非线程安全”问题了。

那么如何来保证“线程安全”呢?对应于这个案例,就是说不同的线程可拥有自己的 static 变量,如何实现呢?下面看看另外一个实现吧。

01 public class SequenceB implements Sequence {
02  
03     private static ThreadLocal<Integer> numberContainer = new ThreadLocal<Integer>() {
04         @Override
05         protected Integer initialValue() {
06             return 0;
07         }
08     };
09  
10     public int getNumber() {
11         numberContainer.set(numberContainer.get() + 1);
12         return numberContainer.get();
13     }
14  
15     public static void main(String[] args) {
16         Sequence sequence = new SequenceB();
17  
18         ClientThread thread1 = new ClientThread(sequence);
19         ClientThread thread2 = new ClientThread(sequence);
20         ClientThread thread3 = new ClientThread(sequence);
21  
22         thread1.start();
23         thread2.start();
24         thread3.start();
25     }
26 }

通过 ThreadLocal 封装了一个 Integer 类型的 numberContainer 静态成员变量,并且初始值是0。再看 getNumber() 方法,首先从 numberContainer 中 get 出当前的值,加1,随后 set 到 numberContainer 中,最后将 numberContainer 中 get 出当前的值并返回。

是不是很恶心?但是很强大!确实稍微饶了一下,我们不妨把 ThreadLocal 看成是一个容器,这样理解就简单了。所以,这里故意用 Container 这个单词作为后缀来命名 ThreadLocal 变量。

运行结果如何呢?看看吧。

Thread-0 => 1
Thread-0 => 2
Thread-0 => 3
Thread-2 => 1
Thread-2 => 2
Thread-2 => 3
Thread-1 => 1
Thread-1 => 2
Thread-1 => 3

每个线程相互独立了,同样是 static 变量,对于不同的线程而言,它没有被共享,而是每个线程各一份,这样也就保证了线程安全。 也就是说,TheadLocal 为每一个线程提供了一个独立的副本!

搞清楚 ThreadLocal 的原理之后,有必要总结一下 ThreadLocal 的 API,其实很简单。

  1. public void set(T value):将值放入线程局部变量中
  2. public T get():从线程局部变量中获取值
  3. public void remove():从线程局部变量中移除值(有助于 JVM 垃圾回收)
  4. protected T initialValue():返回线程局部变量中的初始值(默认为 null) 

为什么 initialValue() 方法是 protected 的呢?就是为了提醒程序员们,这个方法是要你们来实现的,请给这个线程局部变量一个初始值吧。

了解了原理与这些 API,其实想想 ThreadLocal 里面不就是封装了一个 Map 吗?自己都可以写一个 ThreadLocal 了,尝试一下吧。

01 public class MyThreadLocal<T> {
02  
03     private Map<Thread, T> container = Collections.synchronizedMap(new HashMap<Thread, T>());
04  
05     public void set(T value) {
06         container.put(Thread.currentThread(), value);
07     }
08  
09     public T get() {
10         Thread thread = Thread.currentThread();
11         T value = container.get(thread);
12         if (value == null && !container.containsKey(thread)) {
13             value = initialValue();
14             container.put(thread, value);
15         }
16         return value;
17     }
18  
19     public void remove() {
20         container.remove(Thread.currentThread());
21     }
22  
23     protected T initialValue() {
24         return null;
25     }
26 }
以上完全山寨了一个 ThreadLocal,其中中定义了一个同步 Map(为什么要这样?请读者自行思考),代码应该非常容易读懂。

下面用这 MyThreadLocal 再来实现一把看看。

01 public class SequenceC implements Sequence {
02  
03     private static MyThreadLocal<Integer> numberContainer = new MyThreadLocal<Integer>() {
04         @Override
05         protected Integer initialValue() {
06             return 0;
07         }
08     };
09  
10     public int getNumber() {
11         numberContainer.set(numberContainer.get() + 1);
12         return numberContainer.get();
13     }
14  
15     public static void main(String[] args) {
16         Sequence sequence = new SequenceC();
17  
18         ClientThread thread1 = new ClientThread(sequence);
19         ClientThread thread2 = new ClientThread(sequence);
20         ClientThread thread3 = new ClientThread(sequence);
21  
22         thread1.start();
23         thread2.start();
24         thread3.start();
25     }
26 }

以上代码其实就是将 ThreadLocal 替换成了 MyThreadLocal,仅此而已,运行效果和之前的一样,也是正确的。

其实 ThreadLocal 可以单独成为一种设计模式,就看你怎么看了。 

ThreadLocal 具体有哪些使用案例呢?

我想首先要说的就是:通过 ThreadLocal 存放 JDBC Connection,以达到事务控制的能力。

如何实现呢?下回分解!

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值