webapp/wmservices/src/main/java/org/apache/oodt/cas/wmservices/exceptionmappers/NotFoundExceptionMapper.java

/*
 * 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.oodt.cas.wmservices.exceptionmappers;

import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import javax.ws.rs.ext.ExceptionMapper;
import javax.ws.rs.ext.Provider;
import org.apache.oodt.cas.wmservices.enums.ErrorType;
import org.apache.oodt.cas.wmservices.errors.ErrorMessage;
import org.apache.oodt.cas.wmservices.exceptions.NotFoundException;

/**
 * This is an exception mapper which maps "NotFoundException" to "ErrorMessage JSON payload"
 *
 * @author ngimhana (Nadeeshan Gimhana)
 */
@Provider
public class NotFoundExceptionMapper implements ExceptionMapper<NotFoundException> {

  /** Maps NotFoundException to HTTP Response * */
  @Override
  public Response toResponse(NotFoundException exception) {

    // Initialising ErrorMessage Entity for Mapping to Response
    ErrorMessage errorMessageEntity =
        new ErrorMessage(
            Response.Status.NOT_FOUND.getStatusCode(),
            ErrorType.NOT_FOUND_EXCEPTION.getErrorType(),
            exception.getMessage());

    // Maps Error Status 404 to Response
    return Response.status(Response.Status.NOT_FOUND)
        .entity(errorMessageEntity)
        .type(MediaType.APPLICATION_JSON)
        .build();
  }
}