psiprobe.controllers.ErrorHandlerController.java Source code

Introduction

Here is the source code for psiprobe.controllers.ErrorHandlerController.java

Source

/**
 * Licensed under the GPL License. You may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 *
 * THIS PACKAGE IS PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
 * WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE.
 */
package psiprobe.controllers;

import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.AbstractController;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * The ErrorHandlerController will show two different views depending on whether the failed request
 * was AJAX or not.
 *
 * @author Vlad Ilyushchenko.
 */
public class ErrorHandlerController extends AbstractController {

    /** The view name. */
    private String viewName;

    /** The ajax view name. */
    private String ajaxViewName;

    /** The ajax extension. */
    private String ajaxExtension = ".ajax";

    /**
     * Gets the view name.
     *
     * @return the view name
     */
    public String getViewName() {
        return viewName;
    }

    /**
     * Sets the view name.
     *
     * @param viewName the new view name
     */
    public void setViewName(String viewName) {
        this.viewName = viewName;
    }

    /**
     * Gets the ajax view name.
     *
     * @return the ajax view name
     */
    public String getAjaxViewName() {
        return ajaxViewName;
    }

    /**
     * Sets the ajax view name.
     *
     * @param ajaxViewName the new ajax view name
     */
    public void setAjaxViewName(String ajaxViewName) {
        this.ajaxViewName = ajaxViewName;
    }

    /**
     * Gets the ajax extension.
     *
     * @return the ajax extension
     */
    public String getAjaxExtension() {
        return ajaxExtension;
    }

    /**
     * Sets the ajax extension.
     *
     * @param ajaxExtension the new ajax extension
     */
    public void setAjaxExtension(String ajaxExtension) {
        this.ajaxExtension = ajaxExtension;
    }

    @Override
    protected ModelAndView handleRequestInternal(HttpServletRequest request, HttpServletResponse response)
            throws Exception {

        String originalUri = (String) request.getAttribute("javax.servlet.error.request_uri");
        if (originalUri != null && originalUri.endsWith(ajaxExtension)) {
            return new ModelAndView(ajaxViewName);
        }
        return new ModelAndView(viewName);
    }
}