io.gatling.http.cache.HttpContentCacheSupport.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gatling-http Show documentation
gatling-http
There is a newer version: 3.13.1
/**
 * Copyright 2011-2016 GatlingCorp (http://gatling.io)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.gatling.http.cache

import io.gatling.core.config.GatlingConfiguration
import io.gatling.core.session.{ Session, SessionPrivateAttributes }
import io.gatling.core.util.cache.SessionCacheHandler
import io.gatling.http.HeaderNames
import io.gatling.http.protocol.HttpProtocol
import io.gatling.http.response.Response

import org.asynchttpclient.Request
import org.asynchttpclient.uri.Uri

object ContentCacheKey {
  def apply(request: Request): ContentCacheKey =
    new ContentCacheKey(request.getUri, request.getMethod, new Cookies(request.getCookies))
}

case class ContentCacheKey(uri: Uri, method: String, cookies: Cookies)

case class ContentCacheEntry(expires: Option[Long], etag: Option[String], lastModified: Option[String])

object HttpContentCacheSupport {
  val HttpContentCacheAttributeName = SessionPrivateAttributes.PrivateAttributePrefix + "http.cache.contentCache"
}

trait HttpContentCacheSupport extends ExpiresSupport {

  import HttpContentCacheSupport._

  def configuration: GatlingConfiguration

  val httpContentCacheHandler = new SessionCacheHandler[ContentCacheKey, ContentCacheEntry](HttpContentCacheAttributeName, configuration.http.perUserCacheMaxCapacity)

  def cacheContent(httpProtocol: HttpProtocol, request: Request, response: Response): Session => Session =
    if (httpProtocol.requestPart.cache) {

      val expires = getResponseExpires(response)
      val etag = response.header(HeaderNames.ETag)
      val lastModified = response.header(HeaderNames.LastModified)

      if (expires.isDefined || etag.isDefined || lastModified.isDefined) {
        val key = ContentCacheKey(request.getUri, request.getMethod, new Cookies(request.getCookies))
        val value = ContentCacheEntry(expires, etag, lastModified)
        httpContentCacheHandler.addEntry(_, key, value)
      } else
        Session.Identity
    } else
      Session.Identity

  def contentCacheEntry(session: Session, request: Request): Option[ContentCacheEntry] =
    httpContentCacheHandler.getEntry(session, ContentCacheKey(request))

  def clearContentCache(session: Session, request: Request): Session =
    httpContentCacheHandler.removeEntry(session, ContentCacheKey(request))
}