在Docker化Rust Lambda函数中运行headless_chrome故障排查求助

你遇到的问题是Docker化的Rust Lambda函数创建headless Chrome实例时崩溃，本地运行镜像正常，但Lambda环境下报错Trying to work with closed connection，这是Lambda环境的特有限制导致的，以下是针对性的解决要点：

1. 配置Chrome可写临时目录 #

Lambda容器仅允许/tmp目录可写，Chrome默认的缓存/用户数据目录无法写入会导致崩溃，需显式指定到/tmp：

修改Rust代码的LaunchOptions：

use headless_chrome::{Browser, LaunchOptions};
use std::path::PathBuf;

tracing::debug!("Creating browser");
let user_data_dir = PathBuf::from("/tmp/chrome-user-data");
let options = LaunchOptions::default_builder()
    .headless(true)
    .sandbox(false)
    .window_size(Some((1920, 1080)))
    .user_data_dir(Some(user_data_dir)) // 指定可写的用户数据目录
    .build()
    .expect("Couldn't find appropriate Chrome binary.");

同时在Dockerfile中添加临时目录环境变量：

ENV TMPDIR=/tmp

2. 补充Chrome运行依赖 #

Alpine镜像的Chromium缺少Lambda环境必需的系统依赖，需在Dockerfile中补充：

RUN apk add --no-cache bash chromium chromium-chromedriver \
    fontconfig ttf-dejavu libstdc++ glib-dev nss-dev freetype-dev harfbuzz-dev ca-certificates

这些依赖是Chrome渲染页面、加载网络资源的必要组件。

3. 添加Lambda适配的Chrome启动参数 #

Lambda环境资源有限，需添加专属启动参数避免崩溃：

let options = LaunchOptions::default_builder()
    .headless(true)
    .sandbox(false)
    .window_size(Some((1920, 1080)))
    .user_data_dir(Some(user_data_dir))
    .args(vec![
        "--disable-dev-shm-usage", // 禁用小容量的/dev/shm，改用/tmp
        "--disable-gpu", // 无头模式无需GPU
        "--single-process", // 单进程减少资源占用
        "--disable-extensions",
        "--disable-plugins",
        "--disable-background-networking",
    ])
    .build()
    .expect("Couldn't find appropriate Chrome binary.");

其中--disable-dev-shm-usage是核心，Lambda的/dev/shm仅64MB，会导致Chrome内存不足。

4. 提升Lambda资源配置 #

Chrome运行需要足够内存，需将Lambda的内存配置至少调整为1024MB，对应的CPU配额也会提升，确保Chrome能正常启动。

5. 显式指定Chrome二进制路径 #

避免路径查找失败，在LaunchOptions中直接指定Chrome路径：

let options = LaunchOptions::default_builder()
    .chrome_path(PathBuf::from("/usr/bin/chromium-browser"))
    // ... 其他参数
    .build()
    .expect("Couldn't find appropriate Chrome binary.");

6. 清理残留缓存（可选） #

Lambda容器会复用，/tmp目录的旧缓存可能导致冲突，可在函数启动时清理：

use std::fs;

// 创建浏览器前清理用户数据目录
if user_data_dir.exists() {
    if let Err(e) = fs::remove_dir_all(&user_data_dir) {
        tracing::warn!("Failed to clean Chrome user data dir: {}", e);
    }
}

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