InheritableThreadLocal源码学习

InheritableThreadLocal继承了ThreadLocal，重写了3个方法

1. childValue
2. getMap
3. createMap

ThreadLocal是不支持childValue方法的，getMap返回线程的inheritableThreadLocals属性（同样是ThreadLocalMap类型），createMap则同样是创建Map只是将map赋值到inheritableThreadLocals
那么如何实现父子线程继承value呢？

Thread构造器

看下Thread线程的初始化方法

1. 默认线程构造器启用可继承的inheritableThreadLocals
2. 如果父线程inheritableThreadLocals可继承的ThreadLocalMap不为空则基于父类map创建子线程的inheritableThreadLocals，调用java.lang.ThreadLocal#createInheritedMap构建

public Thread() {
    init(null, null, "Thread-" + nextThreadNum(), 0);
}
private void init(ThreadGroup g, Runnable target, String name,
                  long stackSize) {
    init(g, target, name, stackSize, null, true);
}
private void init(ThreadGroup g, Runnable target, String name,
                  long stackSize, AccessControlContext acc,
                  boolean inheritThreadLocals) {
    ...
    this.name = name;
    Thread parent = currentThread();
    SecurityManager security = System.getSecurityManager();
    if (g == null) {
        /* Determine if it's an applet or not */
        /* If there is a security manager, ask the security manager
           what to do. */
        if (security != null) {
            g = security.getThreadGroup();
        }
        /* If the security doesn't have a strong opinion of the matter
           use the parent thread group. */
        if (g == null) {
            g = parent.getThreadGroup();
        }
    }
    ...
    g.addUnstarted();
    this.group = g;
    this.daemon = parent.isDaemon();
    this.priority = parent.getPriority();
    if (security == null || isCCLOverridden(parent.getClass()))
        this.contextClassLoader = parent.getContextClassLoader();
    else
        this.contextClassLoader = parent.contextClassLoader;
    this.inheritedAccessControlContext =
            acc != null ? acc : AccessController.getContext();
    this.target = target;
    setPriority(priority);
    if (inheritThreadLocals && parent.inheritableThreadLocals != null)
        this.inheritableThreadLocals =
            ThreadLocal.createInheritedMap(parent.inheritableThreadLocals);
    /* Stash the specified stack size in case the VM cares */
    this.stackSize = stackSize;
    /* Set thread ID */
    tid = nextThreadID();
}

Thread创建可继承的Map

java.lang.ThreadLocal#createInheritedMap，基于父线程的ThreadLocalMap创建ThreadLocalMap：java.lang.ThreadLocal.ThreadLocalMap#ThreadLocalMap(java.lang.ThreadLocal.ThreadLocalMap)

private ThreadLocalMap(ThreadLocalMap parentMap) {
    Entry[] parentTable = parentMap.table;
    int len = parentTable.length;
    setThreshold(len);
    table = new Entry[len];
    // 1. 遍历父类table
    for (int j = 0; j < len; j++) {
        Entry e = parentTable[j];
        if (e != null) {
            @SuppressWarnings("unchecked")
            ThreadLocal<Object> key = (ThreadLocal<Object>) e.get();
            // 2. ThreadLocalMap的key是ThreadLocal实例的弱引用，可能会被回收因此需要判空。
            // 如果为空则不会被继承
            if (key != null) {
                // 3. 可继承的ThreadLocal的childValue方法直接返回入参的value（即父类的value）
                Object value = key.childValue(e.value);
                // 4. 创建子value Entry
                Entry c = new Entry(key, value);
                // 5. 复制父类至子线程的ThreadLocalMap
                int h = key.threadLocalHashCode & (len - 1);
                while (table[h] != null)
                    h = nextIndex(h, len);
                table[h] = c;
                size++;
            }
        }
    }
}

总结