如何使用 Gatling 遍历目录中的所有文件？

Question

我需要遍历目录中的多个文件，以便对每个文件执行文件上传。该目录位于src/test/resources。

我了解 Gatling 的文件馈送器，但我没有看到任何允许我从目录中查找文件的文件（文件的名称是任意的，如果可能的话，不应该是测试中的硬代码） .

最好的方法是什么？

Answer 1

首先，您需要带文件的进纸器。那是由Maps 组成的Array。此映射需要以 String 作为键，并且每个文件都有自己的映射，其中包含我们需要的内容。

假设我们只需要一个名字，那么这样的东西应该可以工作：

def getFilePropsFromDir(dir: String): Array[Map[String, String]] = {
  val d = new File(dir)

  if (d.exists && d.isDirectory) {
    d.listFiles.filter(_.isFile).map(x => Map("path" -> x.toString))
  } else {
    Array()
  }
}

val feederWithFiles = getFilePropsFromDir("src/test/resources/dir_with_files/")

那么你需要这样的场景（我不上传任何东西）

val sut = scenario("Just feed files and query google")
    .feed(feederWithFiles)
    .exec(session => {
        val path = session("path").as[String] // get values by key from map - we had only "path" there
        println(path)
        val fileToUpload = getFileToUpload(path) // dummy function

        session.setAll(
          // prepare data for using later. Also k -> v
          ("fileToUpload", fileToUpload), 
          // another entry to illustrate how to use session elements
          ("googleAddress","http://google.com") 
        )
      }
    )
    .exec(
      exec(
          http("Should do something that uploads, but I just GET google")
            .get("${googleAddress}") // access key "googleAddress" from session
        )
    )

  setUp(
    sut.inject(rampUsers(1).during(1.seconds))
  ).protocols(http)

  def getFileToUpload(path: String): String = {
    path
  }

我创建了 2 个文件，GET 被执行了 2 次。现在你需要弄清楚如何上传。

我拥有的进口商品：

import io.gatling.core.Predef._
import io.gatling.core.body.StringBody
import io.gatling.core.structure.ChainBuilder
import io.gatling.http.Predef.http
import java.io.File
import scala.concurrent.duration._
import scala.io.Source