Selenium Java如何等待无提示的Excel导出完成后关闭浏览器

前置配置 #

首先需要固定浏览器下载路径，避免文件下载到系统默认目录无法追踪，同时禁用下载弹窗，配置示例如下：

// 提前创建好项目内的下载目录，示例路径为当前项目下的downloads文件夹
String downloadPath = System.getProperty("user.dir") + File.separator + "downloads";
ChromeOptions options = new ChromeOptions();
Map<String, Object> prefs = new HashMap<>();
// 配置固定下载路径
prefs.put("download.default_directory", downloadPath);
// 禁用下载确认弹窗
prefs.put("download.prompt_for_download", false);
prefs.put("profile.default_content_settings.popups", 0);
options.setExperimentalOption("prefs", prefs);
// 如需无头模式运行，需添加新无头模式参数，避免下载失效
// options.addArguments("--headless=new");
WebDriver driver = new ChromeDriver(options);

方案1：文件系统轮询检测（适配所有浏览器，兼容性最强） #

核心逻辑是轮询检测下载目录的文件状态，连续两次检测到文件大小不变即判定下载完成，避免下载中的临时文件误判。

工具方法实现 #

import org.openqa.selenium.TimeoutException;
import java.io.File;
import java.util.Arrays;
import java.util.Comparator;
import java.util.Optional;

/**
 * 等待文件下载完成
 * @param downloadPath 配置的下载目录路径
 * @param expectedFileName 预期文件名，未知文件名可传null自动匹配最新的Excel文件
 * @param timeoutSeconds 最大等待超时时间，单位秒
 * @return 下载完成的文件对象
 */
public File waitForDownloadComplete(String downloadPath, String expectedFileName, int timeoutSeconds) {
    File downloadDir = new File(downloadPath);
    long endTime = System.currentTimeMillis() + timeoutSeconds * 1000L;
    long lastFileSize = -1;
    int stableCount = 0;
    File targetFile = null;

    while (System.currentTimeMillis() < endTime) {
        // 匹配目标文件
        if (expectedFileName != null) {
            targetFile = new File(downloadDir, expectedFileName);
        } else {
            // 未知文件名时取最新修改的Excel文件
            File[] excelFiles = downloadDir.listFiles((dir, name) -> 
                name.endsWith(".xlsx") || name.endsWith(".xls")
            );
            if (excelFiles != null && excelFiles.length > 0) {
                Arrays.sort(excelFiles, Comparator.comparingLong(File::lastModified).reversed());
                targetFile = excelFiles[0];
            }
        }

        // 检测文件是否存在且大小稳定
        if (targetFile != null && targetFile.exists()) {
            long currentSize = targetFile.length();
            if (currentSize == lastFileSize) {
                stableCount++;
                // 连续2次检测大小一致，判定下载完成
                if (stableCount >= 2) {
                    return targetFile;
                }
            } else {
                stableCount = 0;
                lastFileSize = currentSize;
            }
        }
        // 每隔1秒轮询一次
        try {
            Thread.sleep(1000);
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
            throw new RuntimeException("等待下载被中断", e);
        }
    }
    throw new TimeoutException("文件下载超时，超时时间：" + timeoutSeconds + "秒");
}

调用逻辑 #

// 点击导出按钮前可先清空下载目录旧文件，避免干扰
driver.findElement(By.id("exportExcelBtn")).click();
// 等待下载完成，示例超时时间设为300秒
waitForDownloadComplete(downloadPath, "订单导出数据.xlsx", 300);
// 下载完成后再关闭浏览器
driver.quit();

方案2：CDP事件监听（仅适配Chrome内核浏览器，Selenium4+支持，精度更高） #

通过Chrome DevTools Protocol直接监听浏览器的下载进度事件，不需要扫描文件系统，适合高版本Chrome浏览器场景。

import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.devtools.DevTools;
import org.openqa.selenium.devtools.v120.browser.Browser;
import java.util.concurrent.atomic.AtomicBoolean;

// 初始化CDP监听
DevTools devTools = ((ChromeDriver) driver).getDevTools();
devTools.createSession();
devTools.send(Browser.enable());
AtomicBoolean downloadCompleted = new AtomicBoolean(false);
// 监听下载进度事件
devTools.addListener(Browser.downloadProgress(), progress -> {
    if ("completed".equals(progress.getState())) {
        downloadCompleted.set(true);
    }
});

// 点击导出按钮
driver.findElement(By.id("exportExcelBtn")).click();

// 等待下载完成，超时300秒
long endTime = System.currentTimeMillis() + 300 * 1000L;
while (System.currentTimeMillis() < endTime && !downloadCompleted.get()) {
    Thread.sleep(1000);
}
if (!downloadCompleted.get()) {
    throw new TimeoutException("下载超时");
}
driver.quit();

注意事项 #

• 大文件下载场景可适当调高超时时间，避免误判超时
• 跨平台运行时路径统一用File.separator代替硬编码的分隔符
• 未知导出文件名时优先使用方案1，适配性更强

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