org.saiku.plugin.EnunciateJerseyPluginServlet.java Source code

Introduction

Here is the source code for org.saiku.plugin.EnunciateJerseyPluginServlet.java

Source

/*  
 *   Copyright 2012 OSBI Ltd
 *
 *   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.saiku.plugin;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.codehaus.enunciate.modules.jersey.EnunciateJerseyServletContainer;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.ConfigurableApplicationContext;

import com.sun.jersey.api.core.ResourceConfig;
import com.sun.jersey.core.spi.component.ioc.IoCComponentProviderFactory;
import com.sun.jersey.spi.container.WebApplication;
import com.sun.jersey.spi.spring.container.SpringComponentProviderFactory;

/**
 * This should only be used by a plugin in the plugin.spring.xml file to initialize a Jersey.  The
 * presence of this servlet in the spring file will make it possible to write JAX-RS POJOs in your
 * plugin.
 * @author Aaron Phillips
 */
public class EnunciateJerseyPluginServlet extends EnunciateJerseyServletContainer
        implements ApplicationContextAware {

    private static final long serialVersionUID = 457538570048660945L;

    private ApplicationContext applicationContext;

    private static final Log logger = LogFactory.getLog(EnunciateJerseyPluginServlet.class);

    private Map<String, String> initParams = new HashMap<String, String>();

    public Map<String, String> getInitParams() {
        return initParams;
    }

    public void setInitParams(Map<String, String> initParams) {
        this.initParams = initParams;
    }

    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.applicationContext = applicationContext;
    }

    @Override
    public void service(final HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        logger.debug("servicing request for resource " + request.getPathInfo()); //$NON-NLS-1$
        //Strip out the content generator id from the URL here so Jersey can match the @Path of the resource
        HttpServletRequestWrapper wrapper = new HttpServletRequestWrapper(request) {

            @Override
            public String getRequestURI() {
                String uri = super.getRequestURI().replace("content/", "");
                return uri;
            }

            @Override
            public StringBuffer getRequestURL() {
                String url = super.getRequestURL().toString();
                return new StringBuffer(url.replace("content/", ""));
            }

            @Override
            public Map getParameterMap() {
                return request.getParameterMap();
            }

            @Override
            public String getQueryString() {
                return super.getQueryString();
            }

        };
        super.service(wrapper, response);
    }

    @Override
    public void service(ServletRequest req, ServletResponse res) throws ServletException, IOException {
        super.service(req, res);
    }

    @Override
    protected void initiate(ResourceConfig rc, WebApplication wa) {
        if (logger.isDebugEnabled()) {
            rc.getFeatures().put(ResourceConfig.FEATURE_TRACE, true);
            rc.getFeatures().put(ResourceConfig.FEATURE_TRACE_PER_REQUEST, true);
        }
        super.initiate(rc, wa);
    }

    @Override
    public void init(ServletConfig config) throws ServletException {
        ((MutableServletConfig) config).setInitParameters(initParams);
        super.init(config);
    }

    @SuppressWarnings("unchecked")
    @Override
    protected IoCComponentProviderFactory loadResourceProviderFacotry(ResourceConfig rc) {
        try {
            return new SpringComponentProviderFactory(rc, (ConfigurableApplicationContext) applicationContext);
        } catch (Throwable e) {
            throw new IllegalStateException("Unable to load the spring component provider factory.", e);
        }
    }

}