com.technophobia.substeps.runner.ExecutionContext Maven / Gradle / Ivy

Go to download
/*
 *	Copyright Technophobia Ltd 2012
 *
 *   This file is part of Substeps.
 *
 *    Substeps is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU Lesser General Public License as published by
 *    the Free Software Foundation, either version 3 of the License, or
 *    (at your option) any later version.
 *
 *    Substeps is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU Lesser General Public License for more details.
 *
 *    You should have received a copy of the GNU Lesser General Public License
 *    along with Substeps.  If not, see .
 */
package com.technophobia.substeps.runner;

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

import com.technophobia.substeps.model.Scope;

/**
 * Singleton class holding state a various scopes for the duration of execution,
 * state will be cleared at appropriate points through execution
 * 
 * @author imoore
 * 
 */
public final class ExecutionContext {

	private static final ThreadLocal executionContextThreadLocal = new ThreadLocal() {
		@Override
		protected ExecutionContext initialValue() {
			return new ExecutionContext();
		};
	};

	private ExecutionContext() {
		scopedData = new HashMap>();
	}

	private Map> scopedData = null;

	public static void put(final Scope scope, final String key, final Object value) {
		executionContextThreadLocal.get().putInternal(scope, key, value);
	}

	public static Object get(final Scope scope, final String key) {
		return executionContextThreadLocal.get().getInternal(scope, key);
	}

	private void putInternal(final Scope scope, final String key, final Object value) {
		// NB. this is not currently synchronised

		Map map = scopedData.get(scope);
		if (map == null) {
			map = new HashMap();
			scopedData.put(scope, map);
		}
		map.put(key, value);
	}

	private Object getInternal(final Scope scope, final String key) {
		Object rtn = null;
		final Map map = scopedData.get(scope);
		if (map != null) {
			rtn = map.get(key);
		}
		return rtn;
	}

	public static void clear(final Scope scope) {
		executionContextThreadLocal.get().scopedData.remove(scope);
	}
}