com.twitter.finatra.thrift.ThriftClientExceptionMapper.scala Maven / Gradle / Ivy

Go to download
package com.twitter.finatra.thrift

import com.twitter.finagle.http.Request
import com.twitter.finagle.http.Response
import com.twitter.finagle.http.Status
import com.twitter.finatra.http.exceptions.ExceptionMapper
import com.twitter.finatra.http.response.ResponseBuilder
import com.twitter.inject.utils.ExceptionUtils._
import com.twitter.inject.thrift.ThriftClientException
import com.twitter.inject.thrift.ThriftClientExceptionSource
import javax.inject.Inject
import javax.inject.Singleton

@Singleton
class ThriftClientExceptionMapper @Inject() (
  response: ResponseBuilder,
  source: ThriftClientExceptionSource)
    extends ExceptionMapper[ThriftClientException] {

  override def toResponse(
    request: Request,
    exception: ThriftClientException
  ): Response = {
    response
      .status(Status.ServiceUnavailable)
      .jsonError
      .failure(
        request,
        source(exception),
        details = Seq(
          exception.method.serviceName,
          exception.method.name,
          toExceptionDetails(exception.cause)
        ),
        message = toExceptionMessage(exception.cause)
      )
  }
}