Android Handler消息机制中的ThreadLocal源码分析

系列文章

1. 前言

ThreadLocal定义如下:

public class ThreadLocal<T>

可以看出其是一个泛型类,同一个 ThreadLocal 所包含的对象(对ThreadLocal< String>而言即为 String 类型变量),在不同的线程中有不同的副本(实际是不同的实例,后文会详细阐述)

每个线程内有自己的实例副本,且该副本只能由当前线程使用。这是也是 ThreadLocal 命名的由来

1.1 使用场景

ThreadLocal 适用于每个线程需要自己独立的实例且该实例需要在多个方法中被使用,也即变量在线程间隔离而在方法或类间共享的场景。在Android中的Handler中便使用了,主要用来存放和获取不同线程中的Looper。

摘自Java进阶(七)正确理解Thread Local的原理与适用场景

2. Looper

Android消息机制Handler一文中已经分析过Android的Handler机制,其中的Looper起到了消息循环器的作用,Looper的获取便使用到了ThreadLocal:

// Looper源码中sThreadLocal变量
static final ThreadLocal<Looper> sThreadLocal = new ThreadLocal<Looper>();

大家注意,子线程默认是没有Looper的,而UI线程(ActivityThread)帮我们开启了Looper,因为ActivityThread被创建时就会初始化Looper,这也是在主线程中默认可以使用Handler的原因。这一点在Android消息机制Handler中已经分析过了。

2.1 创建Looper

首先看一下构造方法,在构造方法中它会创建一个MessageQueue消息队列,然后将当前线程的对象保存起来。

private Looper(boolean quitAllowed) {
    mQueue = new MessageQueue(quitAllowed);
    mThread = Thread.currentThread();
}

Handler的工作需要Looper,没有Looper的线程就会报错,那么如何为线程创建Looper呢?通过Looper.prepare()即可为当前线程创建一个Looper,同时会将Looper保存在ThreadLocal中,Handler的构造函数便是从ThreadLocal中获取Looper对象,从代码如下:

private static void prepare(boolean quitAllowed) {
    if (sThreadLocal.get() != null) {
        throw new RuntimeException("Only one Looper may be created per thread");
    }
    sThreadLocal.set(new Looper(quitAllowed));
}

Looper除了prepare()方法外,还提供了prepareMainLooper()方法,这个方法主要是给主线程即ActivityThread创建Looper使用的,本质也是通过prepare()方法来实现的,源码如下:

public static void prepareMainLooper() {
    prepare(false);
    synchronized (Looper.class) {
        if (sMainLooper != null) {
            throw new IllegalStateException("The main Looper has already been prepared.");
        }
        sMainLooper = myLooper();
    }
}

在程序启动的时候,系统已经帮我们自动调用了Looper.prepare()方法。ActivityThread中的main()方法调用了Looper.prepareMainLooper()方法,而这个方法又会再去调用Looper.prepare()方法。因此我们应用程序的主线程中会始终存在一个Looper对象,从而不需要再手动去调用Looper.prepare()方法了。

由于主线程Looper比较特殊,所以Looper提供了一个getMainLooper()方法,通过它可以在任何地方获取主线程的Looper。

2.2 为什么Handler中使用Looper

Looper的作用域就是线程并且不同线程具有不同的Looper,这个时候通过ThreadLocal就可以轻松实现Looper在线程中的存取,如果不采用ThreadLocal,那么系统就必须提供一个全局的哈希表供Handler查找指定线程的Looper,这样一来就必须提供一个类似于LooperManager的类了,但是系统并没有这么做而是选择了ThreadLocal,这就是ThreadLocal的好处。

摘自Android的消息机制之ThreadLocal的工作原理

3. ThreadLocal源码分析

我们注意到在Looper.prepare()方法中,我们会先判断sThreadLocal.get()方法为不为空,不为空再去调用sThreadLocal.set()方法,下面我们就来具体分析下ThreadLocal的get和set方法的源码。

3.1 ThreadLocal.get

get方法源码如下:

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;
        }
    }
    return setInitialValue();
}

读取实例时,线程首先通过getMap(t)方法获取线程自身的 ThreadLocalMap,ThreadLocalMap是ThreadLocal的内部类,真正保存了数据,同时每个线程中都有一个ThreadLocalMap对象,定义如下:

// Thread源码
ThreadLocal.ThreadLocalMap threadLocals = null;

getMap(t)源码:

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

