关于ThreadLocal相对合理的打开方式

• 一、旧的方式
• 二、新的方式

ThreadLocal这个类的原理想必大家都不陌生，这里不再对其原理进行过多的阐述，本文从程序设计层面来讨论一下对它的合理使用。通常网上以及实际业务开发过程中对ThreadLocal的使用都会自己实现一个工具类来方便项目中其他开发人员来使用，而且这个工具类都是比较通用的。大致见以下代码：

public final class ThreadLocalUtil {

    
    private static final ThreadLocal> THREAD_LOCAL = ThreadLocal.withInitial(() -> new HashMap<>(16));

    private ThreadLocalUtil() {}

    
    public static Map getThreadLocal() {
        return THREAD_LOCAL.get();
    }

    
    public static  T get(String key) {
        Map map = THREAD_LOCAL.get();
        return (T)map.get(key);
    }

    
    public static  T get(String key, T defaultValue) {
        Map map = THREAD_LOCAL.get();
        return map.get(key) == null ? defaultValue : (T)map.get(key);
    }

    
    public static void set(String key, Object value) {
        Map map = THREAD_LOCAL.get();
        map.put(key, value);
    }

    
    public static void set(Map keyValueMap) {
        Map map = THREAD_LOCAL.get();
        map.putAll(keyValueMap);
    }

    
    public static  Map fetchVarsByPrefix(String prefix) {
        Map vars = new HashMap<>();
        if( prefix == null ){
            return vars;
        }
        Map map = THREAD_LOCAL.get();
        Set set = map.entrySet();

        for( Map.Entry entry : set ){
            Object key = entry.getKey();
            if( key instanceof String && ((String) key).startsWith(prefix)){
                    vars.put((String)key,(T)entry.getValue());
            }
        }
        return vars;
    }

    
    public static  T remove(String key) {
        Map map = THREAD_LOCAL.get();
        return (T)map.remove(key);
    }

    
    public static void clear(String prefix) {
        if( prefix == null ){
            return;
        }
        Map map = THREAD_LOCAL.get();
        Set set = map.entrySet();
        List removeKeys = new ArrayList<>();

        for( Map.Entry entry : set ){
            Object key = entry.getKey();
            if( key instanceof String && ((String) key).startsWith(prefix)){
                    removeKeys.add((String)key);
            }
        }

        for( String key : removeKeys ){
            map.remove(key);
        }
    }

    
    public static void remove() {
        THREAD_LOCAL.remove();
    }

上面代码的优点：通用，几乎所有不同业务的代码都可以用这个类，简单。缺点：ThreadLocal泛型没有利用到，导致通过工具类返回的结果都是Object类型，虽然可以强转，但是我觉得不优雅而且不够规范合理。涉及到业务相关的就不应该总想着往通用的层面靠拢

直接参考Spring团队RequestContextHolder[1]这个类对threadlocal的实现方式。这里不直接使用ThreadLocal这个类，而是使用Spring给我提供的一个可命名的NamedThreadLocal（当然也可以使用原生的ThreadLocal，效果一样）。这里我们可以看到这种设计针对不同的业务需要往本地线程副本里面存的内容通过泛型指定好了， 整个一下来，代码并没有任何由于强转导致的黄色代码警告，而且和hystrix线程隔离有异曲同工的地方。每个线程副本管理的东西都非常清楚。

public abstract class RequestUserInfoHolder {

    private static final ThreadLocal userInfoHolder =
            new NamedThreadLocal<>("Request UserInfo");

    
    public static void resetRequestUserInfo() {
        userInfoHolder.remove();
    }

    public static void setRequestUserInfo(@Nullable UserInfoDTO userInfo) {
        if (userInfo == null) {
            resetRequestUserInfo();
        } else {
            userInfoHolder.set(userInfo);
        }
    }


    @Nullable
    public static UserInfoDTO getRequestUserInfo() {
        return userInfoHolder.get();
    }


    
    @NonNull
    public static UserInfoDTO currentRequestUserInfo() {
        UserInfoDTO info = getRequestUserInfo();
        if (info == null) {
            ThrowableUtil.commonException(ErrorCodeEnum.LOGIN_NONE);
        }
        return info;
    }
    
}