Http stacktracing using headers to see which hosts this request was passing

1 files changed, 25 insertions, 10 deletions

diff --git a/src/main/scala/xyz/driver/core/app.scala b/src/main/scala/xyz/driver/core/app.scala
index e35c300..5a6dbbc 100644
--- a/src/main/scala/xyz/driver/core/app.scala
+++ b/src/main/scala/xyz/driver/core/app.scala
@@ -31,7 +31,8 @@ import scalaz.syntax.equal._
 
 object app {
 
-  class DriverApp(version: String,
+  class DriverApp(appName: String,
+                  version: String,
                   gitHash: String,
                   modules: Seq[Module],
                   time: TimeProvider = new SystemTimeProvider(),
@@ -78,14 +79,11 @@ object app {
                 { ctx =>
                   val trackingId = rest.extractTrackingId(ctx.request)
                   MDC.put("trackingId", trackingId)
-                  MDC.put("origin", origin)
-                  MDC.put("xForwardedFor",
-                          extractHeader(ctx.request)("x-forwarded-for")
-                            .orElse(extractHeader(ctx.request)("x_forwarded_for"))
-                            .getOrElse("unknown"))
-                  MDC.put("remoteAddress", extractHeader(ctx.request)("remote-address").getOrElse("unknown"))
-                  MDC.put("userAgent", extractHeader(ctx.request)("user-agent").getOrElse("unknown"))
-                  MDC.put("ip", ip.toOption.map(_.getHostAddress).getOrElse("unknown"))
+
+                  val updatedStacktrace = (rest.extractStacktrace(ctx.request) ++ Array(appName)).mkString("->")
+                  MDC.put("stack", updatedStacktrace)
+
+                  storeRequestContextToMdc(ctx.request, origin, ip)
 
                   def requestLogging: Future[Unit] = Future {
                     log.info(
@@ -93,7 +91,10 @@ object app {
                   }
 
                   val contextWithTrackingId =
-                    ctx.withRequest(ctx.request.addHeader(RawHeader(ContextHeaders.TrackingIdHeader, trackingId)))
+                    ctx.withRequest(
+                      ctx.request
+                        .addHeader(RawHeader(ContextHeaders.TrackingIdHeader, trackingId))
+                        .addHeader(RawHeader(ContextHeaders.StacktraceHeader, updatedStacktrace)))
 
                   handleExceptions(ExceptionHandler(exceptionHandler))({ c =>
                     requestLogging.flatMap { _ =>
@@ -111,6 +112,20 @@ object app {
       }
     }
 
+    private def storeRequestContextToMdc(request: HttpRequest, origin: String, ip: RemoteAddress) = {
+
+      MDC.put("origin", origin)
+      MDC.put("ip", ip.toOption.map(_.getHostAddress).getOrElse("unknown"))
+      MDC.put("remoteHost", ip.toOption.map(_.getHostName).getOrElse("unknown"))
+
+      MDC.put("xForwardedFor",
+              extractHeader(request)("x-forwarded-for")
+                .orElse(extractHeader(request)("x_forwarded_for"))
+                .getOrElse("unknown"))
+      MDC.put("remoteAddress", extractHeader(request)("remote-address").getOrElse("unknown"))
+      MDC.put("userAgent", extractHeader(request)("user-agent").getOrElse("unknown"))
+    }
+
     /**
       * Override me for custom exception handling
       *