Package org.apache.axis2.dispatchers

Source Code of org.apache.axis2.dispatchers.HTTPLocationBasedDispatcher

/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.dispatchers;

import org.apache.axis2.AxisFault;
import org.apache.axis2.Constants;
import org.apache.axis2.context.MessageContext;
import org.apache.axis2.description.AxisBinding;
import org.apache.axis2.description.AxisBindingOperation;
import org.apache.axis2.description.AxisEndpoint;
import org.apache.axis2.description.AxisOperation;
import org.apache.axis2.description.AxisService;
import org.apache.axis2.description.HandlerDescription;
import org.apache.axis2.description.WSDL2Constants;
import org.apache.axis2.engine.AbstractDispatcher;
import org.apache.axis2.transport.http.HTTPConstants;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import java.util.Collection;
import java.util.Iterator;
import java.util.Map;

/**
* Dispatches the operation based on the information from the target endpoint URL.
*/
public class HTTPLocationBasedDispatcher extends AbstractDispatcher {

    public static final String NAME = "HTTPLocationBasedDispatcher";
    private static final Log log = LogFactory.getLog(HTTPLocationBasedDispatcher.class);

    /*
     *  (non-Javadoc)
     * @see org.apache.axis2.engine.AbstractDispatcher#findOperation(org.apache.axis2.description.AxisService, org.apache.axis2.context.MessageContext)
     */
    public AxisOperation findOperation(AxisService service, MessageContext messageContext)
            throws AxisFault {

        AxisService axisService = messageContext.getAxisService();
        if (axisService != null && messageContext.getTo() != null) {
            String uri = messageContext.getTo().getAddress();
            String httpLocation = parseRequestURL(uri, messageContext
                    .getConfigurationContext().getServiceContextPath());
            String httpMethod = (String) messageContext.getProperty(HTTPConstants.HTTP_METHOD);

            if (httpLocation != null) {
                httpLocation = httpMethod + httpLocation;
                AxisEndpoint axisEndpoint = (AxisEndpoint) messageContext
                        .getProperty(WSDL2Constants.ENDPOINT_LOCAL_NAME);
                // Here we check whether the request was dispatched to the correct endpoint. If it
                // was we can dispatch the operation using the HTTPLocationDispatcher table of that
                // specific endpoint. In most cases we wont be able to do this. So as a last resort
                // iterate through the endpoint map and try to dispatch the operation.
                if (axisEndpoint != null) {
                    Map httpLocationTable = (Map) axisEndpoint.getBinding()
                            .getProperty(WSDL2Constants.HTTP_LOCATION_TABLE);
                    if (httpLocationTable != null) {
                        return getOperationFromHTTPLocation(httpLocation, httpLocationTable);
                    }
                } else {
                    Map endpoints = axisService.getEndpoints();
                    if (endpoints != null) {
                        Iterator iterator = endpoints.values().iterator();
                        while (iterator.hasNext()) {
                            AxisEndpoint endpoint = (AxisEndpoint) iterator.next();
                            Map httpLocationTable = (Map) endpoint.getBinding()
                                    .getProperty(WSDL2Constants.HTTP_LOCATION_TABLE);
                            if (httpLocationTable != null) {
                                AxisOperation axisOperation =
                                        getOperationFromHTTPLocation(httpLocation,
                                                                     httpLocationTable);
                                if (axisOperation != null) {
                                    messageContext.setProperty(WSDL2Constants.ENDPOINT_LOCAL_NAME,
                                                               endpoint);
                                    AxisBinding axisBinding = endpoint.getBinding();
                                    if (axisBinding != null) {
                                        AxisBindingOperation axisBindingOperation =
                                                (AxisBindingOperation) axisBinding
                                                        .getChild(axisOperation.getName());
                                        messageContext.setProperty(Constants.AXIS_BINDING_OPERATION,
                                                                   axisBindingOperation);
                                        return axisOperation;
                                    }
                                }
                            }
                        }
                    }
                }
            } else {
                log.debug("Attempt to check for Operation using HTTP Location failed");
                return null;
            }
        }
        return null;
    }

    /*
     *  (non-Javadoc)
     * @see org.apache.axis2.engine.AbstractDispatcher#findService(org.apache.axis2.context.MessageContext)
     */
    public AxisService findService(MessageContext messageContext) throws AxisFault {
        // This Dispatcher does not need to resolve the service, as that is handled
        // by the RequestURIBasedDispatcher.
        return null;
    }

    public void initDispatcher() {
        init(new HandlerDescription(NAME));
    }

    private String parseRequestURL(String path, String servicePath) {

        int index = path.lastIndexOf(servicePath);
        String service = null;

        if (-1 != index) {
            int serviceStart = index + servicePath.length();
            if (path.length() > serviceStart + 1) {
                service = path.substring(serviceStart + 1);
            }
        }

        if (service != null) {
            index = service.indexOf("/");
            if (-1 != index) {
                service = service.substring(index);
            } else {
                int queryIndex = service.indexOf("?");
                if (queryIndex != -1) {
                    service = service.substring(queryIndex);
                }
            }
        }
        return service;
    }

    /**
     * Given the requestPath that the request came to his method returns the corresponding axisOperation
     *
     * @param requestPath - Part of the request url which is the part after the service name
     * @param httpLocationTable - The httpLocationTable stored in the relavant binding
     * @return AxisOperation - The corresponding AxisOperation
     */
    private AxisOperation getOperationFromHTTPLocation(String requestPath, Map httpLocationTable) {

        Collection httpLocations = httpLocationTable.keySet();
        Iterator iter = httpLocations.iterator();
        while (iter.hasNext()) {
            String location = (String) iter.next();
            int index = requestPath.indexOf(location);
            if (index == 0) {
                return (AxisOperation) httpLocationTable.get(location);
            }

        }
        return null;
    }
}
TOP

Related Classes of org.apache.axis2.dispatchers.HTTPLocationBasedDispatcher

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.