网站建站软件,泰安中文网站建设电话,网站建设内容规划表,怎么租域名做网站Blaze 是快手自研的基于Rust语言和DataFusion框架开发的Spark向量化执行引擎#xff0c;旨在通过本机矢量化执行技术来加速Spark SQL的查询处理。Blaze在快手内部上线的数仓生产作业也观测到了平均30%的算力提升#xff0c;实现了较大的降本增效。本文将深入剖析blaze的技术原…Blaze 是快手自研的基于Rust语言和DataFusion框架开发的Spark向量化执行引擎旨在通过本机矢量化执行技术来加速Spark SQL的查询处理。Blaze在快手内部上线的数仓生产作业也观测到了平均30%的算力提升实现了较大的降本增效。本文将深入剖析blaze的技术原理、实现细节及在快手实际生产环境中的真实表现。
一、研究背景
当下Spark 的重要发展方向之一是通过向量化执行进一步提升性能。向量化执行的思想是将算子的执行粒度从每次处理一行变成每次处理一个行组以此来避免大量的函数调用。通过对行组内部处理按列进行计算同时利用编译技术减少分支判断检查以及更多的 SIMD 优化执行计划。
Blaze 是快手自研的基于Rust语言和DataFusion框架开发的Spark向量化执行引擎旨在通过本机矢量化执行技术来加速Spark SQL的查询处理。在性能方面Blaze展现出显著的优势在TPC-DS 1TB的测试中Blaze相较于Spark 3.3版本减少了60%的计算时间、Spark 3.5版本减少了40%的计算时间并大幅降低了集群资源的消耗此外Blaze在快手内部上线的数仓生产作业也观测到了平均30%的算力提升实现了较大幅度的降本增效。
如今Blaze已开源拥抱更广阔的开发者社区。开源版本全面兼容Spark 3.0~3.5用户能够轻松集成Blaze至现有Spark环境中仅需简单添加Jar包即可解锁Blaze带来的极致性能优化享受前所未有的数据处理速度与资源效率。
Github地址 https://github.com/kwai/blaze
二、Blaze的整体架构及核心
Spark on Blaze架构的整体流向
SparkBlaze 的架构设计原理如下图 对比Spark原生的执行流程引入Blaze Session Extension组件所带来的作用是显著的特别是在提升数据处理效率和性能方面。
Spark原生执行流程主要依赖于Java虚拟机JVM进行任务的执行尽管JVM在提供跨平台、内存管理等方面有着卓越的表现但在大数据处理场景下尤其是涉及大规模数据计算和复杂查询时JVM的性能开销可能会成为瓶颈。
Blaze Session Extension组件的引入巧妙地解决了这一问题。该组件在Spark生成物理执行计划之后介入通过其翻译逻辑将这一计划转换为等效的、native向量化引擎可以识别的形式随后提交到Executor端由native引擎执行计算从而实现了数据处理效率的飞跃。 而这一切的背后离不开Native向量化引擎这一核心模块的支持。该引擎由Rust语言实现凭借其卓越的性能、安全性和并发处理能力成功实现了Spark中大多数关键算子的等效替代包括但不限于Project、Filter、Sort等。这些经过优化的算子在执行过程中通过向量化技术显著提升了计算效率使得数据处理过程更加流畅、快速。
四大核心组件
Blaze 架构中的核心模块有四个共同驱动着大数据性能的显著提升。这些模块分别为 Native Engine基于 Datafusion 框架实现的与 Spark 功能一致的 Native 算子以及相关内存管理、FFI 交互等功能。 ProtoBuf定义用于 JVM 和 native 之间的算子描述协议对 Datafusion 执行计划进行序列化和反序列化。 JNI Bridge实现 Spark Extension 和 Native Engine 之间的互相调用。 Spark ExtensionSpark 插件实现 Spark 算子到 Native 算子之间的翻译。
具体的执行过程中遵循以下步骤 物理执行计划的转换首先Spark Extension将 Spark 生成的物理执行计划转换为对应的 Native Plan 生成和提交Native Plan转换完成后Native Plan通过JNI Bridge被提交给Native Engine进行进一步的处理。 Native 执行层最后Native Engine利用其高效的内存管理机制和向量化处理技术对Native Plan进行真正的执行。执行结果通过JNI Bridge返回给Spark从而完成整个数据处理流程。
三、Blaze的技术优势面向生产的深度优化
在跑通 tpch 和 tpcds 测试集并取得预期性能提升后我们面向线上生产环境进一步做了系列深度优化包括性能和稳定性等方面工作
细粒度的FailBack机制
我们针对Spark执行效率的提升设计并实现了演进式向量化执行方案。这一方案旨在逐步优化算子与表达式的向量化覆盖同时解决Java UDF无法直接转化为Native执行的问题。通过引入细粒度的FailBack机制Blaze在翻译过程中遇到暂无Native实现的算子、单个表达式或UDF时支持算子/单个表达式粒度的回退能够灵活回退到Spark原生执行。此机制首先确定算子/表达式的依赖参数列利用Arrow FFI技术将这些参数列作为行传递给Spark进行处理然后将结果以列的形式回传至Blaze从而在JVM与Native执行之间构建了一座桥梁。
此方案不仅加速了向量化执行的全面部署还确保了即便在用户SQL中有少量UDF等不支持的场景细粒度回退单个表达式开销较小作业整体依然可以得到优化。 更高效的向量化数据传输格式
在Spark中Shuffle操作因其复杂的数据流转过程成为性能瓶颈涉及编码、压缩、网络传输、解压及解码等多个环节。原生Spark采用基于行的序列化与压缩方式而业界向量化数据则倾向于Arrow格式传输但实践中发现Arrow与主流轻量压缩算法适配不佳导致压缩率低下且存在冗余信息。针对此问题Blaze定制了优化的数据传输格式不仅去除了列名、数据类型等冗余数据还使用了byte-transpose列式数据序列化技术通过重组整型/浮点型数据的字节顺序显著提升数据压缩效率。这一改进大幅减少了Shuffle过程中的数据传输量并在实际测试与TPC-DS基准测试中展现出显著的性能提升与资源消耗降低有效解决了原有问题并优化了系统整体性能。 线上2000多个作业的真实数据上线后输入数据量小幅上涨的情况下Shuffle数据量相比spark降近30%
减少用户成本的多级内容管理策略
面对Spark与Native执行模式在内存管理上的差异我们设计了跨堆内堆外的自适应内存管理机制。该机制根据任务的向量化覆盖情况智能调整内存分配确保资源高效利用。同时我们构建了堆外内存、堆内内存与磁盘文件之间的多级管理体系有效防止了内存不足及频繁数据溢写的问题。这些措施不仅保障了向量化引擎上线后任务的稳定运行无需用户手动调整内存参数大幅降低了用户操作成本提升了整体系统的易用性与可靠性。 复杂度更优的聚合算法实现
为深度适配Spark的复杂需求Blaze在aggregate、sort、shuffle等关键算子的实现上并未直接采用DataFusion的现成方案而是进行了定制化开发。以HashAggregate为例当面对大规模group-by聚合且内存不足时Spark会转而采用基于排序的聚合这涉及高复杂度的排序与归并过程。而在Blaze中我们采用了基于分桶的归并方式利用基数排序在spill时进行分桶、溢写并在合并阶段通过hash 表快速合并整个流程保持O(n)的复杂度显著提升了聚合算子的执行效率与资源利用率。 向量化计算场景的表达式重复计算优化
针对SQL执行中算子间常见的重复表达式计算问题Blaze借鉴了Spark的Whole-stage codegen技术应用了这一项优化策略。该策略能够智能识别并合并包含重复表达式的算子如下图中的Project与Filter合并为一个大算子并在其中对表达式计算结果进行缓存、复用达到了减少重复计算、提高执行效率的目的。这一优化在应对复杂计算逻辑如JSON解析多个字段、UDF调用时尤为显著能将执行效率提升一倍以上。特别是在内部业务场景中对于高频调用的重负载UDF该优化成功减少了约40%的计算开销显著增强了系统的整体性能与响应速度。 四、当前进展及未来规划
当前进展 Blaze 作为一款高性能数据处理引擎已取得了显著进展全面支持多项核心功能展现出强大的技术实力与广泛的应用潜力。具体而言Blaze 目前已具备以下关键能力 Parquet向量化读写能力实现了对Parquet格式数据的高效向量化读写极大地提升了数据处理的速度与效率。 全面算子与表达式支持覆盖了线上常用的所有算子与表达式少量不支持的表达式和UDF也可以细粒度回退确保用户能够无缝迁移并享受向量化处理带来的性能提升。 Remote Shuffle Service集成内部集成了自研的Remote Shuffle Service同时我们也在和阿里合作增加对Apache Celeborn 的支持预计9月份可以提交到社区。 TPC-H/TPC-DS测试优异表现在业界权威的TPC-H/TPC-DS基准测试中Blaze成功通过全部测试场景并以TPC-H平均3倍以上、TPC-DS 2.5倍的性能提升展示了其在复杂查询处理上的卓越能力。 在真实的生产环境中,向量化引擎大规模上线应用算力平均提升 30%成本节约年化数千万元。
未来规划 1. 持续迭代优化内部线上推全通过不断收集用户反馈与性能数据我们将精准定位并修复潜在问题同时引入更多先进的算法与优化策略以进一步提升Blaze的性能与稳定性。
2. 支持更多引擎或场景例如数据湖等为了满足用户日益多样化的数据处理需求我们将不断拓展Blaze的应用场景支持更多类型的数据处理引擎与场景如数据湖等。通过加强与业界主流技术的兼容性我们将为用户提供更加灵活、便捷的数据处理方案助力用户解锁数据价值推动业务创新与发展。
3. 加强开源社区运营建设共建生态繁荣我们深知开源社区对于技术发展与生态繁荣的重要性。因此我们将在之后加强Blaze开源社区的运营建设积极构建一个开放、包容、协作的社区环境。当前我们已经与阿里、B站、携程、联通云等公司达成合作。
如果您对该项目感兴趣欢迎您为项目点个star。
项目地址https://github.com/kwai/blaze
