From 530f781a3d1a30dd381e16e8ee67becad2f17f07 Mon Sep 17 00:00:00 2001 From: vlad Date: Mon, 26 Mar 2018 01:21:45 -0700 Subject: Correct `trackingId` placement to the MDC --- src/main/scala/xyz/driver/core/rest/DriverRoute.scala | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) (limited to 'src/main/scala/xyz') diff --git a/src/main/scala/xyz/driver/core/rest/DriverRoute.scala b/src/main/scala/xyz/driver/core/rest/DriverRoute.scala index 8878ef4..c23c44d 100644 --- a/src/main/scala/xyz/driver/core/rest/DriverRoute.scala +++ b/src/main/scala/xyz/driver/core/rest/DriverRoute.scala @@ -26,10 +26,9 @@ trait DriverRoute { protected def defaultResponseHeaders: Directive0 = { extractRequest flatMap { request => - // Needs to happen before any request processing, so all the log messages // associated with processing of this request are having this `trackingId` - val trackingId = rest.extractTrackingId(request) + val trackingId = rest.extractTrackingId(request) val tracingHeader = RawHeader(ContextHeaders.TrackingIdHeader, trackingId) MDC.put("trackingId", trackingId) @@ -54,10 +53,8 @@ trait DriverRoute { case cm: ConcurrentModificationException => ctx => log.warn(s"Concurrent modification of the resource ${ctx.request.method} ${ctx.request.uri}", cm) - errorResponse( - StatusCodes.Conflict, - "Resource was changed concurrently, try requesting a newer version", - cm)(ctx) + errorResponse(StatusCodes.Conflict, "Resource was changed concurrently, try requesting a newer version", cm)( + ctx) case se: SQLException => ctx => @@ -97,10 +94,7 @@ trait DriverRoute { } } - protected def errorResponse[T <: Exception]( - statusCode: StatusCode, - message: String, - exception: T): Route = { + protected def errorResponse[T <: Exception](statusCode: StatusCode, message: String, exception: T): Route = { complete(HttpResponse(statusCode, entity = message)) } } -- cgit v1.2.3