package concurrent;

import java.util.concurrent.atomic.AtomicInteger;

/**
Created by chengxiao on 2016/12/12.


*/


public class ThreadLocalDemo {


public static void main(String []args){


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


final Thread t = new Thread(){


@Override


public void run(){


System.out.println("当前线程:"+Thread.currentThread().getName()+",已分配ID:"+ThreadId.get());


}


};


t.start();


}


}


static   class ThreadId{


//一个递增的序列，使用AtomicInger原子变量保证线程安全


private static final AtomicInteger nextId = new AtomicInteger(0);


//线程本地变量，为每个线程关联一个唯一的序号


private static final ThreadLocal<Integer> threadId =


new ThreadLocal<Integer>() {


@Override


protected Integer initialValue() {


return nextId.getAndIncrement();//相当于nextId++,由于nextId++这种操作是个复合操作而非原子操作，会有线程安全问题(可能在初始化时就获取到相同的ID，所以使用原子变量


                    }


};
</span><span style="color: #008000;">//</span><span style="color: #008000;">返回当前线程的唯一的序列，如果第一次get，会先调用initialValue，后面看源码就了解了</span>

 <span style="color: #0000ff;">public</span> <span style="color: #0000ff;">static</span> <span style="color: #0000ff;">int</span><span style="color: #000000;"> get() {

     </span><span style="color: #0000ff;">return</span><span style="color: #000000;"> threadId.get();

 }


}


}

当前线程:Thread-4,已分配ID:1

当前线程:Thread-0,已分配ID:0

当前线程:Thread-2,已分配ID:3

当前线程:Thread-1,已分配ID:4

当前线程:Thread-3,已分配ID:2 

 public void set(T value) {

    Thread t = Thread.currentThread();//1.首先获取当前线程对象

        ThreadLocalMap map = getMap(t);//2.获取该线程对象的ThreadLocalMap

        if (map != null)

            map.set(this, value);//如果map不为空，执行set操作，以当前threadLocal对象为key，实际存储对象为value进行set操作

        else

            createMap(t, value);//如果map为空，则为该线程创建ThreadLocalMap

    }

ThreadLocalMap getMap(Thread t) {

    return t.threadLocals;//线程对象持有ThreadLocalMap的引用

}

    ThreadLocal.ThreadLocalMap threadLocals = null;

 public T get() {

     Thread t = Thread.currentThread();//1.首先获取当前线程

         ThreadLocalMap map = getMap(t);//2.获取线程的map对象

         if (map != null) {//3.如果map不为空，以threadlocal实例为key获取到对应Entry，然后从Entry中取出对象即可。

             ThreadLocalMap.Entry e = map.getEntry(this);

             if (e != null)

                 return (T)e.value;

         }

         return setInitialValue();//如果map为空，也就是第一次没有调用set直接get（或者调用过set，又调用了remove）时，为其设定初始值

     }

  private T setInitialValue() {

         T value = initialValue();//获取初始值

         Thread t = Thread.currentThread();

         ThreadLocalMap map = getMap(t);

         if (map != null)

             map.set(this, value);

         else

             createMap(t, value);

         return value;

     }

 protected T initialValue() {

         return null;

     }

作者：
dreamcatcher-cx

出处：
<http://www.cnblogs.com/chengxiao/>

本文版权归作者和博客园共有，欢迎转载，但未经作者同意必须保留此段声明，且在页面明显位置给出原文链接。