本地线程管理、仿ThreadLocal

在实际项目中,通常有这样的要求,对于同一个线程内多次请求获取对象,对象要求必须为同一个(如数据多表库多表进行事务处理,请求的连接对象必须为同一个),这就要用到线程管理技术。

线程管理,涉及到一个核心的类,java.lang.ThreadLocal,通过这个类来仅实现相应的功能。
这里写图片描述

代码演示:

public class ThreadLocalDemo {

   private static ThreadLocal<Object> t1 = new ThreadLocal<Object>();

   public static Object getValue(){
       Object o = t1.get();//不用给key,因为t1内部会自动获取当前线程的thread对象,并以上作为key到它的池中去

取obj
       if(o==null){
           System.out.println("空的....");
           Random r = new Random();
           o = r.nextInt(1000);
           t1.set(o);
       }
       return o;
   }

   @Test
   public void test(){
       Object obj = getValue();
       Object obj2 = getValue();
       System.out.println(obj+","+obj2);
       System.out.println(obj==obj2);   //true

      A a = new A();
      Object obj3 = a.aa();
      System.out.println(obj==obj3);    //true

      B b = new B();
      Object obj4 = b.bb();
      System.out.println(obj3==obj4);   //true

      System.out.println("=================");

      new Thread(){
        @Override
        public void run() {
             A a = new A();
             Object obj5 = a.aa();
             System.out.println("obj5:"+obj5);
             B b = new B();
              Object obj6 = b.bb();
              System.out.println("obj6:"+obj6);
              System.out.println(obj5==obj6);  //true
        }
      }.start();

   }

}

class A{
    public Object aa(){
        Object obj3 = ThreadLocalDemo.getValue();
        System.out.println(obj3);
        return obj3;
    }
}
class B{
    public Object bb(){
        Object obj4 = ThreadLocalDemo.getValue();
        System.out.println(obj4);
        return obj4;
    }
}

仿ThreadLocal功能:

public class MyThreadLocal {
    private Map<Thread,Object>map = new HashMap<Thread,Object>();

    public Object get(){
        Thread currentThread = Thread.currentThread();
        Object obj = map.get(currentThread);
        return obj;
    }

    public void set(Object obj){
        Thread currentThread = Thread.currentThread();
        map.put(currentThread,obj);
    }
}
发布了76 篇原创文章 · 获赞 53 · 访问量 27万+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章