其中直接返回当前线程的threadLocals变量,但是线程默认该变量为null,因此第一次获取时肯定为null,所以get()方法便会直接调用setInitialValue()方法:

private T setInitialValue() {
    T value = initialValue(); // 此方法直接返回null
    Thread t = Thread.currentThread();
    ThreadLocalMap map = getMap(t); // 再次获取map还是null
    if (map != null)
        map.set(this, value); 
    else
        createMap(t, value); // 创建map
    return value;
}

所以从上面看出,最后会调用createMap创建ThreadLocalMap对象:

void createMap(Thread t, T firstValue) {
    // 初始化当前线程中的ThreadLocalMap对象
    t.threadLocals = new ThreadLocalMap(this, firstValue);
}

我们再回到get方法中,如果我们map != null发生,那么从源码中可以看出会从ThreadLocalMap中获取一个Entry对象,从中获取value并返回,这说明我们真正的数据都是保存在ThreadLocalMap.Entry中,后面我们再来具体分析。

3.2 ThreadLocal.set

接下来看下set方法源码:

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

类似get方法,首先获取当前线程ThreadLocalMap对象,如果不为空,就调用map.set方法,如果为空就创建ThreadLocalMap,上面已经分析过了。

4. ThreadLocalMap

成员变量

// 初始容量,必须是2的整数次幂
private static final int INITIAL_CAPACITY = 16;

// Entry数组,必要时rehash,长度必须是2的整数次幂
private Entry[] table;

// table数组真正数据大小
private int size = 0;

// 容量阈值,达到此值需rehash
private int threshold; 

4.1 ThreadLocalMap.Entry

上面我们提到,真正保存数据的是ThreadLocalMap.Entry对象,下面我们先来看下其定义:

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

可以看出,其构造函数接受两个参数,一个是ThreadLocal对象,另一个是待保存的数据。ThreadLocalMap 的每个 Entry 都是一个对键 的弱引用,这是为了防止内存泄漏, 对值的强引用。而且ThreadLocalMap中还定了一个Entry数组,可见最终的数据都是存放在此数组中:

// ThreadLocalMap源码
private Entry[] table;

4.2 ThreadLocalMap.set

private void set(ThreadLocal<?> key, Object value) {
    Entry[] tab = table;
    int len = tab.length;
    // 获取到key对应在数组中的下标位置
    int i = key.threadLocalHashCode & (len-1);
    for (Entry e = tab[i];
         e != null;
         e = tab[i = nextIndex(i, len)]) {
        // 获取到的entry不为null
        // 获取保存在entry中的弱引用key,即ThreadLocal对象
        ThreadLocal<?> k = e.get();
        // 如果键值相等就更新value
        if (k == key) {
            e.value = value;
            return;
        }
        // 如果键值为null就通过 replaceStaleEntry 方法将所有键为 null
        // 的 Entry 的值设置为 null,从而使得该值可被回收
        if (k == null) {
            replaceStaleEntry(key, value, i);
            return;
        }
    }
    // 上面获取到的entry为null,说明该位置还没有被初始化
    // 初始化该位置的Entry
    tab[i] = new Entry(key, value);
    // table数组的大小+1
    int sz = ++size;
    if (!cleanSomeSlots(i, sz) && sz >= threshold)
        rehash();
}

关于set方法的解释基本都写在代码注释中了,如上。threadLocalHashCode & (size - 1)的算法,这相当于取模运算threadLocalHashCode % size的一个更高效的实现。

这里注意下在获取到key对应在数组中的下标位置时,是通过threadLocalHashCode变量与table长度位与实现的,在ThreadLocalMap中的Entry的数组长度规定了必须是2的整数次幂,而threadLocalHashCode是定义在ThreadLocal中的final变量,每个ThreadLocal对象的此变量是唯一的,threadLocalHashCode 是专门为 ThreadLocalMap 优化过的,可以减少哈希碰撞。

同时如果出现hash碰撞了,便会获取下一个index,下标不断加 1,不是 HashMap 的链地址法,采用的是线性探测法:

private static int nextIndex(int i, int len) {
    return ((i + 1 < len) ? i + 1 : 0);
}

4.3 ThreadLocalMap.replaceStaleEntry(替换陈旧Entry)

上述set过程中,如果出现了key为null的情况,表明此entry为staled entry(陈旧的Entry),就将其替换为当前的key和value:

