如何在Gatling执行exitHereIfFailed前触发失败清理代码？

你可以通过以下两种方式实现“失败时先清理数据库再调用exitHereIfFailed”的需求：

方式一：结合tryMax的failure分支处理 #

将核心业务逻辑用tryMax包裹，在失败分支中先执行数据库清理动作，再退出会话：

步骤1：封装数据库清理逻辑 #

先把清理数据库的操作封装成Gatling的ChainBuilder，方便复用：

import io.gatling.jdbc.Predef._

def cleanCorruptedDatabaseData(): ChainBuilder = exec(
  jdbc("Clean corrupted records")
    .update("DELETE FROM target_table WHERE status = 'incomplete'") // 替换为你的清理SQL
)

步骤2：在场景中整合失败处理 #

val yourCoreScenario = scenario("Business Flow")
  .tryMax(1) { // 尝试执行核心逻辑1次
    exec(yourBusinessStep1)
    .exec(yourBusinessStep2)
    // 其他核心业务步骤
  }
  .failure { // 核心逻辑执行失败时进入此分支
    exec(cleanCorruptedDatabaseData()) // 优先执行数据库清理
    .exitHereIfFailed // 清理完成后退出当前会话
  }

方式二：通过请求的failureHandler触发清理 #

如果需要针对单个请求的失败做清理，可以使用请求的failureHandler回调，在请求失败时先执行清理，再标记会话为失败：

exec(
  http("Critical API Call")
    .post("/api/submit")
    .body(StringBody("${payload}"))
    .check(status.is(201))
    .failureHandler { (session, response) =>
      // 执行数据库清理
      session.exec(cleanCorruptedDatabaseData())
        .markAsFailed // 手动标记会话为失败，后续可触发exitHereIfFailed
    }
)
.exitHereIfFailed // 清理完成后退出

注意事项 #

• 确保清理SQL的准确性，避免误删正常数据；
• Gatling的Session是不可变对象，在failureHandler中必须返回处理后的Session实例；
• 如果使用JDBC操作数据库，需要提前在Gatling配置中配置好数据库连接信息。

内容的提问来源于stack exchange，提问作者deven you