sttp.model.MultiQueryParams.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of model_native0.3_2.11 Show documentation

model

There is a newer version: 2.0.0-RC4

package sttp.model

/**
  * Represents query parameters, where each parameter can have 0, 1, or more values.
  */
class MultiQueryParams(ps: Seq[(String, Seq[String])]) {
  def toMap: Map[String, String] = toSeq.toMap
  def toMultiMap: Map[String, Seq[String]] = ps.toMap
  def toSeq: Seq[(String, String)] = ps.flatMap { case (k, vs) => vs.map((k, _)) }
  def toMultiSeq: Seq[(String, Seq[String])] = ps

  def get(s: String): Option[String] = ps.find(_._1 == s).flatMap(_._2.headOption)
  def getMulti(s: String): Option[Seq[String]] = ps.find(_._1 == s).map(_._2)

  def param(k: String, v: String): MultiQueryParams = new MultiQueryParams(ps :+ (k -> List(v)))
  def param(k: String, v: Seq[String]): MultiQueryParams = new MultiQueryParams(ps :+ (k -> v))
  def param(p: Map[String, String]): MultiQueryParams =
    new MultiQueryParams(ps ++ p.map { case (k, v) => (k -> List(v)) })
}

object MultiQueryParams {
  def apply() = new MultiQueryParams(Nil)
  def fromMap(m: Map[String, String]): MultiQueryParams = new MultiQueryParams(m.mapValues(List(_)).toSeq)
  def fromSeq(s: Seq[(String, String)]): MultiQueryParams =
    new MultiQueryParams(s.groupBy(_._1).mapValues(_.map(_._2)).toSeq)
  def fromMultiMap(m: Map[String, Seq[String]]): MultiQueryParams = new MultiQueryParams(m.toSeq)
  def fromMultiSeq(m: Seq[(String, Seq[String])]): MultiQueryParams = new MultiQueryParams(m)
}