ThreadLocal应用场景及源码分析

原创/朱季谦

在登录模块里，当用户完成一次登录会话后，往往需要将其登录成功的信息进行缓存。不同的登录会话，属于不同的会话线程，彼此需要互不影响。这就意味着，登录成功的信息，只属于该次会话线程的本地变量，这时，就可以基于Th readLo cal缓存属于该会话线程的用户信息，类似线程的私有本地变量。

可以这样实现，先定义一个存储用户信息的POJO类——

@Data
public class User {
    private String username;
    private String role;
    ......
}

写一个专门缓存会话用户信息的类Sess ionCsc he，在该类里定义一个 static 修饰的ThreadLo cal变量，再定义一个将User信息缓存到ThreadLo cal的方法，以及从ThreadLo cal取出的方法——

public class SessionCsche {
    //定义一个static修饰的ThreadLocal变量
    private static ThreadLocal&lt;User&gt; threadCache = new ThreadLocal&lt;&gt;();

    //缓存用户信息
    public static void put(String username, String role) {
        User user = new User();
        user.setUsername(username);
        user.setRole(role);
        threadCache.set(user);
    }

  //读取缓存在该会话线程里的用户信息
    public static String getUsername() {
        User user = threadCache.get();
        if (user == null) {
            return null;
        }
        return user.getUsername();
    }
}

以上，就是基于ThreadLocal实现一个线程隔离缓存变量的场景。除此之外，在DataSource 数据源动态切换场景下，也可以基于ThreadLocal实现不同的线程获取到不同的DataSource数据源，进而发起一次数据库访问会话。

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

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

ThreadLocalMap(ThreadLocal<?> firstKey, Object firstValue) {
    table = new Entry[INITIAL_CAPACITY];
    int i = firstKey.threadLocalHashCode &amp; (INITIAL_CAPACITY - 1);
    table[i] = new Entry(firstKey, firstValue);
    size = 1;
    setThreshold(INITIAL_CAPACITY);
}

static class Entry extends WeakReference<ThreadLocal<?>> {
    /** The value associated with this ThreadLocal. */
    Object value;

    Entry(ThreadLocal<?> k, Object v) {
        super(k);
        value = v;
    }
}

  User user = threadCache.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();
}