• Home
• com.lightbend.akka.grpc
• akka-grpc-runtime_2.11
• 0.5.0
• source code
• GrpcExceptionHandler.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

akka.grpc.scaladsl.GrpcExceptionHandler.scala Maven / Gradle / Ivy

/*
 * Copyright (C) 2018-2019 Lightbend Inc. 
 */

package akka.grpc.scaladsl

import akka.actor.ActorSystem
import akka.grpc.GrpcServiceException
import akka.grpc.internal.GrpcResponseHelpers
import akka.http.scaladsl.model.HttpResponse
import io.grpc.Status

import scala.concurrent.{ ExecutionException, Future }

object GrpcExceptionHandler {
  def default(mapper: PartialFunction[Throwable, Status])(implicit system: ActorSystem): PartialFunction[Throwable, Future[HttpResponse]] = {
    mapper
      .orElse(defaultMapper(system))
      .andThen(s => Future.successful(GrpcResponseHelpers.status(s)))
  }

  def defaultMapper(system: ActorSystem): PartialFunction[Throwable, Status] = {
    case e: ExecutionException ⇒
      if (e.getCause == null) Status.INTERNAL
      else defaultMapper(system)(e.getCause)
    case grpcException: GrpcServiceException ⇒ grpcException.status
    case _: NotImplementedError ⇒ Status.UNIMPLEMENTED
    case _: UnsupportedOperationException ⇒ Status.UNIMPLEMENTED
    case other ⇒
      system.log.error(other, other.getMessage)
      Status.INTERNAL
  }

  def default(implicit system: ActorSystem): PartialFunction[Throwable, Future[HttpResponse]] = default(defaultMapper(system))
}