1_2/modules/clustering/src/org/apache/axis2/cluster/handlers/ReplicationHandler.java

/*
 * Copyright 2004,2005 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.axis2.cluster.handlers;

import org.apache.axis2.AxisFault;
import org.apache.axis2.cluster.ClusterManager;
import org.apache.axis2.cluster.ClusteringFault;
import org.apache.axis2.cluster.context.ContextManager;
import org.apache.axis2.context.ConfigurationContext;
import org.apache.axis2.context.MessageContext;
import org.apache.axis2.context.ServiceContext;
import org.apache.axis2.context.ServiceGroupContext;
import org.apache.axis2.engine.AxisConfiguration;
import org.apache.axis2.handlers.AbstractHandler;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

public class ReplicationHandler extends AbstractHandler {

    private static final Log log = LogFactory.getLog(ReplicationHandler.class);
    
	public InvocationResponse invoke(MessageContext msgContext) throws AxisFault {

		replicateState(msgContext);

		return InvocationResponse.CONTINUE;
	}

	public void flowComplete(MessageContext msgContext) {
		super.flowComplete(msgContext);

		try {
			replicateState(msgContext);
		} catch (ClusteringFault e) {
			String message = "Could not replicate the state";
			log.error(message);
		}
	}

	private void replicateState(MessageContext message) throws ClusteringFault {

		ConfigurationContext configurationContext = message.getConfigurationContext();
		AxisConfiguration axisConfiguration = configurationContext.getAxisConfiguration();
		ClusterManager clusterManager = axisConfiguration.getClusterManager();

		if (clusterManager != null) {
			
			ContextManager contextManager = clusterManager.getContextManager();
			if (contextManager==null) {
				if (log.isDebugEnabled()) {
					String msg = "Cannot replicate contexts since the ClusterManager has not defined a ContextManager";
					log.error(msg);
				}
			}
			
			ServiceContext serviceContext = message.getServiceContext();
			ServiceGroupContext serviceGroupContext = message.getServiceGroupContext();

			
			contextManager.updateState(configurationContext);

			if (serviceGroupContext != null) {
				contextManager.updateState(serviceGroupContext);
			}

			if (serviceContext != null) {
				contextManager.updateState(serviceContext);
			}

		}

	}

}