一個優(yōu)雅的 ThreadLocal 工具類

Java 版

import java.util.*;

public final class ThreadLocalUtil {
    private static final ThreadLocal<Map<String, Object>> threadLocal = new ThreadLocal() {
        protected Map<String, Object> initialValue() {
            return new HashMap(4);
        }
    };

    public static Map<String, Object> getThreadLocal(){
        return threadLocal.get();
    }
    public static <T> T get(String key) {
        Map map = (Map)threadLocal.get();
        return (T)map.get(key);
    }

    public static <T> T get(String key,T defaultValue) {
        Map map = (Map)threadLocal.get();
        return (T)map.get(key) == null ? defaultValue : (T)map.get(key);
    }

    public static void set(String key, Object value) {
        Map map = (Map)threadLocal.get();
        map.put(key, value);
    }

    public static void set(Map<String, Object> keyValueMap) {
        Map map = (Map)threadLocal.get();
        map.putAll(keyValueMap);
    }

    public static void remove() {
        threadLocal.remove();
    }

    public static <T> Map<String,T> fetchVarsByPrefix(String prefix) {
        Map<String,T> vars = new HashMap<>();
        if( prefix == null ){
            return vars;
        }
        Map map = (Map)threadLocal.get();
        Set<Map.Entry> set = map.entrySet();

        for( Map.Entry entry : set ){
            Object key = entry.getKey();
            if( key instanceof String ){
                if( ((String) key).startsWith(prefix) ){
                    vars.put((String)key,(T)entry.getValue());
                }
            }
        }
        return vars;
    }

    public static <T> T remove(String key) {
        Map map = (Map)threadLocal.get();
        return (T)map.remove(key);
    }

    public static void clear(String prefix) {
        if( prefix == null ){
            return;
        }
        Map map = (Map)threadLocal.get();
        Set<Map.Entry> set = map.entrySet();
        List<String> removeKeys = new ArrayList<>();

        for( Map.Entry entry : set ){
            Object key = entry.getKey();
            if( key instanceof String ){
                if( ((String) key).startsWith(prefix) ){
                    removeKeys.add((String)key);
                }
            }
        }
        for( String key : removeKeys ){
            map.remove(key);
        }
    }
}

Kotlin 版

import java.util.ArrayList
import java.util.HashMap

open class ThreadLocalUtil {

    companion object {

        private val threadLocal = object : ThreadLocal<Map<String, Any>>() {
            protected override fun initialValue(): Map<String, Any> {
                return HashMap(4)
            }
        }

        fun getThreadLocal(): Map<String, Any> {
            return threadLocal.get()
        }

        operator fun <T> get(key: String): T {
            val map: Map<String, Any>  = threadLocal.get() as HashMap<String, Any>
            return map[key] as T
        }

        operator fun <T> get(key: String, defaultValue: T): T {
            val map = threadLocal.get() as HashMap<String, Any>
            return if (map[key] as T == null) defaultValue else map[key] as T
        }

        operator fun set(key: String, value: Any) {
            val amap = threadLocal.get() as HashMap<String, Any>
            amap.put(key, value)
        }

        fun set(keyValueMap: Map<String, Any>) {
            val map = threadLocal.get() as HashMap<String, Any>
            map.putAll(keyValueMap)
        }

        fun remove() {
            threadLocal.remove()
        }

        fun <T> fetchVarsByPrefix(prefix: String?): Map<String, T> {
            val vars = HashMap<String, T>()
            if (prefix == null) {
                return vars
            }
            val map = threadLocal.get() as HashMap<String, Any>
            val set = map.entries

            for ((key, value) in set) {
                if (key is String) {
                    if (key.startsWith(prefix)) {
                        vars.put(key, value as T)
                    }
                }
            }
            return vars
        }

        fun <T> remove(key: String): T {
            val map = threadLocal.get() as HashMap<String, Any>
            return map.remove(key) as T
        }

        fun clear(prefix: String?) {
            if (prefix == null) {
                return
            }
            val map = threadLocal.get() as HashMap<String, Any>
            val set = map.entries
            val removeKeys = ArrayList<String>()

            for ((key) in set) {
                if (key is String) {
                    if (key.startsWith(prefix)) {
                        removeKeys.add(key)
                    }
                }
            }
            for (key in removeKeys) {
                map.remove(key)
            }
        }
    }

}
?著作權歸作者所有,轉載或內容合作請聯系作者
  • 序言:七十年代末放妈,一起剝皮案震驚了整個濱河市柠衅,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖雏吭,帶你破解...
    沈念sama閱讀 212,718評論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現場離奇詭異处铛,居然都是意外死亡熟吏,警方通過查閱死者的電腦和手機,發(fā)現死者居然都...
    沈念sama閱讀 90,683評論 3 385
  • 文/潘曉璐 我一進店門娶吞,熙熙樓的掌柜王于貴愁眉苦臉地迎上來垒迂,“玉大人,你說我怎么就攤上這事妒蛇』希” “怎么了?”我有些...
    開封第一講書人閱讀 158,207評論 0 348
  • 文/不壞的土叔 我叫張陵绣夺,是天一觀的道長吏奸。 經常有香客問我,道長陶耍,這世上最難降的妖魔是什么奋蔚? 我笑而不...
    開封第一講書人閱讀 56,755評論 1 284
  • 正文 為了忘掉前任,我火速辦了婚禮烈钞,結果婚禮上旺拉,老公的妹妹穿的比我還像新娘产上。我一直安慰自己棵磷,他們只是感情好蛾狗,可當我...
    茶點故事閱讀 65,862評論 6 386
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著仪媒,像睡著了一般沉桌。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上算吩,一...
    開封第一講書人閱讀 50,050評論 1 291
  • 那天留凭,我揣著相機與錄音,去河邊找鬼偎巢。 笑死蔼夜,一個胖子當著我的面吹牛,可吹牛的內容都是我干的压昼。 我是一名探鬼主播求冷,決...
    沈念sama閱讀 39,136評論 3 410
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼窍霞!你這毒婦竟也來了匠题?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 37,882評論 0 268
  • 序言:老撾萬榮一對情侶失蹤但金,失蹤者是張志新(化名)和其女友劉穎韭山,沒想到半個月后,有當地人在樹林里發(fā)現了一具尸體冷溃,經...
    沈念sama閱讀 44,330評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡钱磅,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 36,651評論 2 327
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現自己被綠了似枕。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片盖淡。...
    茶點故事閱讀 38,789評論 1 341
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖菠净,靈堂內的尸體忽然破棺而出禁舷,到底是詐尸還是另有隱情,我是刑警寧澤毅往,帶...
    沈念sama閱讀 34,477評論 4 333
  • 正文 年R本政府宣布牵咙,位于F島的核電站,受9級特大地震影響攀唯,放射性物質發(fā)生泄漏洁桌。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 40,135評論 3 317
  • 文/蒙蒙 一侯嘀、第九天 我趴在偏房一處隱蔽的房頂上張望另凌。 院中可真熱鬧谱轨,春花似錦、人聲如沸吠谢。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,864評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽工坊。三九已至献汗,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間王污,已是汗流浹背罢吃。 一陣腳步聲響...
    開封第一講書人閱讀 32,099評論 1 267
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留昭齐,地道東北人尿招。 一個月前我還...
    沈念sama閱讀 46,598評論 2 362
  • 正文 我出身青樓,卻偏偏與公主長得像阱驾,于是被迫代替她去往敵國和親就谜。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 43,697評論 2 351

推薦閱讀更多精彩內容