[OpenSource] 토비 Spring 3 - ConfigurableDispatcherServlet

토비 Spring 3 - ConfigurableDispatcherServlet.java Sample

package springbook.learningtest.spring.web;

 

import java.io.IOException;

 

import javax.servlet.ServletException;

import javax.servlet.ServletRequest;

import javax.servlet.ServletResponse;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

 

import org.springframework.beans.BeansException;

import org.springframework.beans.factory.support.DefaultListableBeanFactory;

import org.springframework.beans.factory.xml.XmlBeanDefinitionReader;

import org.springframework.context.ApplicationContext;

import org.springframework.context.annotation.AnnotatedBeanDefinitionReader;

import org.springframework.util.ClassUtils;

import org.springframework.web.context.WebApplicationContext;

import org.springframework.web.context.support.AbstractRefreshableWebApplicationContext;

import org.springframework.web.servlet.DispatcherServlet;

import org.springframework.web.servlet.ModelAndView;

 

@SuppressWarnings("serial")

public class ConfigurableDispatcherServlet extends DispatcherServlet {

    private Class<?>[] classes;

    private String[] locations;

     

    private ModelAndView modelAndView;

     

    public ConfigurableDispatcherServlet(String[] locations) {

        this.locations = locations;

    }

     

    public ConfigurableDispatcherServlet(Class<?> ...classes) {

        this.classes = classes;

    }

     

    public void setLocations(String ...locations) {

        this.locations = locations;

    }

     

    @SuppressWarnings("rawtypes")

    public void setRelativeLocations(Class clazz, String ...relativeLocations) {

        String[] locations = new String[relativeLocations.length];

        String currentPath = ClassUtils.classPackageAsResourcePath(clazz) + "/";

         

        for(int i = 0; i < relativeLocations.length; i++) {

            locations[i] = currentPath + relativeLocations[i];

        }

         

        this.setLocations(locations);

    }

     

    public void setClasses(Class<?> ...classes) {

        this.classes = classes;

    }

     

    public void service(ServletRequest req, ServletResponse res) throws ServletException, IOException {

        modelAndView = null;

        super.service(req, res);

    }

     

    protected WebApplicationContext createWebApplicationContext(ApplicationContext parent) {

        AbstractRefreshableWebApplicationContext wac = new AbstractRefreshableWebApplicationContext() {

            protected void loadBeanDefinitions(DefaultListableBeanFactory beanFactory) throws BeansException, IOException {

                if(locations != null) {

                    XmlBeanDefinitionReader xmlReader = new XmlBeanDefinitionReader(beanFactory);

                    xmlReader.loadBeanDefinitions(locations);

                }

                 

                if(classes != null) {

                    AnnotatedBeanDefinitionReader reader = new AnnotatedBeanDefinitionReader(beanFactory);

                    reader.register(classes);

                }

            }

        };

         

        wac.setServletContext(getServletContext());

        wac.setServletConfig(getServletConfig());

        wac.refresh();

         

        return wac;

    }

     

    protected void render(ModelAndView mv, HttpServletRequest request, HttpServletResponse response) throws Exception {

        this.modelAndView = mv;

        super.render(mv,  request, response);

    }

     

    public ModelAndView getModelAndView() {

        return modelAndView;

    }

}