package com.thinkbiganalytics.rest.exception;
/*-
* #%L
* thinkbig-service-app
* %%
* Copyright (C) 2017 ThinkBig Analytics
* %%
* 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 com.thinkbiganalytics.rest.model.RestResponseStatus;
import org.springframework.context.annotation.Configuration;
import javax.annotation.Nonnull;
import javax.ws.rs.WebApplicationException;
import javax.ws.rs.core.Response;
import javax.ws.rs.ext.ExceptionMapper;
import javax.ws.rs.ext.Provider;
/**
* Maps {@link WebApplicationException} objects to a standard response format.
*/
@Provider
@Configuration
public class WebApplicationExceptionMapper extends BaseExceptionMapper implements ExceptionMapper<WebApplicationException> {
@Nonnull
@Override
public Response toResponse(@Nonnull final WebApplicationException exception) {
if (exception.getResponse().getEntity() == null) {
final RestResponseStatus.ResponseStatusBuilder builder = new RestResponseStatus.ResponseStatusBuilder()
.message(exception.getMessage())
.url(req.getRequestURI());
if (debugMode && exception.getCause() != null) {
builder.setDeveloperMessage(exception.getCause());
}
return Response.fromResponse(exception.getResponse()).entity(builder.buildError()).build();
} else {
return exception.getResponse();
}
}
}