CurrentThreadContext.java

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.HashMap;

/**
 * User: ehaskah
 * Date: 26/10/11
 * Time: 22:29
 */
public class CurrentThreadContext {
    private static final Logger logger = LoggerFactory.getLogger(CurrentThreadContext.class);

    private static final ThreadLocal<HashMap<String, Object>> SCOPE_CONTEXT = new ThreadLocal<HashMap<String, Object>>();

    private CurrentThreadContext() {
    }

    /**
     * Developer is supposed to call this method at the end of current transaction or at the beginning of a new transaction
     *
     * It is the developer's responsibility to call this method at the right time.
     */
    public static void clear() {
        // logger.debug("Clearing thread context");

        HashMap map = SCOPE_CONTEXT.get();
        if (map != null) {
            map.clear();
        }
        SCOPE_CONTEXT.remove();
    }

    public static void put(String key, Object value) {
        HashMap<String, Object> map = SCOPE_CONTEXT.get();
        if (map == null) {
            map = new HashMap<>();
            SCOPE_CONTEXT.set(map);
        }
        map.put(key, value);
    }

    public static Object get(String key) {
        HashMap map = SCOPE_CONTEXT.get();
        if (map == null) {
            return null;
        }
        return map.get(key);
    }
}