如何在AWS Lambda（Java环境）中运行Selenium+Chrome浏览器

一、构建Selenium相关Lambda Layer #

Lambda Layer需打包Java依赖、Headless Chrome及ChromeDriver，注意适配Lambda的Amazon Linux 2环境（x86_64架构）：

1. 准备依赖文件 #

• Selenium Java客户端: 下载对应版本（推荐4.x）的核心jar包，放入java/lib目录（Lambda Java Layer的固定依赖路径）
• Headless Chrome: 下载适配Amazon Linux 2的Chromium二进制包，解压后将可执行文件放入bin目录
• ChromeDriver: 下载与Chrome版本完全匹配的x86_64 Linux版本，放入bin目录，执行chmod +x chromedriver赋予执行权限

2. 打包并上传Layer #

将java和bin目录打包为selenium-java-layer.zip，结构如下：

selenium-java-layer.zip
├── java
│   └── lib
│       ├── selenium-java-4.11.0.jar
│       ├── selenium-api-4.11.0.jar
│       └── ...（其他Selenium核心依赖jar）
└── bin
    ├── chrome
    └── chromedriver

上传该压缩包到AWS Lambda Layer，选择兼容的Java运行时（如java17）。

二、编写Java Lambda测试函数 #

Lambda函数需配置ChromeDriver适配Lambda环境，核心代码示例：

import com.amazonaws.services.lambda.runtime.Context;
import com.amazonaws.services.lambda.runtime.RequestHandler;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;

public class SeleniumTestHandler implements RequestHandler<String, String> {

    @Override
    public String handleRequest(String testCaseParam, Context context) {
        // 配置Headless Chrome参数
        ChromeOptions options = new ChromeOptions();
        options.addArguments("--headless=new");
        options.addArguments("--no-sandbox");
        options.addArguments("--disable-dev-shm-usage");
        options.addArguments("--disable-gpu");
        options.addArguments("--window-size=1920,1080");
        
        // 指定Layer中的Chrome和ChromeDriver路径
        System.setProperty("webdriver.chrome.driver", "/opt/bin/chromedriver");
        options.setBinary("/opt/bin/chrome");

        // 执行测试用例
        try (WebDriver driver = new ChromeDriver(options)) {
            // 此处替换为具体测试逻辑，比如根据testCaseParam加载对应测试步骤
            driver.get("https://example.com");
            String pageTitle = driver.getTitle();
            context.getLogger().log("Test executed: " + testCaseParam + " | Page title: " + pageTitle);
            return "Test passed: " + testCaseParam;
        } catch (Exception e) {
            context.getLogger().log("Test failed: " + testCaseParam + " | Error: " + e.getMessage());
            return "Test failed: " + testCaseParam;
        }
    }
}

函数部署注意事项 #

• 将编译后的class文件打包为jar上传到Lambda
• 关联之前创建的Selenium Layer
• 配置Lambda资源：内存至少512MB（建议1024MB），超时时间根据测试用例时长调整（如30-60秒）

三、批量测试用例执行方案 #

针对数百个测试用例，实现按需扩缩容：

• SQS触发: 将每个测试用例参数作为消息发送到SQS队列，配置Lambda为SQS触发器，自动按消息批量执行，Lambda会根据消息量自动扩缩容
• Step Functions编排: 创建状态机实现测试用例的并行执行、失败重试、结果汇总，适配复杂测试场景
• 并发控制: 调整Lambda预留并发数，避免超出AWS资源限制

四、关键排查点 #

• 确保Chrome与ChromeDriver版本完全一致，否则会启动失败
• 检查Lambda CloudWatch Logs，排查Chrome启动报错、测试逻辑异常
• 精简Layer依赖，避免解压后体积超过250MB的Lambda限制
• 配置Lambda权限：需CloudWatch Logs写入权限，若用SQS/Step Functions则需对应服务权限

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