Add service exception json formatters and pass through exceptions in HttpRestServiceTransport (#168)v1.9.4

1 files changed, 9 insertions, 3 deletions

diff --git a/src/main/scala/xyz/driver/core/rest/DriverRoute.scala b/src/main/scala/xyz/driver/core/rest/DriverRoute.scala
index c23c44d..fe3790f 100644
--- a/src/main/scala/xyz/driver/core/rest/DriverRoute.scala
+++ b/src/main/scala/xyz/driver/core/rest/DriverRoute.scala
@@ -90,11 +90,17 @@ trait DriverRoute {
     }
 
     { (ctx: RequestContext) =>
-      errorResponse(statusCode, serviceException.message, serviceException)(ctx)
+      import xyz.driver.core.json.serviceExceptionFormat
+      val entity =
+        HttpEntity(ContentTypes.`application/json`, serviceExceptionFormat.write(serviceException).toString())
+      errorResponse(statusCode, entity, serviceException)(ctx)
     }
   }
 
-  protected def errorResponse[T <: Exception](statusCode: StatusCode, message: String, exception: T): Route = {
-    complete(HttpResponse(statusCode, entity = message))
+  protected def errorResponse[T <: Exception](statusCode: StatusCode, message: String, exception: T): Route =
+    errorResponse(statusCode, HttpEntity(message), exception)
+
+  protected def errorResponse[T <: Exception](statusCode: StatusCode, entity: ResponseEntity, exception: T): Route = {
+    complete(HttpResponse(statusCode, entity = entity))
   }
 }