Scala中如何通过返回值在函数间传递变量（Gatling报错求助）

错误原因分析 #

1. 全局变量无法适配Gatling会话模型：你定义的全局变量a、b不属于Gatling的用户会话（Session）上下文，不同场景的用户会话是完全隔离的，fun2无法直接访问fun1设置的全局变量，这就是报错No attribute named 'a'的核心原因。
2. 函数定义不符合Gatling运行要求：exec方法需要接收返回ChainBuilder类型的操作，你写的普通函数无法融入Gatling的会话流程，且fun2存在语法错误（缺少返回类型声明）。

解决方案 #

根据你的需求，分两种场景给出正确实现：

场景1：同一用户会话内传递值（最常用） #

将两个函数整合到同一个场景中，通过Gatling的Session存储和读取数据：

import io.gatling.core.Predef.*
import io.gatling.core.scenario.Simulation
import io.gatling.core.structure.ChainBuilder
import io.gatling.http.Predef.*

// 定义fun1：将值存入Session，返回ChainBuilder
def fun1(): ChainBuilder = {
  exec(session => {
    val a = "Test1"
    val b = "Test2"
    // 将a、b存入会话上下文
    session.set("a", a).set("b", b)
  })
}

// 定义fun2：从Session读取值，返回ChainBuilder
def fun2(): ChainBuilder = {
  exec(session => {
    // 从会话中取出a、b，指定类型为String
    val a = session("a").as[String]
    val b = session("b").as[String]
    val c = "Test3"
    val d = a + b + c
    println(d) // Gatling中建议用println输出到控制台
    session // 必须返回会话对象，维持会话流程
  })
}

class TestSimulation extends Simulation {
  val httpurl = http.baseUrl("https://your-base-url.com") // 替换为你的实际baseUrl

  val testScenario = scenario("TestScenario")
    .exec(fun1())
    .exec(fun2())

  setUp(
    testScenario.inject(atOnceUsers(1))
      .protocols(httpurl)
  )
}

场景2：跨场景/跨用户共享值（特殊需求） #

如果必须在scen1和scen2之间共享数据，需要使用线程安全的全局存储容器（Gatling是多线程运行的，普通全局变量会有并发问题）：

import io.gatling.core.Predef.*
import io.gatling.core.scenario.Simulation
import io.gatling.core.structure.ChainBuilder
import io.gatling.http.Predef.*
import java.util.concurrent.atomic.AtomicReference

// 线程安全的全局共享容器
val sharedData = new AtomicReference[(String, String)]()

def fun1(): ChainBuilder = {
  exec(session => {
    val a = "Test1"
    val b = "Test2"
    sharedData.set((a, b)) // 将值存入全局容器
    session
  })
}

def fun2(): ChainBuilder = {
  exec(session => {
    val (a, b) = sharedData.get() // 从全局容器取出值
    val c = "Test3"
    val d = a + b + c
    println(d)
    session
  })
}

class TestSimulation extends Simulation {
  val httpurl = http.baseUrl("https://your-base-url.com")
  val httpurl1 = http.baseUrl("https://your-other-url.com")

  val scen1 = scenario("Test1")
    .exec(fun1())
    .inject(atOnceUsers(1))
    .protocols(httpurl)

  val scen2 = scenario("Test2")
    .exec(fun2())
    .inject(atOnceUsers(1))
    .protocols(httpurl1)

  setUp(scen1.andThen(scen2))
}

关键注意点 #

• 永远不要用普通全局变量在Gatling中传递用户数据，必须依赖Session或线程安全的全局容器。
• 所有供exec调用的函数必须返回ChainBuilder类型，内部通过操作Session来处理数据。
• 跨场景共享数据要谨慎，避免并发情况下的数据覆盖问题。

内容的提问来源于stack exchange，提问作者Quick learner