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

但是前端同學(xué)說(shuō)傳了token了，那為什么還會(huì)報(bào)token為空的錯(cuò)誤呢

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

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：?jiǎn)栴}復(fù)現(xiàn)

我們寫一個(gè)簡(jiǎn)單的測(cè)試

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

可以看到可以拿到我們?cè)O(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());
}

我們的問(wèn)題復(fù)現(xiàn)了，報(bào)了token為空的錯(cuò)誤。但是這還是隨機(jī)會(huì)出現(xiàn)的情況，并不是每次都會(huì)出現(xiàn)。所以導(dǎo)致我們?cè)谡{(diào)試的時(shí)候并沒(méi)有出現(xiàn)這個(gè)問(wèn)題

3：分析問(wèn)題

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

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

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

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

4：如何解決

解決辦法就很簡(jiǎn)單了，只需要把ThreadLocalUtil.getUserId()單獨(dú)拿出來(lái)執(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ú)拿出來(lái)執(zhí)行
    Integer userId = ThreadLocalUtil.getUserId();
    list.parallelStream().filter(x -> x.equals(userId)).collect(Collectors.toList());
}