Gatling（Scala）如何将失败用户的username写入文件？

核心思路 #

别用全局的After()钩子了——它本来就是整个模拟结束才执行一次的，根本没法处理单个用户的场景。要实现单个用户失败或退出时写入文件，得用绑定到用户会话的退出钩子，或者针对请求失败的回调逻辑，同时必须注意多线程下文件写入的线程安全问题。

方案1：用exitHook捕获所有用户退出场景 #

exitHook是跟单个用户会话绑定的，不管用户是正常跑完场景，还是中途失败退出，都会触发这个钩子。配合线程安全的文件写入方法：

import io.gatling.core.Predef._
import io.gatling.http.Predef._
import java.nio.file.{Files, Paths, StandardOpenOption}
import java.nio.charset.StandardCharsets

class UserFailureLoggingSimulation extends Simulation {
  // 加载CSV数据源
  val userFeeder = csv("users.csv").circular

  // 线程安全的文件写入工具方法
  private def logFailedUser(username: String): Unit = {
    val logContent = s"$username\n"
    Files.write(
      Paths.get("file.txt"),
      logContent.getBytes(StandardCharsets.UTF_8),
      StandardOpenOption.CREATE, // 文件不存在则创建
      StandardOpenOption.APPEND  // 追加内容而非覆盖
    )
  }

  val testScenario = scenario("UserBehaviorScenario")
    .feed(userFeeder)
    // 给每个用户绑定退出钩子：会话结束时执行
    .exitHook { session =>
      // 从会话中取出username，存在则写入文件
      session("username").asOption[String].foreach(logFailedUser)
      session // 必须返回会话对象，否则会破坏会话流程
    }
    .exec(
      http("TargetRequest")
        .get("/your-api-endpoint")
        // 可选：如果需要在请求失败时额外标记（比如注明是请求失败），可以加onFailure
        .onFailure { session =>
          session("username").asOption[String].foreach(username => 
            Files.write(Paths.get("file.txt"), s"$username (request failed)\n".getBytes, StandardOpenOption.CREATE, StandardOpenOption.APPEND)
          )
          session
        }
    )

  // 配置并发用户数和基础HTTP协议
  setUp(testScenario.inject(atOnceUsers(20))).protocols(http.baseUrl("http://your-target-host.com"))
}

方案2：精准捕获请求失败场景 #

如果只需要在特定请求失败时写入文件，而不是所有退出情况，可以用tryMax包裹请求块，失败时触发回调：

val testScenario = scenario("CriticalRequestScenario")
  .feed(userFeeder)
  // 尝试执行1次请求，失败则进入onFailure块
  .tryMax(1) {
    exec(http("CriticalAPI").get("/critical-endpoint"))
  }
  .onFailure { session =>
    session("username").asOption[String].foreach(logFailedUser)
    session
  }

重要注意点 #

• 线程安全是必须的：Gatling是多用户并行运行的，不能用普通的FileWriter这种非线程安全的类，Files.write自带的文件追加逻辑在单JVM环境下是线程安全的，能避免多用户同时写入时的内容错乱。
• 会话属性要做空检查：用asOption[String]而不是直接as[String]，防止会话中没有username时抛出异常，保证脚本稳定。
• 全局钩子别乱用：After()、Before()这类都是全局钩子，只会在整个模拟启动/结束时执行一次，完全不适合处理单个用户的场景。

内容的提问来源于stack exchange，提问作者Alex Michael Raj