com.github.tomato.util.StaticContext Maven / Gradle / Ivy

Go to download
package com.github.tomato.util;

import com.github.tomato.constant.TomatoConstant;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 基于线程的上下文工具,用于获取每个线程中的幂等建键息
 *
 * @author liuxin
 * 2020-01-04 22:11
 */
public final class StaticContext {

    private static final ThreadLocal> CONTEXT = new InheritableThreadLocal<>();

    private static Map getAttributeAsMap() {
        Map stringStringMap = CONTEXT.get();
        if (stringStringMap == null) {
            stringStringMap = new ConcurrentHashMap<>();
            CONTEXT.set(stringStringMap);
        }
        return stringStringMap;
    }

    public static String getToken() {
        return getAttributeAsMap().get(TomatoConstant.TOKEN);
    }

    public static void setToken(String tokenValue) {
        addAttribute(TomatoConstant.TOKEN, tokenValue);
    }

    public static void addAttribute(String key, String value) {
        getAttributeAsMap().put(key, value);
    }

    public static void clear() {
        CONTEXT.remove();
    }
}