Search code examples
scalaakkaakka-http

How does one log Akka HTTP client requests


I need to log akka http client requests as well as their responses. While there seems to be a hint of API for logging these requests, there is no clear documentation on how it should be done. My approach has been to create a logged request which transparently wraps Http().singleRequest(req) as follows:

def loggedRequest(req: HttpRequest)
                  (implicit system: ActorSystem, ctx: ExecutionContext, m: Materializer): Future[HttpResponse] = {

  Http().singleRequest(req).map { resp ⇒
    Unmarshal(resp.entity).to[String].foreach{s ⇒
      system.log.info(req.toString)
      system.log.info(resp.toString + "\n" + s)
    }
    resp
  }
}

Unfortunately, I have to grab the future either through the unmarshal or by simply requesting resp.entity.dataBytes in order to recover the body of the response. I get the logging but the promise gets completed and I can no longer unmarshal the entity to the actual data. A working solution would log the request and response and pass this test case without an IllegalStateException with "Promise already completed" being thrown:

describe("Logged rest requests") {

  it("deliver typed responses") {
    val foo = Rest.loggedRequest(Get(s"http://127.0.0.1:9000/some/path"))
    val resp = foo.futureValue(patience)
    resp.status shouldBe StatusCodes.OK
    val res = Unmarshal(resp.entity).to[MyClass].futureValue
  }
}

Ideas welcome.


Solution

  • One of the solution I've found is to use a:

    import akka.http.scaladsl.server.directives.DebuggingDirectives
    
    val clientRouteLogged = DebuggingDirectives.logRequestResult("Client ReST", Logging.InfoLevel)(clientRoute)
    Http().bindAndHandle(clientRouteLogged, interface, port)
    

    Which can easily log the request and result in raw(bytes) format. The problem is that those logs are completely unreadable. And here is place where it became complicated.

    Here is my example that encode the entity of the request/response and write it to the logger.

    You can pass a function to:

    DebuggingDirectives.logRequestResult
    
    def logRequestResult(magnet: LoggingMagnet[HttpRequest ⇒ RouteResult ⇒ Unit])
    

    That is function written using magnet pattern:

    LoggingMagnet[HttpRequest ⇒ RouteResult ⇒ Unit]
    

    Where:

    LoggingMagnet[T](f: LoggingAdapter ⇒ T)
    

    Thanks to that we have access to all parts that we need to log the request and result. We have LoggingAdapter, HttpRequest and RouteResult

    In my case I've create an inside function. I don't want to pass all the parameters again.

    def logRequestResult(level: LogLevel, route: Route)
                          (implicit m: Materializer, ex: ExecutionContext) = {
      def myLoggingFunction(logger: LoggingAdapter)(req: HttpRequest)(res: Any): Unit = {
        val entry = res match {
          case Complete(resp) =>
            entityAsString(resp.entity).map(data ⇒ LogEntry(s"${req.method} ${req.uri}: ${resp.status} \n entity: $data", level))
          case other =>
            Future.successful(LogEntry(s"$other", level))
        }
        entry.map(_.logTo(logger))
      }
      DebuggingDirectives.logRequestResult(LoggingMagnet(log => myLoggingFunction(log)))(route)
    }
    

    The most important part is the last line where I put myLoggingFunction in to logRequestResult.

    The function called myLoggingFunction, simple matched the result of server computation and create a LogEntry based on it.

    The last thing is a method that allows to decode the result entity from a stream.

    def entityAsString(entity: HttpEntity)
                       (implicit m: Materializer, ex: ExecutionContext): Future[String] = {
    entity.dataBytes
      .map(_.decodeString(entity.contentType().charset().value))
      .runWith(Sink.head)
    }
    

    The method can be easily add to any akka-http route.

    val myLoggedRoute = logRequestResult(Logging.InfoLevel, clientRoute)
    Http().bindAndHandle(myLoggedRoute, interface, port)