com.liferay.httpservice.internal.servlet.BundleRequestDispatcher.java Source code

Introduction

Here is the source code for com.liferay.httpservice.internal.servlet.BundleRequestDispatcher.java

Source

/**
 * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
 *
 * This library 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 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library 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.
 */

package com.liferay.httpservice.internal.servlet;

import com.liferay.portal.kernel.util.JavaConstants;
import com.liferay.portal.kernel.util.StringPool;
import com.liferay.portal.kernel.util.StringUtil;
import com.liferay.portal.util.ClassLoaderUtil;

import java.io.IOException;
import java.io.UnsupportedEncodingException;

import java.net.URLDecoder;

import javax.servlet.RequestDispatcher;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletRequestEvent;
import javax.servlet.ServletRequestListener;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;

/**
 * @author Raymond Aug
 * @author Miguel Pastor
 */
public class BundleRequestDispatcher implements RequestDispatcher {

    public BundleRequestDispatcher(String servletMapping, boolean extensionMapping, String requestURI,
            BundleServletContext bundleServletContext, BundleFilterChain bundleFilterChain) {

        _servletMapping = servletMapping;
        _extensionMapping = extensionMapping;
        _requestURI = StringUtil.replace(requestURI, StringPool.DOUBLE_SLASH, StringPool.SLASH);
        _bundleServletContext = bundleServletContext;
        _bundleFilterChain = bundleFilterChain;

        if (!_extensionMapping) {
            _servletPath = _servletMapping;
        } else {
            _servletPath = _requestURI;
        }

        if ((_servletPath != null) && _requestURI.startsWith(_servletPath)
                && (_requestURI.length() > _servletPath.length())) {

            _pathInfo = _requestURI.substring(_servletPath.length());

            try {
                _pathInfo = URLDecoder.decode(_pathInfo, StringPool.UTF8);
            } catch (UnsupportedEncodingException uee) {
                throw new RuntimeException(uee);
            }
        }
    }

    public void doDispatch(ServletRequest servletRequest, ServletResponse servletResponse)
            throws IOException, ServletException {

        ClassLoader contextClassLoader = ClassLoaderUtil.getContextClassLoader();

        try {
            ClassLoader pluginClassLoader = _bundleServletContext.getClassLoader();

            ClassLoaderUtil.setContextClassLoader(pluginClassLoader);

            for (ServletRequestListener servletRequestListener : _bundleServletContext
                    .getServletRequestListeners()) {

                ServletRequestEvent servletRequestEvent = new ServletRequestEvent(_bundleServletContext,
                        servletRequest);

                servletRequestListener.requestInitialized(servletRequestEvent);
            }

            _bundleFilterChain.doFilter(servletRequest, servletResponse);

            for (ServletRequestListener servletRequestListener : _bundleServletContext
                    .getServletRequestListeners()) {

                ServletRequestEvent servletRequestEvent = new ServletRequestEvent(_bundleServletContext,
                        servletRequest);

                servletRequestListener.requestDestroyed(servletRequestEvent);
            }
        } finally {
            ClassLoaderUtil.setContextClassLoader(contextClassLoader);
        }
    }

    @Override
    public void forward(ServletRequest servletRequest, ServletResponse servletResponse)
            throws IOException, ServletException {

        BundleServletRequest bundleServletRequest = new BundleServletRequest(this,
                (HttpServletRequest) servletRequest);

        doDispatch(bundleServletRequest, servletResponse);
    }

    @Override
    public void include(ServletRequest servletRequest, ServletResponse servletResponse)
            throws IOException, ServletException {

        BundleServletRequest bundleServletRequest = new BundleServletRequest(this,
                (HttpServletRequest) servletRequest);

        String contextPath = _bundleServletContext.getContextPath();

        if (_requestURI != null) {
            bundleServletRequest.setAttribute(JavaConstants.JAVAX_SERVLET_INCLUDE_REQUEST_URI,
                    _bundleServletContext.getContextPath().concat(_requestURI));
        }

        if (contextPath != null) {
            bundleServletRequest.setAttribute(JavaConstants.JAVAX_SERVLET_INCLUDE_CONTEXT_PATH, contextPath);
        }

        if (_servletPath != null) {
            bundleServletRequest.setAttribute(JavaConstants.JAVAX_SERVLET_INCLUDE_SERVLET_PATH, _servletPath);
        }

        if (_queryString != null) {
            bundleServletRequest.setAttribute(JavaConstants.JAVAX_SERVLET_INCLUDE_QUERY_STRING, _queryString);
        }

        if (_pathInfo != null) {
            bundleServletRequest.setAttribute(JavaConstants.JAVAX_SERVLET_INCLUDE_PATH_INFO, _pathInfo);
        }

        doDispatch(bundleServletRequest, servletResponse);
    }

    protected BundleServletContext getBundleServletContext() {
        return _bundleServletContext;
    }

    protected String getPathInfo() {
        return _pathInfo;
    }

    protected String getRequestURI() {
        return _requestURI;
    }

    protected String getServletPath() {
        return _servletPath;
    }

    private BundleFilterChain _bundleFilterChain;
    private BundleServletContext _bundleServletContext;
    private boolean _extensionMapping;
    private String _pathInfo;
    private String _queryString;
    private String _requestURI;
    private String _servletMapping;
    private String _servletPath;

}