com.github.valfirst.slf4jtest.TestMDCAdapter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of slf4j-test Show documentation

An in-memory SLF4J implementation focused on aiding unit testing.

There is a newer version: 3.0.1

package com.github.valfirst.slf4jtest;

import com.google.common.collect.ImmutableMap;
import java.util.HashMap;
import java.util.Map;
import org.slf4j.spi.MDCAdapter;

public class TestMDCAdapter implements MDCAdapter {

    private final ThreadLocal> value = ThreadLocal.withInitial(HashMap::new);

    public void put(final String key, final String val) {
        value.get().put(key, String.valueOf(val));
    }

    public String get(final String key) {
        return value.get().get(key);
    }

    public void remove(final String key) {
        value.get().remove(key);
    }

    public void clear() {
        value.get().clear();
    }

    public ImmutableMap getCopyOfContextMap() {
        return ImmutableMap.copyOf(value.get());
    }

    @SuppressWarnings("unchecked")
    public void setContextMap(final Map contextMap) {
        value.set(new HashMap(contextMap));
    }
}