org.ocelotds.context.ThreadLocalContextHolder Maven / Gradle / Ivy

Go to download

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

There is a newer version: 2.12.0

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
package org.ocelotds.context;

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

/**
 *
 * @author hhfrancois
 */
public class ThreadLocalContextHolder {

	private static final ThreadLocal> THREAD_WITH_CONTEXT = new ThreadLocal<>();

	ThreadLocalContextHolder() {
	}

	public static void put(String key, Object payload) {
		Map context = THREAD_WITH_CONTEXT.get();
		if (context == null) {
			context = new HashMap<>();
			THREAD_WITH_CONTEXT.set(context);
		}
		if(null == payload) {
			if(context.containsKey(key)) {
				context.remove(key);
			}
		} else {
			context.put(key, payload);
		}
	}

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

	public static void cleanupThread() {
		THREAD_WITH_CONTEXT.remove();
	}

}