skinny.engine.ContentEncodingSupport.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of skinny-engine_2.11 Show documentation

skinny-engine

The newest version!

package skinny.engine

import javax.servlet.http.{ HttpServletRequest, HttpServletResponse }

/**
 * SkinnyEngine handler for gzipped responses.
 */
trait ContentEncodingSupport extends Handler { self: SkinnyEngineBase =>

  abstract override def handle(req: HttpServletRequest, res: HttpServletResponse): Unit = {
    withRequestResponse(req, res) {
      super.handle(decodedRequest(req), encodedResponse(req, res))
    }
  }

  /** Encodes the response if necessary. */
  private def encodedResponse(req: HttpServletRequest, res: HttpServletResponse): HttpServletResponse = {
    ContentNegotiation.preferredEncoding(req).map { encoding =>
      val encoded = encoding(res)
      SkinnyEngineBase.onRenderedCompleted { _ => encoded.end() }(context)
      encoded
    }.getOrElse(res)
  }

  /** Decodes the request if necessary. */
  private def decodedRequest(req: HttpServletRequest): HttpServletRequest = {
    (for {
      name <- Option(req.getHeader("Content-Encoding"))
      enc <- ContentEncoding.forName(name)
    } yield enc(req)).getOrElse(req)
  }

}