com.twitter.finatra.http.internal.exceptions.FinatraDefaultExceptionMapper.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of finatra-http_2.11 Show documentation

finatra-http

The newest version!

package com.twitter.finatra.http.internal.exceptions

import com.google.common.net.MediaType
import com.twitter.finagle.http.{Request, Response}
import com.twitter.finagle.{CancelledRequestException, Failure}
import com.twitter.finatra.http.exceptions.{DefaultExceptionMapper, HttpException, HttpResponseException}
import com.twitter.finatra.http.internal.exceptions.FinatraDefaultExceptionMapper._
import com.twitter.finatra.http.response.{ErrorsResponse, ResponseBuilder}
import javax.inject.{Inject, Singleton}

object FinatraDefaultExceptionMapper {
  private val MaxDepth = 5

  private def unwrapFailure(failure: Failure, depth: Int): Throwable = {
    if (depth == 0)
      failure
    else
      failure.cause match {
        case Some(inner: Failure) => unwrapFailure(inner, depth - 1)
        case Some(cause) => cause
        case None => failure
      }
  }
}

@Singleton
class FinatraDefaultExceptionMapper @Inject()(
  response: ResponseBuilder)
  extends DefaultExceptionMapper {

  override def toResponse(request: Request, throwable: Throwable): Response = {
    throwable match {
      case e: HttpException => //TODO: Optionally pass handled exception
        val builder = response.status(e.statusCode)
        if (e.mediaType.is(MediaType.JSON_UTF_8))
          builder.json(ErrorsResponse(e.errors))
        else
          builder.plain(e.errors.mkString(", "))
      case e: HttpResponseException => //TODO: Optionally pass handled exception
        e.response
      case e: CancelledRequestException =>
        response
          .clientClosed
          .handled(request, e)
      case e: Failure =>
        unwrapFailure(e, MaxDepth) match {
          case cause: Failure =>
            response
              .internalServerError
              .unhandled(request, e)
              .jsonError
          case cause =>
            toResponse(request, cause)
        }
      case e =>
        // ExceptionMappingFilter protects us against fatal exceptions.
        response
          .internalServerError
          .unhandled(request, e)
          .jsonError
    }
  }
}