io.gatling.http.request.builder.package.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gatling-http
gatling-http
There is a newer version: 3.0.3
/**
 * Copyright 2011-2014 eBusiness Information, Groupe Excilys (www.ebusinessinformation.fr)
 *
 * 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.request

import scala.annotation.tailrec
import scala.collection.breakOut

import com.ning.http.client.Param

import io.gatling.core.session.Session
import io.gatling.core.validation._

import java.util.{ List => JList, ArrayList => JArrayList }

package object builder {

  implicit class HttpParams(val params: List[HttpParam]) extends AnyVal {

    def resolveParams(session: Session): Validation[List[(String, String)]] =

      params.foldLeft(HttpParam.EmptyParamListSuccess) { (resolvedParams, param) =>
        val newParams: Validation[List[(String, String)]] = param match {
          case SimpleParam(key, value) =>
            for {
              key <- key(session)
              value <- value(session)
            } yield List(key -> value.toString)

          case MultivaluedParam(key, values) =>
            for {
              key <- key(session)
              values <- values(session)
            } yield values.map(key -> _.toString)(breakOut)

          case ParamSeq(seq) =>
            for {
              seq <- seq(session)
            } yield seq.map { case (key, value) => key -> value.toString }(breakOut)

          case ParamMap(map) =>
            for {
              map <- map(session)
            } yield map.map { case (key, value) => key -> value.toString }(breakOut)
        }

        for {
          newParams <- newParams
          resolvedParams <- resolvedParams
        } yield newParams ::: resolvedParams
      }

    def resolveParamJList(session: Session): Validation[JList[Param]] = {

        def update(ahcParams: JList[Param], param: HttpParam): Validation[JList[Param]] = param match {
          case SimpleParam(key, value) =>
            for {
              key <- key(session)
              value <- value(session)
            } yield {
              ahcParams.add(new Param(key, value.toString))
              ahcParams
            }

          case MultivaluedParam(key, values) =>
            for {
              key <- key(session)
              values <- values(session)
            } yield {
              values.foreach(value => ahcParams.add(new Param(key, value.toString)))
              ahcParams
            }

          case ParamSeq(seq) =>
            for {
              seq <- seq(session)
            } yield {
              seq.foreach { case (key, value) => ahcParams.add(new Param(key, value.toString)) }
              ahcParams
            }

          case ParamMap(map) =>
            for {
              map <- map(session)
            } yield {
              map.foreach { case (key, value) => ahcParams.add(new Param(key, value.toString)) }
              ahcParams
            }
        }

        @tailrec
        def resolveParamJList(ahcParams: JList[Param], currentParams: List[HttpParam]): Validation[JList[Param]] =
          currentParams match {
            case Nil => ahcParams.success
            case head :: tail =>
              update(ahcParams, head) match {
                case Success(newAhcParams) => resolveParamJList(newAhcParams, tail)
                case f                     => f
              }
          }

      resolveParamJList(new JArrayList[Param](params.size), params)
    }
  }
}