ThreadLocal工具类-优快云博客

本文链接：https://blog.youkuaiyun.com/alashanMt/article/details/126035355

这个Java类实现了ThreadLocal来创建线程局部变量的工具类，主要用于存储线程安全的缓存。类中提供了获取、设置、移除缓存的方法，确保每个线程拥有独立的变量副本，避免数据冲突。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package com.demo.util;

import cn.hutool.core.collection.CollUtil;

import java.util.HashMap;
import java.util.Map;
import java.util.Objects;

/**
 * @author: mt
 * @description: 线程变量工具类
 * @create: 16:50 2022/7/26
 */
public class ThreadLocalUtil {

    private static final ThreadLocal<Map<Object, Object>> MAP_THREAD_LOCAL = ThreadLocal.withInitial(HashMap::new);

    private static Map<Object, Object> cacheMap;

    static {
        cacheMap = MAP_THREAD_LOCAL.get();
    }

    public static Map<Object, Object> getThreadLocal() {
        return MAP_THREAD_LOCAL.get();
    }

    public static <T> T get(Object key) {
        if (CollUtil.isEmpty(cacheMap)) {
            return null;
        }
        return (T) cacheMap.get(key);
    }

    public static <T> T get(Object key, Object defaultValue) {
        if (CollUtil.isEmpty(cacheMap)) {
            return (T) defaultValue;
        }
        return (T) cacheMap.get(key);
    }

    public static Object set(Object key, Object value) {
        return cacheMap.put(key, value);
    }

    public static void set(Map<Object, Object> map) {
        cacheMap.putAll(map);
    }

    public static void removeValue(Object key) {
        MAP_THREAD_LOCAL.remove();
        if (!Objects.isNull(cacheMap.get(key))) {
            cacheMap.remove(key);
        }
    }
}