在AWS Lambda中为Java+Gradle的Selenium测试提供chromedriver.exe

我来帮你梳理下这两个方案的具体操作和优劣势，你可以根据自己的场景来选择：

首先要注意一个关键细节：Lambda的执行环境是Linux系统，所以你需要下载Linux版本的chromedriver，而不是Windows的.exe文件（虽然你写了.exe，但这个后缀在Linux下没用，反而会导致执行失败）。

具体操作步骤：

1. 把下载好的Linux版chromedriver放到项目的src/main/resources目录下（或者你自定义的资源目录）
2. 修改Gradle构建脚本，确保资源文件被正确打包到部署包中：
   • 如果是Groovy风格的build.gradle：

     sourceSets {
         main {
             resources {
                 srcDirs = ['src/main/resources']
                 includes = ['chromedriver']
             }
         }
     }
     

   • 如果是Kotlin DSL风格的build.gradle.kts：

     sourceSets {
         main {
             resources {
                 srcDirs("src/main/resources")
                 include("chromedriver")
             }
         }
     }
     

3. 如果你用Gradle的Shadow插件打包Lambda部署包，需要确保chromedriver被包含进去，比如：

   shadowJar {
       mergeServiceFiles()
       manifest {
           attributes 'Main-Class': 'your.package.YourLambdaHandler'
       }
       // 将chromedriver从资源目录复制到部署包的根目录（或指定路径）
       from('src/main/resources') {
           include 'chromedriver'
           into '.' // 直接放到根目录，方便后续读取
       }
   }
   

4. 在Lambda代码中读取并配置chromedriver：
   Lambda的/var/task目录是只读的，所以需要把chromedriver复制到可写的/tmp目录，并设置可执行权限：

   import java.io.File;
   import java.io.InputStream;
   import java.nio.file.Files;
   import java.nio.file.StandardCopyOption;
   
   public class YourLambdaHandler {
       public void handleRequest() {
           try {
               // 从部署包中读取chromedriver资源
               InputStream driverStream = getClass().getResourceAsStream("/chromedriver");
               File driverFile = new File("/tmp/chromedriver");
               // 复制到/tmp目录
               Files.copy(driverStream, driverFile.toPath(), StandardCopyOption.REPLACE_EXISTING);
               // 设置可执行权限
               driverFile.setExecutable(true);
               // 配置ChromeDriver路径
               System.setProperty("webdriver.chrome.driver", driverFile.getAbsolutePath());
               
               // 后续的Selenium测试逻辑...
           } catch (Exception e) {
               e.printStackTrace();
           }
       }
   }
   

这个方案适合需要频繁更新chromedriver，或者部署包体积接近Lambda上限的场景。

具体操作步骤：

1. 下载Linux版chromedriver，上传到你的S3存储桶（比如路径s3://your-bucket-name/chromedriver）
2. 给Lambda的IAM执行角色添加S3读取权限，允许它从目标桶获取对象
3. 在Lambda代码启动时，从S3下载chromedriver到/tmp目录并配置：

   import com.amazonaws.services.s3.AmazonS3;
   import com.amazonaws.services.s3.AmazonS3ClientBuilder;
   import com.amazonaws.services.s3.model.GetObjectRequest;
   import java.io.File;
   
   public class YourLambdaHandler {
       public void handleRequest() {
           try {
               // 初始化S3客户端
               AmazonS3 s3Client = AmazonS3ClientBuilder.defaultClient();
               String bucketName = "your-bucket-name";
               String driverKey = "chromedriver";
               File driverFile = new File("/tmp/chromedriver");
               
               // 从S3下载到/tmp
               s3Client.getObject(new GetObjectRequest(bucketName, driverKey), driverFile);
               // 设置可执行权限
               driverFile.setExecutable(true);
               // 配置ChromeDriver路径
               System.setProperty("webdriver.chrome.driver", driverFile.getAbsolutePath());
               
               // 后续的Selenium测试逻辑...
           } catch (Exception e) {
               e.printStackTrace();
           }
       }
   }
   

两种方案的优劣势对比 #

• 直接打包进部署包
  • ✅ 优点：部署流程简单，不需要额外依赖S3，Lambda冷启动速度更快（无需下载文件）
  • ❌ 缺点：部署包体积增大（chromedriver约150MB），可能接近Lambda的部署包上限（直接上传50MB，通过S3上传250MB）；更新chromedriver需要重新构建并部署Lambda
• 存储在S3
  • ✅ 优点：部署包体积更小，更新chromedriver只需替换S3中的文件，无需重新部署Lambda；适合驱动版本频繁迭代的场景
  • ❌ 缺点：Lambda冷启动时需要额外的下载步骤，增加启动时间；需要配置IAM权限，多了一层S3依赖

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