快捷導(dǎo)航

解決ThreadLocal獲取不到值大坑

更新時間：2023年05月08日 11:09:53 作者：我是小趴菜

這篇文章主要介紹了解決ThreadLocal獲取不到值大坑

1：問題起因

今天項(xiàng)目上測試環(huán)境，再給領(lǐng)導(dǎo)演示的時候出現(xiàn)了bug，很尷尬。于是我跟前端同學(xué)通過模擬請求，最后發(fā)現(xiàn)在調(diào)一個接口的時候返回了一個 token為空的錯誤。

但是前端同學(xué)說傳了token了，那為什么還會報token為空的錯誤呢

我們項(xiàng)目使用的JWT生成用戶token，每次請求都要經(jīng)過攔截器校驗(yàn)。因?yàn)樵谡埱蟮臅r候我們經(jīng)常需要用到當(dāng)前用戶登錄的ID，所以我們使用到了ThhreadLocal這個工具類。

Map<String, Claim> claimMap = JwtUtil.verifyToken(headerToken);
if (null == claimMap) {
    throw new GlobalException(ResponseEnums.TOKEN_INVALID_ERROR);
}
//將參數(shù)放入上下文中
Map<String, Object> result = new HashMap<>();
Set<Map.Entry<String, Claim>> entrySet = claimMap.entrySet();
for (Map.Entry<String, Claim> claimEntry : entrySet) {
    result.put(claimEntry.getKey(), claimEntry.getValue().asString());
}
//將用戶ID存到ThreadLocal中，以便后續(xù)的獲取使用
ThreadLocalUtil.getInstance().setContext(result);

這里也貼上ThreadLocalUtil工具類代碼

@Slf4j
public class ThreadLocalUtil {
    private static final ThreadLocal<Map<String, Object>> CONTEXT = new ThreadLocal<>();
    private ThreadLocalUtil() {}
    public static ThreadLocalUtil getInstance() {
        return SingletonHolder.INSTANCE;
    }
    private static class SingletonHolder {
        private static final ThreadLocalUtil INSTANCE = new ThreadLocalUtil();
    }
    public void setContext(Map<String, Object> map) {
        CONTEXT.set(map);
    }
    public static Map<String, Object> getContext() {
        return CONTEXT.get();
    }
    public void clear() {
        CONTEXT.remove();
    }
    public static Integer getUserId() {
        Map<String, Object> context = getContext();
        if(context == null || !context.containsKey(JwtUtil.USER_ID)) {
            throw new GlobalException(ResponseEnums.TOKEN_IS_NULL_ERROR);
        }
        return Integer.parseInt(String.valueOf(context.get(JwtUtil.USER_ID)));
    }
}

2：問題復(fù)現(xiàn)

我們寫一個簡單的測試

public static void main(String[] args) {
    HashMap<String,Object> map = new HashMap<>();
    map.put(JwtUtil.USER_ID,"1");
    ThreadLocalUtil.getInstance().setContext(map);
    System.out.println(ThreadLocalUtil.getUserId());
}

可以看到可以拿到我們設(shè)置的值。

但是如果將ThreadLocal跟Java8的Stream一起配合使用呢？

public static void main(String[] args) {
    HashMap<String,Object> map = new HashMap<>();
    map.put(JwtUtil.USER_ID,"1");
    ThreadLocalUtil.getInstance().setContext(map);
    List<Integer> list = new ArrayList<>();
    list.add(1);
    list.add(2);
    list.add(3);
    //ThreadLocal獲取值放在stream里面執(zhí)行
    list.parallelStream().filter(x -> x.equals(ThreadLocalUtil.getUserId())).collect(Collectors.toList());
}

我們的問題復(fù)現(xiàn)了，報了token為空的錯誤。但是這還是隨機(jī)會出現(xiàn)的情況，并不是每次都會出現(xiàn)。所以導(dǎo)致我們在調(diào)試的時候并沒有出現(xiàn)這個問題

3：分析問題

咋一看并不能知道為什么會這樣，所以我在獲取用戶id的打印了一下日志

看出問題了吧，竟然有三個線程來獲取，因?yàn)槲覀冊O(shè)置值的線程就是main線程，所以前面二個線程獲取到的值就是空的，所以就拋出了異常

所以現(xiàn)在只需要知道這個 ForkJoinPool是在哪就好了，最終在翻看源碼，找到原來就是在jdk8的Stream里面。

這是為什么呢？因?yàn)镴dk8的Stream底層使用了ForkJoinPool線程池，這就導(dǎo)致當(dāng)我們調(diào)用 ThreadLocalUtil.getUserId()的時候，是直接提交到了ForkJoinPool線程池中去了，這時候就會有其它線程去調(diào)用這個方法，所以就拿不到值了

4：如何解決

解決辦法就很簡單了，只需要把ThreadLocalUtil.getUserId()單獨(dú)拿出來執(zhí)行就可以了

public static void main(String[] args) {
    HashMap<String,Object> map = new HashMap<>();
    map.put(JwtUtil.USER_ID,"1");
    ThreadLocalUtil.getInstance().setContext(map);
    List<Integer> list = new ArrayList<>();
    list.add(1);
    list.add(2);
    list.add(3);
    //單獨(dú)拿出來執(zhí)行
    Integer userId = ThreadLocalUtil.getUserId();
    list.parallelStream().filter(x -> x.equals(userId)).collect(Collectors.toList());
}