0

0

优化 Selenium 表格爬取与 Excel 写入性能的实战指南

霞舞

霞舞

发布时间:2026-02-21 10:56:02

|

319人浏览过

|

来源于php中文网

原创

优化 Selenium 表格爬取与 Excel 写入性能的实战指南

本文针对使用 selenium + java 爬取网页表格并逐单元格写入 excel 时出现严重性能下降的问题,揭示根本原因(高频文件 i/o),并提供基于内存批量写入的完整优化方案。

本文针对使用 selenium + java 爬取网页表格并逐单元格写入 excel 时出现严重性能下降的问题,揭示根本原因(高频文件 i/o),并提供基于内存批量写入的完整优化方案。

在实际 Web 数据采集项目中,开发者常误将“功能实现”等同于“性能可用”。您提供的 CovidWebTable 示例代码虽能成功抓取 Worldometers 的疫情表格并写入 Excel,但其执行速度随行数增长而急剧恶化——这并非 Selenium 本身慢,而是Excel I/O 策略设计缺陷所致

? 根本问题:每次写入都触发完整文件读-写-关闭循环

观察 XLUtils.setCellData(...) 方法:

  • 每次调用都会 new FileInputStream(xlFile) 重新加载整个 .xls 文件;
  • 创建 HSSFWorkbook 解析二进制结构;
  • 定位 Sheet/Row/Cell 并写入;
  • 最后 wb.write(fo) 全量覆写文件并关闭流。

这意味着:若表格含 231 行 × 15 列 = 3465 个单元格,您的程序将重复打开/解析/写入/关闭 Excel 文件 3465 次。磁盘 I/O 成为绝对瓶颈,且随着写入次数增加,文件系统缓存失效、JVM GC 压力上升,导致“越往后越慢”的典型现象。

Cleanup.pictures
Cleanup.pictures

智能移除图片中的物体、文本、污迹、人物或任何不想要的东西

下载

✅ 正确解法:内存聚合 + 单次持久化

核心思想:将所有待写数据暂存于内存(如 List>),待全部采集完成后再一次性写入 Excel。这相当于用一辆大货车(单次 I/O)替代数千次快递小哥(逐单元格 I/O)。

✅ 优化后的主流程(关键修改部分)

// 1. 内存中构建二维数据结构(替代逐单元格写入)
List<List<String>> excelData = new ArrayList<>();

// 写入表头(跳过第0列,按原逻辑)
List<String> headerRow = new ArrayList<>();
for (int col = 1; col < header.size() - 1; col++) {
    headerRow.add(header.get(col).getText());
}
excelData.add(headerRow);

// 2. 遍历表格行,逐行提取并添加到内存列表
int xlRow = 1;
for (int r = 1; r < rows.size(); r++) {
    WebElement rowElement = rows.get(r);
    if (rowElement.getText().trim().isEmpty()) {
        System.out.println("Skipped empty row: " + r);
        continue;
    }
    System.out.println("Processing row " + r);

    List<String> rowData = new ArrayList<>();
    for (int c = 1; c < header.size(); c++) {
        try {
            // 使用更稳定的 CSS 选择器替代复杂 XPath
            String cellText = rowElement.findElement(By.cssSelector("td:nth-of-type(" + (c + 1) + ")")).getText().trim();
            rowData.add(cellText);
        } catch (Exception e) {
            rowData.add(""); // 容错:缺失单元格填空字符串
        }
    }
    excelData.add(rowData);
}

// 3. 【关键】仅调用一次批量写入方法
xl.writeAllData("Covid Data", excelData); // 新增方法,见下方
System.out.println("✅ Export completed. Total rows written: " + excelData.size());

✅ 在 XLUtils 中新增高效批量写入方法

// 新增方法:接收完整二维数据,单次 I/O 写入
public static void writeAllData(String xlFile, String xlSheet, List<List<String>> data) throws IOException {
    FileInputStream fi = new FileInputStream(xlFile);
    HSSFWorkbook wb = new HSSFWorkbook(fi);
    HSSFSheet ws = wb.getSheet(xlSheet);

    // 清空旧数据(可选,确保干净写入)
    ws.removeAllCells();

    // 逐行写入内存数据
    for (int rowNum = 0; rowNum < data.size(); rowNum++) {
        HSSFRow row = ws.createRow(rowNum);
        List<String> rowData = data.get(rowNum);
        for (int colNum = 0; colNum < rowData.size(); colNum++) {
            HSSFCell cell = row.createCell(colNum);
            cell.setCellValue(rowData.get(colNum));
        }
    }

    // 【唯一一次】写入磁盘并关闭
    FileOutputStream fo = new FileOutputStream(xlFile);
    wb.write(fo);
    fo.close();
    wb.close();
    fi.close();
}

