0

0

Apache Spark 中重复使用 Dataset 两次时是否需要缓存?

碧海醫心

碧海醫心

发布时间:2026-01-14 10:59:19

|

237人浏览过

|

来源于php中文网

原创

Apache Spark 中重复使用 Dataset 两次时是否需要缓存?

当 dataset 仅被复用两次且仅需单列进行轻量聚合(如 count/distinct)时,通常无需显式 cache;spark 的谓词下推与投影下推可大幅减少实际 i/o,盲目缓存反而可能因全列加载和内存开销而降低性能。

在您的代码中,tradesDataset 通过 sparkSession.sql("select * from a_table") 全列加载后立即调用 .cache(),但后续两个操作均只依赖单列:

  • select("uitid").distinct().count() → 仅需 uitid 列
  • filter("TRADE_DATE = ?").count() → 仅需 TRADE_DATE 列(及隐式计数所需的最小数据)

Spark SQL 查询优化器(Catalyst)会在物理执行前自动应用列裁剪(Column Pruning)谓词下推(Predicate Pushdown)。这意味着:
✅ 即使未缓存,两次行动(action)触发的两次执行计划中,底层数据源(如 Parquet、Hive 表、JDBC)实际读取的仅为所需列,而非全表所有字段;
✅ 对于支持下推的格式(如 Parquet/ORC),I/O 与反序列化开销显著低于全表扫描;
❌ 而 .cache() 会强制将 SELECT * 的全部列持久化到内存/磁盘,不仅浪费存储与 GC 压力,还可能挤占其他任务资源。

以下是更优的写法建议:

// ✅ 推荐:按需构建窄依赖 Dataset,避免冗余列
Dataset uitIdOnly = sparkSession.sql("SELECT uitid FROM a_table");
long distinctUitIds = uitIdOnly.distinct().count();

Dataset tradeDateOnly = sparkSession.sql("SELECT TRADE_DATE FROM a_table");
long countForDate = tradeDateOnly
    .filter(col("TRADE_DATE").equalTo(processingDate))
    .count();

或进一步合并为一次扫描(若逻辑允许):

话袋AI笔记
话袋AI笔记

话袋AI笔记, 像聊天一样随时随地记录每一个想法,打造属于你的个人知识库,成为你的外挂大脑

下载
// ✅ 更高效:单次扫描 + 多重聚合(避免重复扫描)
Row result = sparkSession.sql(
        "SELECT COUNT(DISTINCT uitid) AS distinct_uitids, " +
        "       COUNT(*) FILTER (WHERE TRADE_DATE = '" + processingDate + "') AS count_for_date " +
        "FROM a_table")
    .first();
long distinctUitIds = result.getLong(0);
long countForDate = result.getLong(1);

⚠️ 注意事项:

  • 缓存收益取决于数据规模、列宽、存储格式、集群资源——对宽表(数百列)、大字段(如 JSON/BLOB)或低速数据源(如远程 JDBC),不缓存优势更明显;
  • 若后续还需多次访问多列组合复杂计算,再考虑基于具体列子集缓存(如 .select("uitid", "TRADE_DATE").cache());
  • 始终通过 EXPLAIN FORMATTED 验证执行计划,确认是否发生列裁剪(查看 Output: [uitid#123] 类日志);
  • 生产环境建议 A/B 测试:分别运行 cache() 与无 cache 版本,对比 Spark UI > SQL tab 中的 Input Rows/BytesDuration

总结:“复用两次”不是缓存的充分条件;“是否减少整体计算/IO”才是决策核心。 在轻量单列聚合场景下,信任 Catalyst 优化器,优先让 Spark 智能裁剪,而非过早缓存。

相关专题

更多
数据分析工具有哪些
数据分析工具有哪些

数据分析工具有Excel、SQL、Python、R、Tableau、Power BI、SAS、SPSS和MATLAB等。详细介绍:1、Excel,具有强大的计算和数据处理功能;2、SQL,可以进行数据查询、过滤、排序、聚合等操作;3、Python,拥有丰富的数据分析库;4、R,拥有丰富的统计分析库和图形库;5、Tableau,提供了直观易用的用户界面等等。

676

2023.10.12

SQL中distinct的用法
SQL中distinct的用法

SQL中distinct的语法是“SELECT DISTINCT column1, column2,...,FROM table_name;”。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

320

2023.10.27

SQL中months_between使用方法
SQL中months_between使用方法

在SQL中,MONTHS_BETWEEN 是一个常见的函数,用于计算两个日期之间的月份差。想了解更多SQL的相关内容,可以阅读本专题下面的文章。

346

2024.02.23

SQL出现5120错误解决方法
SQL出现5120错误解决方法

SQL Server错误5120是由于没有足够的权限来访问或操作指定的数据库或文件引起的。想了解更多sql错误的相关内容,可以阅读本专题下面的文章。

1095

2024.03.06

sql procedure语法错误解决方法
sql procedure语法错误解决方法

sql procedure语法错误解决办法:1、仔细检查错误消息;2、检查语法规则;3、检查括号和引号;4、检查变量和参数;5、检查关键字和函数;6、逐步调试;7、参考文档和示例。想了解更多语法错误的相关内容,可以阅读本专题下面的文章。

357

2024.03.06

oracle数据库运行sql方法
oracle数据库运行sql方法

运行sql步骤包括:打开sql plus工具并连接到数据库。在提示符下输入sql语句。按enter键运行该语句。查看结果,错误消息或退出sql plus。想了解更多oracle数据库的相关内容,可以阅读本专题下面的文章。

675

2024.04.07

sql中where的含义
sql中where的含义

sql中where子句用于从表中过滤数据,它基于指定条件选择特定的行。想了解更多where的相关内容,可以阅读本专题下面的文章。

571

2024.04.29

sql中删除表的语句是什么
sql中删除表的语句是什么

sql中用于删除表的语句是drop table。语法为drop table table_name;该语句将永久删除指定表的表和数据。想了解更多sql的相关内容,可以阅读本专题下面的文章。

414

2024.04.29

Java 桌面应用开发(JavaFX 实战)
Java 桌面应用开发(JavaFX 实战)

本专题系统讲解 Java 在桌面应用开发领域的实战应用,重点围绕 JavaFX 框架,涵盖界面布局、控件使用、事件处理、FXML、样式美化(CSS)、多线程与UI响应优化,以及桌面应用的打包与发布。通过完整示例项目,帮助学习者掌握 使用 Java 构建现代化、跨平台桌面应用程序的核心能力。

2

2026.01.14

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
WEB前端教程【HTML5+CSS3+JS】
WEB前端教程【HTML5+CSS3+JS】

共101课时 | 8.3万人学习

JS进阶与BootStrap学习
JS进阶与BootStrap学习

共39课时 | 3.2万人学习

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

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