com.tcloud.bee.key.server.jetty.config.WebAppInitializer.java Source code

Introduction

Here is the source code for com.tcloud.bee.key.server.jetty.config.WebAppInitializer.java

Source

/**
 * Copyright 2013 Trend Micro Incorporated
 *
 * 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 com.tcloud.bee.key.server.jetty.config;

import java.util.EnumSet;

import javax.servlet.DispatcherType;
import javax.servlet.FilterRegistration;
import javax.servlet.ServletContext;
import javax.servlet.ServletContextEvent;
import javax.servlet.ServletContextListener;
import javax.servlet.ServletException;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.filter.CharacterEncodingFilter;
import org.springframework.web.filter.DelegatingFilterProxy;
import org.springframework.web.servlet.support.AbstractAnnotationConfigDispatcherServletInitializer;

/**
 * ?web.xml 
 * 
 * @author howie_yu
 * 
 */
public class WebAppInitializer extends AbstractAnnotationConfigDispatcherServletInitializer
        implements ServletContextListener {

    final Logger logger = LoggerFactory.getLogger(getClass());

    /**
     * See {@link org.springframework.web.servlet.support.AbstractAnnotationConfigDispatcherServletInitializer}.
     */
    @Override
    protected Class<?>[] getRootConfigClasses() {
        return null;
    }

    /**
     * Set the application context for the Spring MVC web tier.
     * 
     * @See {@link org.springframework.web.servlet.support.AbstractAnnotationConfigDispatcherServletInitializer}
     */
    @Override
    protected Class<?>[] getServletConfigClasses() {
        return new Class<?>[] { MvcConfiguration.class };
    }

    /**
     * Map the Spring MVC servlet as the root.
     * 
     * @See {@link org.springframework.web.servlet.support.AbstractAnnotationConfigDispatcherServletInitializer}
     */
    @Override
    protected String[] getServletMappings() {
        return new String[] { "/" };
    }

    @Override
    public void contextInitialized(ServletContextEvent servletContextEvent) {
        try {
            onStartup(servletContextEvent.getServletContext());
        } catch (ServletException e) {
            logger.error("Failed to initialize web application", e);
            System.exit(0);
        }
    }

    @Override
    public void contextDestroyed(ServletContextEvent servletContextEvent) {
    }

    /**
     * Overrided to squelch a meaningless log message when embedded.
     */
    @Override
    protected void registerContextLoaderListener(ServletContext servletContext) {
        servletContext.addFilter("HiddenHttpMethodFilter",
                org.springframework.web.filter.HiddenHttpMethodFilter.class);
    }

    private void configureEncodingFilter(ServletContext servletContext) {

        CharacterEncodingFilter encodingFilter = new CharacterEncodingFilter();
        encodingFilter.setEncoding("UTF-8");
        encodingFilter.setForceEncoding(false);
        FilterRegistration.Dynamic encodingFilterDinamic = servletContext.addFilter("charEncodingFilter",
                encodingFilter);
        encodingFilterDinamic.addMappingForUrlPatterns(EnumSet.allOf(DispatcherType.class), true, "/*");

    }

    /**
     * Register Spring Security
     * 
     * @param servletContext
     */
    private void configureSpringSecurityFilter(ServletContext servletContext) {

        DelegatingFilterProxy delegatingFilterProxy = new DelegatingFilterProxy("springSecurityFilterChain");

        FilterRegistration.Dynamic securityFilterDynamic = servletContext.addFilter("securityFilter",
                delegatingFilterProxy);
        securityFilterDynamic.addMappingForUrlPatterns(EnumSet.allOf(DispatcherType.class), true, "/*");
    }

    /**
     * Register JerseyServlet Config
     * 
     * @param servletContext
     */
    private void configurationJersey(ServletContext servletContext) {

        FilterRegistration.Dynamic filterDinamic = servletContext.addFilter("jersey",
                new org.glassfish.jersey.servlet.ServletContainer());
        filterDinamic.setInitParameter("javax.ws.rs.Application",
                "org.apache.hadoop.io.crypto.bee.key.server.jetty.config.JerseyApplicationConfig");
        filterDinamic.addMappingForUrlPatterns(EnumSet.allOf(DispatcherType.class), true, "/api/*");
    }

    @Override
    public void onStartup(ServletContext servletContext) throws ServletException {

        logger.info("On Startup WebAppInitializer.....");

        /* Let super do its thing... , include add RootConfig and ServletConfig */
        super.onStartup(servletContext);

        configureEncodingFilter(servletContext);

        configureSpringSecurityFilter(servletContext);

        configurationJersey(servletContext);
    }

}