⚠️ 关键注意事项与增强建议

  • 避免隐式等待滥用:implicitlyWait 在查找动态表格时易引发不可预测延迟。推荐改用显式等待(WebDriverWait)+ ExpectedConditions.presenceOfAllElementsLocatedBy 精准等待表格加载完成。
  • XPath 性能陷阱:原代码中 rows.get(r).findElement(By.xpath(".//td[...")) 在每行内重复解析 XPath。改用 By.cssSelector("td:nth-of-type(n)") 更轻量且稳定。
  • 异常健壮性:网络波动或 DOM 变化可能导致 findElement 抛异常。务必包裹 try-catch 并提供默认值(如空字符串),防止中断整个流程。
  • 内存安全提醒:对于超大表格(>10 万行),需评估 JVM 堆内存。此时应考虑分块写入(如每 5000 行 flush 一次)或切换至流式 API(Apache POI SXSSFWorkbook)。
  • 现代格式兼容:.xls(HSSF)已属遗留格式。建议升级至 .xlsx(XSSF/SXSSF),支持更大行列数及更好性能。

✅ 性能提升效果预估

操作方式 文件 I/O 次数 预估耗时(231×15 表) CPU/IO 负载
原代码(逐单元格) ~3465 次 8–15 分钟(机械硬盘) 极高
优化后(批量写入) 1 次 极低

? 总结:Selenium 爬虫的性能瓶颈 rarely 在浏览器自动化层,而常源于下游数据处理的反模式。牢记——I/O 是最昂贵的操作,永远优先聚合、延迟执行、批量处理。将数据从“边爬边存”重构为“先存后写”,是提升采集效率最立竿见影的工程实践。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
string转int
string转int

在编程中,我们经常会遇到需要将字符串(str)转换为整数(int)的情况。这可能是因为我们需要对字符串进行数值计算,或者需要将用户输入的字符串转换为整数进行处理。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

810

2023.08.02

js 字符串转数组
js 字符串转数组

js字符串转数组的方法:1、使用“split()”方法;2、使用“Array.from()”方法;3、使用for循环遍历;4、使用“Array.split()”方法。本专题为大家提供js字符串转数组的相关的文章、下载、课程内容,供大家免费下载体验。

595

2023.08.03

js截取字符串的方法
js截取字符串的方法

js截取字符串的方法有substring()方法、substr()方法、slice()方法、split()方法和slice()方法。本专题为大家提供字符串相关的文章、下载、课程内容,供大家免费下载体验。

217

2023.09.04

java基础知识汇总
java基础知识汇总

java基础知识有Java的历史和特点、Java的开发环境、Java的基本数据类型、变量和常量、运算符和表达式、控制语句、数组和字符串等等知识点。想要知道更多关于java基础知识的朋友,请阅读本专题下面的的有关文章,欢迎大家来php中文网学习。

1556

2023.10.24

字符串介绍
字符串介绍

字符串是一种数据类型,它可以是任何文本,包括字母、数字、符号等。字符串可以由不同的字符组成,例如空格、标点符号、数字等。在编程中,字符串通常用引号括起来,如单引号、双引号或反引号。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

641

2023.11.24

java读取文件转成字符串的方法
java读取文件转成字符串的方法

Java8引入了新的文件I/O API,使用java.nio.file.Files类读取文件内容更加方便。对于较旧版本的Java,可以使用java.io.FileReader和java.io.BufferedReader来读取文件。在这些方法中,你需要将文件路径替换为你的实际文件路径,并且可能需要处理可能的IOException异常。想了解更多java的相关内容,可以阅读本专题下面的文章。

986

2024.03.22

php中定义字符串的方式
php中定义字符串的方式

php中定义字符串的方式:单引号;双引号;heredoc语法等等。想了解更多字符串的相关内容,可以阅读本专题下面的文章。

939

2024.04.29

go语言字符串相关教程
go语言字符串相关教程

本专题整合了go语言字符串相关教程,阅读专题下面的文章了解更多详细内容。

186

2025.07.29

pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法
pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法

本专题系统整理pixiv网页版官网入口及登录访问方式,涵盖官网登录页面直达路径、在线阅读入口及快速进入方法说明,帮助用户高效找到pixiv官方网站,实现便捷、安全的网页端浏览与账号登录体验。

928

2026.02.13

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Excel 教程
Excel 教程

共162课时 | 18.3万人学习

成为PHP架构师-自制PHP框架
成为PHP架构师-自制PHP框架

共28课时 | 2.6万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号