简单扣一下ThreadLocal源码

逻辑跃动师
• 阅读 1278

简介

ThreadLocal内部定义了一个静态内部类ThreadLocalMap,这个Map的key就是当前的ThreadLocal,value则是要存储的线程局部变量。准确的说key保存的是ThreadLocal的弱引用。在每个线程内部会维护这样的ThreadLocal.ThreadLocalMap,这样做的好处不言而喻,可以做到线程之间隔离,并且随着线程关闭,ThreadLocalMap以及其中的对象值会被回收。

插入数据

获取当前线程的ThreadLocalMap,如果已初始化就直接插入,如果未初始化就先初始化再插入。

    public void set(T value) {
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        if (map != null)
            map.set(this, value);
        else
            createMap(t, value);
    }

我们先来分析ThreadLocalMapset方法

private void set(ThreadLocal<?> key, Object value) {

            Entry[] tab = table;
            int len = tab.length;
            // 根据Hash值计算数据该插入的位置
            // 这个Hash值得获取是通过一个魔数递增的方式得到
            int i = key.threadLocalHashCode & (len-1);
            // 遍历桶,直到遇到桶中空槽位置停止
            for (Entry e = tab[i];
                 e != null;
                 e = tab[i = nextIndex(i, len)]) {
                ThreadLocal<?> k = e.get();
                //key相同做覆盖操作
                if (k == key) {
                    e.value = value;
                    return;
                }
                // key为null,说明已经失效,作为弱引用,已被GC
                // 此时需要清理过期Key对应的value
                if (k == null) {
                    replaceStaleEntry(key, value, i);
                    return;
                }
            }
            //走到这里,说明遇到了一个空的槽,则直接插入
            tab[i] = new Entry(key, value);
            int sz = ++size;
            //先清理过期的槽,并判断是否要扩容
            if (!cleanSomeSlots(i, sz) && sz >= threshold)
                // 进行扩容,2倍,扩容前也会清理过期的槽
                rehash();
        }

可以看出在一个简单的set方法中有多次涉及到清理过期的槽,虽然清理的方法有多个,但是最终都是调用expungeStaleEntry方法。入参就是某个过期槽在桶中的位置。

 private int expungeStaleEntry(int staleSlot) {
            Entry[] tab = table;
            int len = tab.length;
            // 当前过期的槽置为null,方便GC
            tab[staleSlot].value = null;
            tab[staleSlot] = null;
            size--;

            Entry e;
            int i;
            //从当前过期槽的下一个位置开始遍历,直到遇到空槽
            for (i = nextIndex(staleSlot, len);
                 (e = tab[i]) != null;
                 i = nextIndex(i, len)) {
                ThreadLocal<?> k = e.get();
                //这里将过期的槽置为null
                if (k == null) {
                    e.value = null;
                    tab[i] = null;
                    size--;
                } else {
                    //注意:走到这个分支,当前的槽未失效,前面过期槽的位置已经被腾出
                    //获取槽i处结点原本应存放的位置
                    //因为ThreadLocal解决Hash冲突的方式,h必定小于等于i
                    int h = k.threadLocalHashCode & (len - 1);
                    //h!=i,说明i处存放的结点不是它原本位置,需要挪动
                    if (h != i) {
                        //将该处的结点挪走,置为null
                        tab[i] = null;
                        // 从它原本应存放的位置h,开始遍历,
                        // 找到一个最靠近h位置的空槽,并插入数据
                        while (tab[h] != null)
                            h = nextIndex(h, len);
                        tab[h] = e;
                    }
                }
            }
            // 返回的桶中空槽的位置
            return i;
        }

通过上面源码分析,expungeStaleEntry方法做了两件事:

  • 遍历桶,清理过期的key对应的value和槽(遇到空槽停止)
  • 遇到未失效的槽,尝试将该槽的结点往靠近它原本位置的附近挪动
    对于第二点,我们需要补充一下,ThreadLocal解决Hash冲突的方式:
    在往ThreadLocalMap中set值时,如果当前位置已经被占,且key不相同,那它就往后找一个空的槽存放,也就是说,Map桶中元素的位置可能不是该元素本身应当存放的位置,而是靠后,所以清理过期槽后,尽量把它挪到靠近原本位置附近,这个是为了查询方便

获取值

其实分析完了set方法,我们大致也能猜出查询方法是怎么回事,无非就是根据key计数出桶的位置,如果该位置的key与当前的key不是同一个,那就往后遍历,找到key值相同的结点,遍历过程中顺带手清理一下过期的槽。

  public T get() {
        Thread t = Thread.currentThread();
        ThreadLocalMap map = getMap(t);
        if (map != null) {
            // 核心就是这里
            ThreadLocalMap.Entry e = map.getEntry(this);
            if (e != null) {
                @SuppressWarnings("unchecked")
                T result = (T)e.value;
                return result;
            }
        }
        //往map中存一个null的Value,并返回
        return setInitialValue();
    }

