简单实现一个ThreadLocal

原创于 2018-07-13 16:05:07 发布 · 294 阅读

0 ·

CC 4.0 BY-SA版权

线程安全专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一个简单的ThreadLocal实现，通过使用线程安全的Map来存储每个线程的局部变量。文章详细展示了如何设置、获取和移除特定线程的数据，并提供了一个初始值的默认实现。

package com.smart.threadlocal;

import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
/**
 * 简单线程本地化
 * @author lenovo
 *
 */
public class SimpleThreadLocal {
    //定义一个线程安全的集合(key为线程,value为值)
    private Map<Thread,Object> valueMap = Collections.synchronizedMap(new HashMap<Thread,Object>());

    public void set(Object newValue) {
        valueMap.put(Thread.currentThread(), newValue);
    }

    public Object get() {
        Thread currentThread = Thread.currentThread();
        Object o =valueMap.get(currentThread);

        if(o == null && !valueMap.containsKey(currentThread)) {
             o =initialValue();
             valueMap.put(currentThread, o);
        }
        return o;
    }

    public void remove() {
        valueMap.remove(Thread.currentThread());
    }
    public Object initialValue() {
        return null;
    }

}