Gatling JsonPath

gatling-jsonpath.scala

package utils

import com.fasterxml.jackson.databind.node.TextNode
import com.fasterxml.jackson.databind.{JsonNode, ObjectMapper}
import io.gatling.jsonpath.JsonPath

object JsonPathUtil {

  private lazy val mapper = new ObjectMapper

  private def parseJson(s: String): JsonNode = mapper.readValue(s, classOf[JsonNode])

  def extract(path: String, json: String) = {
    JsonPath.query(path, parseJson(json)) match {
      case Right(v) =>
        v.toVector.map {
          case text: TextNode => text.textValue()
          case other: JsonNode => other.toString
        }
      case Left(e) => println(s"Jsonpath [$path] isn't valid : $e"); Vector()
    }
  }

  def jsonPrettyPrint(value: String): String = {
    val jsonObject = mapper.readValue(value, classOf[Any])
    mapper.writerWithDefaultPrettyPrinter.writeValueAsString(jsonObject)
  }

}