如何用Gatling(Kotlin)预热AWS Lambda以稳定响应时间断言

问题描述 #

我用Gatling(Kotlin)编写测试脚本，通过API Gateway调用Lambda函数并获取响应，采用50个并发用户测试。想要对平均响应时间做断言，但Lambda冷启动导致响应时间波动极大，断言总是失败。

当前代码如下：

override fun before() {
    println("warming up lambda")
    warmerRequest.injectOpen(atOnceUsers(10)).protocols(httpProtocol)
}

init {
    setUp(success.injectOpen(
        atOnceUsers(CONCURRENT_USERS)
    ).protocols(httpProtocol))
        .assertions(
            global().failedRequests().count().shouldBe(0L),
            global().responseTime().mean().lt(5000),
            global().successfulRequests().percent().shouldBe(100.0)
        )
}

我希望在测试开始前完成Lambda预热，让响应时间断言有效，请问有可行的实现方式吗？

解决方案 #

1. 确保预热请求完全执行后再启动主测试 #

你当前用的injectOpen是非阻塞的，Gatling会直接继续执行主测试，不会等待预热请求完成，导致Lambda还没暖起来就开始压测。可以改用阻塞式的injectAndWait方法：

override fun before() {
    println("预热Lambda")
    // 阻塞等待所有预热请求执行完成
    warmerRequest.injectAndWait(atOnceUsers(10)).protocols(httpProtocol)
}

2. 强化预热策略 #

如果一次性10个用户不够，可以延长预热时间、增加预热请求量，确保Lambda实例被创建并保持活跃：

override fun before() {
    println("预热Lambda")
    // 10秒内逐步启动20个用户发送预热请求，覆盖更多实例
    warmerRequest.injectAndWait(rampUsers(20) during (10 seconds)).protocols(httpProtocol)
}

3. 避免预热请求影响统计结果 #

如果预热请求被计入全局统计，会拉高平均响应时间。可以给预热请求打标签，断言时排除这些请求：
首先定义预热场景时添加标签：

val warmerRequest = scenario("Lambda预热")
    .exec(http("预热请求")
        .get("/your-api-endpoint")
        .tag("warmup")) // 给预热请求打标签

然后在断言时过滤掉带warmup标签的请求：

init {
    setUp(success.injectOpen(
        atOnceUsers(CONCURRENT_USERS)
    ).protocols(httpProtocol))
        .assertions(
            global().failedRequests().count().shouldBe(0L),
            // 只统计非预热请求的平均响应时间
            global().responseTime().mean().lt(5000).filter { !it.tags.contains("warmup") },
            global().successfulRequests().percent().shouldBe(100.0)
        )
}

4. 同步HTTP请求预热（备选方案） #

如果Gatling的场景预热方式仍有问题，可以直接用同步HTTP请求在before钩子中手动触发Lambda：

import java.net.URI
import java.net.http.HttpClient
import java.net.http.HttpRequest
import java.net.http.HttpResponse

override fun before() {
    println("预热Lambda")
    val httpClient = HttpClient.newHttpClient()
    val apiUrl = "你的API Gateway完整地址"
    
    // 重复发送10次请求，确保Lambda实例启动
    repeat(10) {
        val request = HttpRequest.newBuilder()
            .uri(URI.create(apiUrl))
            .GET()
            .build()
        httpClient.send(request, HttpResponse.BodyHandlers.ofString())
    }
}

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