Java实现ThreadLocal的原理

可能到了冬天就变得懒惰了,所以都不太想敲代码就在看书上花了些时间.其中看到讲关于ThreadLocal,中文翻译是"本地线程"的这块,就想写写自己的理解.刚接触它的时候我认为它就是字面上的意思,但其实它是一个容器,用来存放线程的局部变量.我看过关于它的故事,其实早在JDK1.2的那个时代Sun公司的工程师们为了解决多线程并发问题而设计这么一个类.只不过设计的有些难用,所以到现在也没有太过广泛的使用,但它还是很有用的,以下我们通过一个例子来看看它的优点吧.
示例1:一个序列号生成器的程序，可能同时会有多个线程并发访问它，要保证每个线程得到的序列号都是自增的，而不能相互干扰。
1.先定义一个接口:

public interface Sequence {
          int getNumber();
}

2.每次调用 getNumber() 方法可获取一个序列号，下次再调用时，序列号会自增。再做一个线程类：

public class ClientThread extends Thread {
    private Sequence sequence;
    public ClientThread(Sequence sequence) {
        this.sequence = sequence;
    }
    @Override
    public void run() {
        for (int i = 0; i < 3; i++) {
            System.out.println(Thread.currentThread().getName() + " => " + sequence.getNumber());
        }
    }

3.我们先不用 ThreadLocal，直接运行下面方法.看看控制台打印各线程名对应的序列号:

public class SequenceA implements Sequence {
    private static int number = 0;
    public int getNumber() {
        number = number + 1;
        return number;
    }
    public static void main(String[] args) {
        Sequence sequence = new SequenceA();
        ClientThread thread1 = new ClientThread(sequence);
        ClientThread thread2 = new ClientThread(sequence);
        ClientThread thread3 = new ClientThread(sequence);
        thread1.start(); 
        thread2.start();
        thread3.start();
    }

序列号初始值是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 变量！这就是所谓的“非线程安全”问题了。
所以问题就来了,我们应该如何保证"线程安全"呢?请看看接下来的这个案例你就明白了.

public class SequenceB implements Sequence {
    private static ThreadLocal<Integer> numberContainer = new ThreadLocal<Integer>() {
        @Override
        protected Integer initialValue() {
            return 0;
        }
    };
    @Override
    public int getNumber() {
        numberContainer.set(numberContainer.get() + 1);
        return numberContainer.get();
    }
    public static void main(String[] args) {
        Sequence sequence = new SequenceB();
        ClientThread thread1 = new ClientThread(sequence);
        ClientThread thread2 = new ClientThread(sequence);
        ClientThread thread3 = new ClientThread(sequence);
        thread1.start();
        thread2.start();
        thread3.start();
    }

我通过 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,其实这个是为了提醒我们这个方法是需要由我们自己来实现的.说到这了,我们可以好好想一下,其实ThreadLocal 里面封装的可以说是一个Map集合.之前我就山寨了一个ThreadLocal，其中中定义了一个同步 Map尝试一下我的结论.以下就是我的代码:

public class MyThreadLocal<T> {
    private Map<Thread, T> container = Collections.synchronizedMap(new HashMap<Thread, T>());
    public void set(T value) { container.put(Thread.currentThread(), value); }
    public T get() {
        Thread thread = Thread.currentThread();
        T value = container.get(thread);
        if (value == null && !container.containsKey(thread)) {
            value = initialValue();
            container.put(thread, value);
        }
        return value;
    }
    public void remove() { 
    container.remove(Thread.currentThread())；
    }
    protected T initialValue() { 
    return null;
    }
}

下面是用这 MyThreadLocal 实现的:

public class SequenceC implements Sequence {
    private static MyThreadLocal<Integer> numberContainer = new MyThreadLocal<Integer>() {
        @Override
        protected Integer initialValue() {
            return 0;
        }
    };
    public int getNumber() {
        numberContainer.set(numberContainer.get() + 1);
        return numberContainer.get();
    }
    public static void main(String[] args) {
        Sequence sequence = new SequenceC();
        ClientThread thread1 = new ClientThread(sequence);
        ClientThread thread2 = new ClientThread(sequence);
        ClientThread thread3 = new ClientThread(sequence);
        thread1.start();
        thread2.start();
        thread3.start();
    }
}

其实无非是将ThreadLocal 换成了 MyThreadLocal,运行的结果和之前的一样.