private void replaceStaleEntry(ThreadLocal<?> key, Object value,
                               int staleSlot) {
    Entry[] tab = table;
    int len = tab.length;
    Entry e;
    
    int slotToExpunge = staleSlot;
    // 从当前slot向前遍历
    for (int i = prevIndex(staleSlot, len);
         (e = tab[i]) != null;
         i = prevIndex(i, len))
        if (e.get() == null)
            slotToExpunge = i;
    // 从当前slot向后遍历
    for (int i = nextIndex(staleSlot, len);
         (e = tab[i]) != null;
         i = nextIndex(i, len)) {
        ThreadLocal<?> k = e.get();
        // If we find key, then we need to swap it
        // with the stale entry to maintain hash table order.
        // The newly stale slot, or any other stale slot
        // encountered above it, can then be sent to expungeStaleEntry
        // to remove or rehash all of the other entries in run.
        // 如果我们找到了key,那么我们替换它和stale Entry以保持table的顺序
        if (k == key) {
            // 替换它和stale Entry
            e.value = value;
            tab[i] = tab[staleSlot];
            tab[staleSlot] = e;
            
            if (slotToExpunge == staleSlot)
                slotToExpunge = i;
            cleanSomeSlots(expungeStaleEntry(slotToExpunge), len);
            return;
        }
        // 向后扫描没有找到stale entry
        if (k == null && slotToExpunge == staleSlot)
            slotToExpunge = i;
    }
    // key没有找到,则重新初始化
    tab[staleSlot].value = null;
    tab[staleSlot] = new Entry(key, value);
    // 清理
    if (slotToExpunge != staleSlot)
        cleanSomeSlots(expungeStaleEntry(slotToExpunge), len);
}

4.4 ThreadLocalMap.getEntry

private Entry getEntry(ThreadLocal<?> key) {
    // 获取到key对应在数组中的下标位置
    int i = key.threadLocalHashCode & (table.length - 1);
    // 获取entry
    Entry e = table[i];
    // entry不为null,且键值相等就返回entry
    if (e != null && e.get() == key)
        return e;
    // entry为null,
    else
        return getEntryAfterMiss(key, i, e);
}

关于getEntry的解释基本都写在代码注释中了,如上。

4.5 ThreadLocalMap.cleanSomeSlots

private boolean cleanSomeSlots(int i, int n) {
    boolean removed = false;
    Entry[] tab = table;
    int len = tab.length;
    do {
        i = nextIndex(i, len);
        Entry e = tab[i];
        if (e != null && e.get() == null) {
            n = len;
            removed = true;
            i = expungeStaleEntry(i);
        }
    } while ( (n >>>= 1) != 0);
    return removed;
}

5. 总结

ThreadLocal.jpg

上图总结了ThreadLocal具体是怎么保存Looper的,首先在Looper类中定义了sThreadLocal变量,然后调用ThreadLocal的get/set来获取/设置当前线程Looper,get/set之前都是要先调用getMap获取当前线程的threadLocals变量,threadLocals对应的是ThreadLocalMap,get/set方法又分别调用的是ThreadLocalMap中的getEntry/set方法,ThreadLocalMap中真正保存数据的是Entry对象,其以ThreadLocal为弱key,传入的值为value。因为每个线程都有私有的threadLocals变量,便可做到数据的线程隔离。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,839评论 6 482
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,543评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 153,116评论 0 344
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,371评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,384评论 5 374
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,111评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,416评论 3 400
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,053评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,558评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,007评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,117评论 1 334
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,756评论 4 324
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,324评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,315评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,539评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,578评论 2 355
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,877评论 2 345

推荐阅读更多精彩内容

  • 写在前面 本次使用的源码来自于Android API 25 Platform. 引子 在Android中,在子线程...
    houtrry阅读 624评论 0 1
  • 1. 背景 ThreadLocal源码解读,网上面早已经泛滥了,大多比较浅,甚至有的连基本原理都说的很有问题,包括...
    时之令阅读 635评论 1 5
  • java版本是 java1.8.0_181 每个版本的具体实现细节都大同小异,接着从以下几个问题进行分析。 为什...
    衣忌破阅读 154评论 0 0
  • 在理解Handler、Looper之前,先来说说ThreadLocal这个类,听名字好像是一个本地线程的意思,实际...
    MrKing5946阅读 244评论 0 1
  • 一. 简介 提醒篇幅较大需耐心。 简介来自ThreadLocal类注释 ThreadLocal类提供了线程局部 (...
    BrightLoong阅读 9,157评论 2 14