package org.dromara.common.core.context;

import com.alibaba.ttl.TransmittableThreadLocal;

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

/**
 * 线程持有类
 *
 * @author Michelle.Chung
 */
public class ThreadLocalHolder {

    /**
     * 初始化 (支持异步)
     */
    private static final ThreadLocal<Map<String, Object>> THREAD_LOCAL = TransmittableThreadLocal.withInitial(HashMap::new);

    /**
     * 设置值
     *
     * @param key 键
     * @param value 值
     */
    public static <T> void set(String key, T value) {
        THREAD_LOCAL.get().put(key, value);
    }

    /**
     * 获取值
     *
     * @param key 键
     * @return 值
     */
    @SuppressWarnings("unchecked")
    public static <T> T get(String key) {
        return (T) THREAD_LOCAL.get().get(key);
    }

    /**
     * 移除值
     *
     * @param key 键
     */
    public static void remove(String key) {
        THREAD_LOCAL.get().remove(key);
    }

    /**
     * 清空值
     */
    public static void clear() {
        THREAD_LOCAL.remove();
    }

}