src/org/apache/kandula/storage/StorageUtils.java

/*
 * Copyright  2004 The Apache Software Foundation.
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 */
package org.apache.kandula.storage;

import org.apache.axis2.context.ConfigurationContext;
import org.apache.axis2.context.MessageContext;
import org.apache.kandula.Constants;
import org.apache.kandula.context.AbstractContext;

public class StorageUtils {
	public static AbstractContext getContext(String id)
	{
		Store store = getStore(MessageContext.getCurrentMessageContext().getServiceContext().getConfigurationContext());
		return (AbstractContext)store.get(id);
	}
	
	public static void putContext(AbstractContext context)
	{
		Store store = getStore(MessageContext.getCurrentMessageContext().getServiceContext().getConfigurationContext());
		store.put(context.getCoordinationContext().getActivityID(),context);
	}
	
	public static void putContext(AbstractContext context, String id, MessageContext  messageContext)
	{
		Store store = getStore(messageContext.getServiceContext().getConfigurationContext());
		store.put(id,context);
	}
	
	public static Store getStore(ConfigurationContext configurationContext) {
		if (configurationContext == null)
			return null;
		Store store;
		Object storeObect =  configurationContext
				.getProperty(Constants.KANDULA_STORE);
		if (storeObect == null) {
			store = new SimpleStore();
			configurationContext.setProperty(Constants.KANDULA_STORE, store);
		}else{
			store = (Store)storeObect;
		}
		return store;
	}
}