CatchExceptionFilter.java

/*
 * JBoss, Home of Professional Open Source
 * Copyright 2010, Red Hat Middleware LLC, and individual contributors
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * 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.jboss.solder.servlet.exception;

import java.io.IOException;

import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;

import org.apache.deltaspike.core.api.exception.control.event.ExceptionToCatchEvent;
import org.jboss.solder.logging.Logger;
//import org.jboss.solder.exception.control.ExceptionToCatch;
import org.jboss.solder.beanManager.BeanManagerAware;
import org.jboss.solder.beanManager.BeanManagerUnavailableException;
import org.jboss.solder.core.Requires;
import org.jboss.solder.servlet.literal.WebRequestLiteral;
import org.jboss.solder.servlet.support.ServletLogger;

/**
 * A bridge that forwards unhandled exceptions to the Solder exception handling facility (Solder Catch).
 *
 * @author <a href="http://community.jboss.org/people/dan.j.allen">Dan Allen</a>
 */
//@Requires({"org.jboss.solder.exception.control.extension.CatchExtension", "javax.servlet.Servlet"})
@Requires({"org.apache.deltaspike.core.impl.exception.control.extension.ExceptionControlExtension", "javax.servlet.Servlet"})
public class CatchExceptionFilter extends BeanManagerAware implements Filter {
    private transient ServletLogger log = Logger.getMessageLogger(ServletLogger.class, ServletLogger.CATEGORY);

    private boolean enabled = false;

    public void init(FilterConfig config) throws ServletException {
        try {
            if (!getBeanManager().getBeans(CatchExceptionFilter.class).isEmpty()) {
                enabled = true;
                log.catchIntegrationEnabled();
            }
        } catch (BeanManagerUnavailableException e) {
            log.catchIntegrationDisabledNoBeanManager();
        }
    }

    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException,
            ServletException {
        if (!enabled) {
            chain.doFilter(request, response);
        } else {
            try {
                chain.doFilter(request, response);
            } catch (Exception e) {
                ExceptionToCatchEvent catchEvent;
                if (request instanceof HttpServletRequest) {
                    catchEvent = new ExceptionToCatchEvent(e, WebRequestLiteral.INSTANCE);
                    // new PathLiteral(HttpServletRequest.class.cast(request).getServletPath()));
                } else {
                    catchEvent = new ExceptionToCatchEvent(e, WebRequestLiteral.INSTANCE);
                }
                getBeanManager().fireEvent(catchEvent);
                // QUESTION should catch handle rethrowing?
                if (!catchEvent.isHandled()) {
                    if (e instanceof ServletException) {
                        throw (ServletException) e;
                    } else if (e instanceof IOException) {
                        throw (IOException) e;
                    } else {
                        throw new ServletException(e);
                    }
                }
            }
        }
    }

    public void destroy() {
    }
}