简单看下map中的get方法

       private Entry getEntry(ThreadLocal<?> key) {
            int i = key.threadLocalHashCode & (table.length - 1);
            Entry e = table[i];
            //查到就返回
            if (e != null && e.get() == key)
                return e;
            else
                //查不到就遍历往后找
                return getEntryAfterMiss(key, i, e);
        }
点赞
收藏
评论区
推荐文章
Easter79 Easter79
3年前
sql:mysql:函数:TIMESTAMPDIFF函数实现TimeStamp字段相减,求得时间差
<divclass"htmledit\_views"id"content\_views"<p&nbsp;函数内指定是minute,则最终结果value值的单位是分钟,如果函数内指定为hours,则最终结果value值单位为小时。</p<preclass"has"name"code"<codeclass"hljssql"<
待兔 待兔
4年前
ThreadLocal源码分析
最近在学多线程并发的知识,发现好像ThreadLoca还挺重要,决定看看源码以及查找各方资料来学习一下。ThreadLocal能够提供线程的局部变量,让每个线程都可以通过set/get来对这个局部变量进行操作,不会和其它线程的局部变量进行冲突,实现了线程的数据隔离。首先是ThreadLocal的结构:每个Thread维护一个ThreadLocalMap,这个
Wesley13 Wesley13
3年前
java中map接口hashMap以及Enty之间的用法和关系
java中map接口hashMap以及Enty之间的转换首先说的是map接口:Map提供了一种映射关系,其中的元素是以键值对(keyvalue)的形式存储的,能够实现根据key快速查找value;建(key值)不可重复,value值可以重复,一个value值可以和很多key值形成对应关系,每个建最多只能映射到一个值。Ma
Wesley13 Wesley13
3年前
Java ThreadLocal的内存泄漏问题
ThreadLocal提供了线程独有的局部变量,可以在整个线程存活的过程中随时取用,极大地方便了一些逻辑的实现。常见的ThreadLocal用法有:\存储单个线程上下文信息。比如存储id等;\使变量线程安全。变量既然成为了每个线程内部的局部变量,自然就不会存在并发问题了;\减少参数传递。比如做一个trace工具,能够输出工程从开始到结
Stella981 Stella981
3年前
MapReduce中combine、partition、shuffle的作用是什么
概括:combine和partition都是函数。中间的步骤应该仅仅有shuffle!1.combinecombine分为map端和reduce端,作用是把同一个key的键值对合并在一起,能够自己定义的。combine函数把一个map函数产生的<key,value对(多个key,value)合并成一个新的<key2,value
Easter79 Easter79
3年前
ThreadLocal 原理和使用场景分析
ThreadLocal不知道大家有没有用过,但至少听说过,今天主要记录一下ThreadLocal的原理和使用场景。使用场景直接定位到ThreadLocal的源码,可以看到源码注释中有很清楚的解释:它是线程的局部变量,这些变量只能在这个线程内被读写,在其他线程内是无法访问的。ThreadLocal定义的通常是
Stella981 Stella981
3年前
HashMap Hashtable 的区别
Hashtable 和 HashMap作为 Map 的基本特性两者都实现了Map接口,基本特性相同\          对同一个Key,只会有一个对应的value值存在\          如何算是同一个Key?首先,两个key对象的hash值相同,其次,key对象的equals方法返回真内部数据结构Hashtab
Wesley13 Wesley13
3年前
Java集合
HashMap\_详解简述实现了什么接口Map和抽象类AbstractMapCloneableSerializable核心内容线程不同步。根据key的hashcode进行存储,内部使用静态内部类Node的数组进行存
Stella981 Stella981
3年前
Redis散列(Hash)的相关命令
散列就像一个减配的Redis内部及其类似Java的Map内容就是key:value结构hash类型在面向对象编程的运用中及其适合,因为它可以直接保存编程语言中的实体类关系增hsethsetkeyfieldvalue设置key指定的哈希集字段的值127.0.0.1:6379h
Wesley13 Wesley13
3年前
Java中的Map集合
Map接口简介Map接口是一种双列集合,它的每个元素都包含一个键对象Key和值对象Value,键和值对象之间存在一种对应关系,称为映射。从Map集合中访问元素时,只要指定了Key,就能找到对应的Value,Map中的键必须是唯一的,不能重复,如果存储了相同的键,后存储的值会覆盖原有的值,简而言之就是键相同,值覆盖。Map常用
Wesley13 Wesley13
3年前
APP 验证码 采用MEMCAHED验证的坑
都知道APP不是用浏览器来的,所以验证码生成sessioncookie是行不通了。一开始APP进入验证码界面,先让他请求一次服务器,生成key:code返回去。再把拿到的KEY带到服务器上面去生成图片。同时生成一个key:code再把key返回到客户端,当客户端验证图片的时候再把key再到服务器上去获取key里的code进行对比。一旦对比成功放行,对比