AbstractExceptionMapper.java example

Explorer
usergrid-master
/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.apache.usergrid.rest.exceptions;


import org.apache.usergrid.rest.ApiResponse;
import org.apache.usergrid.services.exceptions.UnsupportedServiceOperationException;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.http.HttpServletRequest;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.Response.Status;
import javax.ws.rs.ext.ExceptionMapper;

import static javax.ws.rs.core.MediaType.APPLICATION_JSON_TYPE;
import static javax.ws.rs.core.Response.Status.INTERNAL_SERVER_ERROR;
import static javax.ws.rs.core.Response.Status.OK;
import static org.apache.commons.lang.StringUtils.isNotBlank;
import static org.apache.usergrid.rest.utils.JSONPUtils.isJavascript;
import static org.apache.usergrid.rest.utils.JSONPUtils.wrapJSONPResponse;
import static org.apache.usergrid.utils.JsonUtils.mapToJsonString;


public abstract class AbstractExceptionMapper<E extends java.lang.Throwable> implements ExceptionMapper<E> {

    public static final Logger logger = LoggerFactory.getLogger( AbstractExceptionMapper.class );

    @Context
    HttpHeaders hh;

    @Context
    protected HttpServletRequest httpServletRequest;


    public boolean isJSONP() {
        return isJavascript(hh.getAcceptableMediaTypes());
    }


    @Override
    public Response toResponse( E e ) {
        // if we don't know what type of error it is then it's a 500
        return toResponse( INTERNAL_SERVER_ERROR, e );
    }


    public Response toResponse( Status status, E e ) {
        return toResponse( status.getStatusCode(), e );
    }


    public Response toResponse( int status, E e ) {

        if ( status >= 500 ) {
            // only log real errors as errors
            logger.error( "{} 5XX Uncaught Exception ({})", e.getClass().getCanonicalName(), status, e );
        } else {
            if (logger.isDebugEnabled()) {
                logger.debug( "{} Following Exception Thrown ({}), {}", e.getClass().getCanonicalName(), status, e );
            }
        }

        ApiResponse response = new ApiResponse();


        AuthErrorInfo authError = AuthErrorInfo.getForException( e );

        if ( authError != null ) {
            response.setError( authError.getType(), authError.getMessage(), e );
        }
        else {
            response.setError( e );
        }

        String jsonResponse = mapToJsonString( response );

        return toResponse( status, jsonResponse );
    }


    public Response toResponse( Status status, String jsonResponse ) {
        return toResponse( status.getStatusCode(), jsonResponse, false );
    }

    public Response toResponse( Status status, String jsonResponse, boolean skipLogging ) {
        return toResponse( status.getStatusCode(), jsonResponse, skipLogging );
    }

    public Response toResponse( int statusCode, String jsonResponse ) {
        return toResponse( statusCode, jsonResponse, false );
    }

    protected Response toResponse( int status, String jsonResponse, boolean skipLogging ) {
        if ( status >= 500 && !skipLogging) {
            // only log real errors as errors
            logger.error( "Server Error ({}):\n{}", status, jsonResponse );
        } else if ( logger.isDebugEnabled() ) {
            logger.debug( "Client Error ({}):\n{}", status, jsonResponse );
        }

        String callback = httpServletRequest.getParameter( "callback" );
        if ( isJSONP() && isNotBlank( callback ) ) {
            jsonResponse = wrapJSONPResponse( callback, jsonResponse );
            return Response.status( OK ).type( "application/javascript" ).entity( jsonResponse ).build();
        }
        else {
            return Response.status( status ).type( APPLICATION_JSON_TYPE ).entity( jsonResponse ).build();
        }
    }
}