ca.uhn.fhir.rest.server.interceptor.ExceptionHandlingInterceptor.java Source code

Java tutorial

Introduction

Here is the source code for ca.uhn.fhir.rest.server.interceptor.ExceptionHandlingInterceptor.java

Source

package ca.uhn.fhir.rest.server.interceptor;

/*
 * #%L
 * HAPI FHIR - Core Library
 * %%
 * Copyright (C) 2014 - 2016 University Health Network
 * %%
 * 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.
 * #L%
 */
import static org.apache.commons.lang3.StringUtils.isNotBlank;

import java.io.Closeable;
import java.io.IOException;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

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

import org.apache.commons.lang3.exception.ExceptionUtils;
import org.hl7.fhir.instance.model.api.IBaseOperationOutcome;

import ca.uhn.fhir.context.FhirContext;
import ca.uhn.fhir.rest.api.SummaryEnum;
import ca.uhn.fhir.rest.method.RequestDetails;
import ca.uhn.fhir.rest.server.IRestfulResponse;
import ca.uhn.fhir.rest.server.exceptions.BaseServerResponseException;
import ca.uhn.fhir.rest.server.exceptions.InternalErrorException;
import ca.uhn.fhir.rest.server.exceptions.UnclassifiedServerFailureException;
import ca.uhn.fhir.util.OperationOutcomeUtil;

public class ExceptionHandlingInterceptor extends InterceptorAdapter {

    public static final String PROCESSING = "processing";
    private static final org.slf4j.Logger ourLog = org.slf4j.LoggerFactory
            .getLogger(ExceptionHandlingInterceptor.class);
    private Class<?>[] myReturnStackTracesForExceptionTypes;

    @Override
    public boolean handleException(RequestDetails theRequestDetails, BaseServerResponseException theException,
            HttpServletRequest theRequest, HttpServletResponse theResponse) throws ServletException, IOException {
        Closeable writer = (Closeable) handleException(theRequestDetails, theException);
        writer.close();
        return false;
    }

    public Object handleException(RequestDetails theRequestDetails, BaseServerResponseException theException)
            throws ServletException, IOException {
        IRestfulResponse response = theRequestDetails.getResponse();

        FhirContext ctx = theRequestDetails.getServer().getFhirContext();

        IBaseOperationOutcome oo = theException.getOperationOutcome();
        if (oo == null) {
            oo = createOperationOutcome(theException, ctx);
        }

        int statusCode = theException.getStatusCode();

        // Add headers associated with the specific error code
        if (theException.hasResponseHeaders()) {
            Map<String, List<String>> additional = theException.getResponseHeaders();
            for (Entry<String, List<String>> next : additional.entrySet()) {
                if (isNotBlank(next.getKey()) && next.getValue() != null) {
                    String nextKey = next.getKey();
                    for (String nextValue : next.getValue()) {
                        response.addHeader(nextKey, nextValue);
                    }
                }
            }
        }

        String statusMessage = null;
        if (theException instanceof UnclassifiedServerFailureException) {
            String sm = theException.getMessage();
            if (isNotBlank(sm) && sm.indexOf('\n') == -1) {
                statusMessage = sm;
            }
        }

        return response.streamResponseAsResource(oo, true, Collections.singleton(SummaryEnum.FALSE), statusCode,
                statusMessage, false, false);

    }

    @Override
    public BaseServerResponseException preProcessOutgoingException(RequestDetails theRequestDetails,
            Throwable theException, HttpServletRequest theServletRequest) throws ServletException {
        BaseServerResponseException retVal;
        if (!(theException instanceof BaseServerResponseException)) {
            retVal = new InternalErrorException(theException);
        } else {
            retVal = (BaseServerResponseException) theException;
        }

        if (retVal.getOperationOutcome() == null) {
            retVal.setOperationOutcome(
                    createOperationOutcome(theException, theRequestDetails.getServer().getFhirContext()));
        }

        return retVal;
    }

    private IBaseOperationOutcome createOperationOutcome(Throwable theException, FhirContext ctx)
            throws ServletException {
        IBaseOperationOutcome oo = null;
        if (theException instanceof BaseServerResponseException) {
            oo = ((BaseServerResponseException) theException).getOperationOutcome();
        }

        /*
         * Generate an OperationOutcome to return, unless the exception throw by the resource provider had one
         */
        if (oo == null) {
            try {
                oo = OperationOutcomeUtil.newInstance(ctx);

                if (theException instanceof InternalErrorException) {
                    ourLog.error("Failure during REST processing", theException);
                    populateDetails(ctx, theException, oo);
                } else if (theException instanceof BaseServerResponseException) {
                    int statusCode = ((BaseServerResponseException) theException).getStatusCode();

                    // No stack traces for non-server internal errors
                    if (statusCode < 500) {
                        ourLog.warn("Failure during REST processing: {}", theException.toString());
                    } else {
                        ourLog.warn("Failure during REST processing: {}", theException);
                    }

                    BaseServerResponseException baseServerResponseException = (BaseServerResponseException) theException;
                    populateDetails(ctx, theException, oo);
                    if (baseServerResponseException.getAdditionalMessages() != null) {
                        for (String next : baseServerResponseException.getAdditionalMessages()) {
                            OperationOutcomeUtil.addIssue(ctx, oo, "error", next, null, PROCESSING);
                        }
                    }
                } else {
                    ourLog.error("Failure during REST processing: " + theException.toString(), theException);
                    populateDetails(ctx, theException, oo);
                }
            } catch (Exception e1) {
                ourLog.error("Failed to instantiate OperationOutcome resource instance", e1);
                throw new ServletException("Failed to instantiate OperationOutcome resource instance", e1);
            }
        } else {
            ourLog.error("Unknown error during processing", theException);
        }
        return oo;
    }

    private void populateDetails(FhirContext theCtx, Throwable theException, IBaseOperationOutcome theOo) {
        if (myReturnStackTracesForExceptionTypes != null) {
            for (Class<?> next : myReturnStackTracesForExceptionTypes) {
                if (next.isAssignableFrom(theException.getClass())) {
                    String detailsValue = theException.getMessage() + "\n\n"
                            + ExceptionUtils.getStackTrace(theException);
                    OperationOutcomeUtil.addIssue(theCtx, theOo, "error", detailsValue, null, PROCESSING);
                    return;
                }
            }
        }

        OperationOutcomeUtil.addIssue(theCtx, theOo, "error", theException.getMessage(), null, PROCESSING);
    }

    /**
     * If any server methods throw an exception which extends any of the given exception types, the exception stack trace will be returned to the user. This can be useful for helping to diagnose
     * issues, but may not be desirable for production situations.
     * 
     * @param theExceptionTypes
     *           The exception types for which to return the stack trace to the user.
     * @return Returns an instance of this interceptor, to allow for easy method chaining.
     */
    public ExceptionHandlingInterceptor setReturnStackTracesForExceptionTypes(Class<?>... theExceptionTypes) {
        myReturnStackTracesForExceptionTypes = theExceptionTypes;
        return this;
    }

}