ThreadLocal源码阅读

ThreadLocal简介:

threadlocal可以为线程提供线程本地变量,这个数据绑定在当前线程中的,只有当前线程可以访问。

要获取线程本地变量,只需要使用threadlocal.get()方法,要设置本地变量值,只需使用threadlocal,set()方法。

ThreadLocal.get():

首先来看get():

public T get() {
    //获取当前线程
    Thread t = Thread.currentThread();
    //获取当前线程中的ThreadLocalMap
    ThreadLocalMap map = getMap(t);
    if (map != null) {
        //ThreadLocalMap中节点为Entry
        ThreadLocalMap.Entry e = map.getEntry(this);
        if (e != null) {
            @SuppressWarnings("unchecked")
            //获取本地变量
            T result = (T)e.value;
            return result;
        }
    }
    return setInitialValue();
}

get()首先会获取ThreadLocalMap,这个Map实际存储在当前线程中,源码如下:

ThreadLocalMap getMap(Thread t) {
    return t.threadLocals;
}

t中存放着这个map

public class Thread implements Runnable {
    ...
    ThreadLocal.ThreadLocalMap threadLocals = null;
    ...
}

获取到ThreadLocalMap后,会根据当前threadlocal获取ThreadLocalMap中存储的Entry,Entry结构如下:

static class Entry extends WeakReference<ThreadLocal<?>> {
    /** The value associated with this ThreadLocal. */
    Object value;
​
    Entry(ThreadLocal<?> k, Object v) {
        super(k);
        value = v;
    }
}

可见ThreadLocalMap中基本存储单元为Entry,Entry是一个键值对,key为一个threadlocal,而值由线程设置,ThreadLocalMap中有一个Entry数组名为table用于存放这些Entry。自此,拿到了Entry,若之前已经设置过数据,则直接返回value,否则会执行初始化操作:

private T setInitialValue() {
​
    //initialValue()返回null
    T value = initialValue();
    Thread t = Thread.currentThread();
    
    //首先获得当前线程对应的ThreadLocalMap
    ThreadLocalMap map = getMap(t);
    //若map已经初始化,则设置初始值null
    if (map != null) {
        map.set(this, value);
    } else {
        //map没有初始化,则创建map并设置初始值null
        createMap(t, value);
    }
    if (this instanceof TerminatingThreadLocal) {
        TerminatingThreadLocal.register((TerminatingThreadLocal<?>) this);
    }
    return value;
}

再来看createMap():

void createMap(Thread t, T firstValue) {
    t.threadLocals = new ThreadLocalMap(this, firstValue);
}

逻辑很简单,创建一个ThreadLocalMap,并初始化值。

总结:get()方法首先会去获取当前线程的ThreadLocalMap,然后再去获取Map中存放的threadlocal对应的值。这一过程中可能会涉及到一些初始化操作。

ThreadLocal.set():

再来看set():

public void set(T value) {
    //获取当前线程对应的ThreadLocalMap
    Thread t = Thread.currentThread();
    ThreadLocalMap map = getMap(t);
    if (map != null) {
        //map不为null,则设置threadlocal,value键值对
        map.set(this, value);
    } else {
        createMap(t, value);
    }
}

与前述操作基本一致。

ThreadLocal使用:

经过上述分析,其实可以看出一个threadlocal变量可以为不同的线程维护本地变量。

public class test {
    public static void main(String[] args) {
​
        ThreadLocal<String> localStr = new ThreadLocal<>();
​
        Thread thread1 = new Thread(new Runnable() {
            @Override
            public void run() {
                localStr.set("thread1");
                loaclInt.set(1);
                System.out.println("线程1的localstr: " + localStr.get());
            }
        });
        Thread thread2 = new Thread(new Runnable() {
            @Override
            public void run() {
                localStr.set("thread2");
                loaclInt.set(2);
                System.out.println("线程2的localstr: " + localStr.get());
            }
        });
        thread1.start();
        thread2.start();

测试代码中使用同一个localStr为两个线程都维护以他们的本地变量,测试结果如下:

线程1的localstr: thread1 线程2的localstr: thread2

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
ThreadLocal源码是Java中一个关键的类,它提供了一种在多线程环境下实现线程本地变量的机制。在JDK 8之前和之后,ThreadLocal的内部结构有所变化。ThreadLocal源码分为两部分:ThreadLocal类和ThreadLocalMap类。 ThreadLocal类是一个泛型类,它包含了两个核心方法:set()和get()。set()方法用于将一个值与当前线程关联起来,get()方法用于获取当前线程关联的值。 ThreadLocalMap类是ThreadLocal的内部类,它用于存储每个线程的本地变量。在JDK 8之前,ThreadLocalMap是通过线性探测法解决哈希冲突的,每个ThreadLocal对象都对应一个Entry对象,Entry对象包含了ThreadLocal对象和与之关联的值[2]。 在JDK 8之后,ThreadLocalMap的实现方式发生了改变。使用了类似于HashMap的方式,采用了分段锁的机制来提高并发性能。每个线程维护一个ThreadLocalMap对象,其中的Entry对象也是采用链表的形式来解决哈希冲突。 总结起来,ThreadLocal源码主要由ThreadLocal类和ThreadLocalMap类组成。ThreadLocal类提供了set()和get()方法来管理线程本地变量,而ThreadLocalMap类则负责存储每个线程的本地变量,并解决哈希冲突的问题。 史上最全ThreadLocal 详解 ThreadLocal源码分析_02 内核(ThreadLocalMap) 【JDK源码】线程系列之ThreadLocal 深挖ThreadLocal ThreadLocal原理及内存泄露预防 ThreadLocal原理详解——终于弄明白了ThreadLocal ThreadLocal使用与原理 史上最全ThreadLocal 详解。 ThreadLocal源码分析,主要有ThreadLocal源码以及ThreadLocal的内部结构在jdk8前后的变化。 使用方式非常简单,核心就两个方法set/get public class TestThreadLocal { private static final ThreadLocal<String> threadLocal = new ThreadLocal<>(); public static void main(String[] args) { new Thread(new Runnable() { @Override public void run() { try { threadLocal.set("aaa"); Thread.sleep(500); System.out.println("threadA:" threadLocal.get()); } catch (InterruptedException e) { e.printStackTrace(); } } }).start(); new Thread(new Runnable() { @Override public void run() { threadLocal.set("bbb"); System.out.println("threadB:" threadLocal.get()); } }).start(); } }
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值