io.tracee.backend.slf4j.Slf4jTraceeBackend Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tracee-core Show documentation

Please refer to https://github.com/tracee/tracee.

The newest version!

package io.tracee.backend.slf4j;

import io.tracee.BackendBase;
import org.slf4j.MDC;

import java.util.HashMap;
import java.util.Map;
import java.util.Set;

class Slf4jTraceeBackend extends BackendBase {

	/**
	 * This set contains all MDC-Keys managed by tracee.
	 * This bookkeeping is required to ensure that operations like {@link Slf4jTraceeBackend#clear()} do not remove
	 * TracEE unrelated keys from the MDC.
	 */
	protected final ThreadLocal> traceeKeys;

	Slf4jTraceeBackend(ThreadLocal> traceeKeys) {
		this.traceeKeys = traceeKeys;
	}

	@Override
    public boolean containsKey(String key) {
		return key != null && traceeKeys.get().contains(key) && MDC.get(key) != null;
    }

	@Override
    public int size() {
        return traceeKeys.get().size();
    }

	@Override
    public boolean isEmpty() {
        return traceeKeys.get().isEmpty();
    }

	@Override
    public String get(String key) {
        if ((key != null) && traceeKeys.get().contains(key))
			return MDC.get(key);
        else
            return null;
    }

	@Override
    public void put(String key, String value) {
        if (key == null) throw new NullPointerException("null keys are not allowed.");
        if (value == null) throw new NullPointerException("null values are not allowed.");
        final Set registeredKeys = traceeKeys.get();
        if (!registeredKeys.contains(key)) {
            registeredKeys.add(key);
        }
		MDC.put(key, value);
	}

	@Override
    public void remove(String key) {
        if (key == null) throw new NullPointerException("null keys are not allowed.");
        if (traceeKeys.get().remove(key)) {
			MDC.remove(key);
		}
    }

    @Override
    public void clear() {
        for (String key : traceeKeys.get()) {
			MDC.remove(key);
        }
        traceeKeys.remove();
    }

	@Override
    public void putAll(Map entries) {
        for (Map.Entry entry : entries.entrySet()) {
            put(entry.getKey(), entry.getValue());
        }
    }

	@Override
	public Map copyToMap() {
		final Map traceeMap = new HashMap();
		final Set keys = traceeKeys.get();
		for (String traceeKey : keys) {
			final String value = MDC.get(traceeKey);
			if (value != null) {
				traceeMap.put(traceeKey, value);
			}
		}
		return traceeMap;
	}
}