编者按:高可用架构推出 2015 年度案例系列文章,分享在架构领域具有典型意义的年度案例,本文由谭政分享。转载请注明来自高可用架构公众号「ArchNotes」。


谭政,Hulu 网大数据基础平台研发。曾在新浪微博平台工作过。专注于大数据存储和处理,对 Hadoop、HBase 以及 Spark 等等均有深入的了解。

Spark 最新的特性以及功能

大数据pyspark写多表查询 大数据spark案例_大数据pyspark写多表查询

2015 年中 Spark 版本从 1.2.1 升级到当前最新的 1.5.2,1.6.0 版本也马上要进行发布,每个版本都包含了许多的新特性以及重要的性能改进,我会按照时间顺序列举部分改进出来,希望大家对 Spark 版本的演化有一个稍微直观的认识。

由于篇幅关系,这次不能给大家一一讲解其中每一项改进,因此挑选了一些我认为比较重要的特性来给大家讲解。如有遗漏和错误,还请帮忙指正。

Spark 版本演化

首先还是先来看一下 Spark 对应版本的变化:

大数据pyspark写多表查询 大数据spark案例_初始化_02

先来一个整体的介绍:1.2 版本主要集中于 Shuffle 优化, 1.3 版本主要的贡献是 DataFrame API, 1.4 版本引入 R API 并启动 Tungsten 项目阶段,1.5 版本完成了 Tungsten 项目的第一阶段,1.6 版本将会继续进行 Tungsten 项目的第二个阶段。而我下面则重点介绍 DataFrame API 以及 Tungsten 项目。

DataFrame 介绍

DataFrame API 是在 1.3.0 中引入的,其目的是为了统一 Spark 中对结构化数据的处理。在引入 DataFrame 之前,Spark 之有上针对结构化数据的 SQL 查询以及 Hive 查询。


这些查询的处理流程基本类似:查询串先需要经过解析器生成逻辑查询计划,然后经过优化器生成物理查询计划,最终被执行器调度执行。

而不同的查询引擎有不同的优化器和执行器实现,并且使用了不同的中间数据结构,这就导致很难将不同的引擎的优化合并到一起,新增一个查询语言也是非常艰难。


为了解决这个问题,Spark 对结构化数据表示进行了高层抽象,产生了 DataFrame API。简单来说 DataFrame 可以看做是带有 Schema 的 RDD(在1.3之前DataFrame 就叫做 SchemaRDD,受到 R 以及 Python 的启发改为 DataFrame这个名字)。


在 DataFrame 上可以应用一系列的表达式,最终生成一个树形的逻辑计划。这个逻辑计划将会经历 Analysis, Logical Optimization, Physical Planning 以及 Code Generation 阶段最终变成可执行的 RDD,如下图所示:


大数据pyspark写多表查询 大数据spark案例_数据_03


在上图中,除了最开始解析 SQL/HQL 查询串不一样之外,剩下的部分都是同一套执行流程,在这套流程上 Spark 实现了对上层 Spark SQL, Hive SQL, DataFrame 以及 R 语言的支持。


大数据pyspark写多表查询 大数据spark案例_API_04

下面我们来看看这些语言的简单示例:
Spark SQL : val count = sqlContext.sql("SELECT COUNT(*) FROM records").collect().head.getLong(0)


大数据pyspark写多表查询 大数据spark案例_API_05


大数据pyspark写多表查询 大数据spark案例_API_06


大数据pyspark写多表查询 大数据spark案例_API_07


各个语言的使用方式都很类似。除了类 SQL 的表达式操作之外,DataFrame 也提供普通的类似于 RDD 的转换,例如可以写如下代码:


大数据pyspark写多表查询 大数据spark案例_数据_08

另外还值得一提的是,和 DataFrame API 紧密相关的 API -- DataSource API。如果说 DataFrame API 提供的是对结构化数据的高层抽象,那么 DataSource API 提供的则是对于结构化数据统一的读写接口。

DataSource API 支持从 JSON, JDBC, ORC, parquet 中加载结构化数据 (SQLContext 类中的诸多读取方法,均会返回一个 DataFrame 对象),也同时支持将 DataFrame 的数据写入到上述数据源中 (DataFrame 中的 save 系列方法 )。

这两个 API 再加上层多种语言的支持,使得 Spark 对结构化数据拥有强大的处理能力,极大简化了用户编程工作。

Tungsten 项目介绍

在官方介绍中 Tungsten 将会是对 Spark 执行引擎所做的最大的修改,其主要目标是改进 Spark 内存和 CPU 的使用效率,尽可能发挥出机器硬件的最大性能。

之所以将优化的重点集中在内存和 CPU 而不是 IO 之上是社区实践发现现在很多的大数据应用的瓶颈在 CPU 。例如目前很多网络 IO 链路的速度达到 10Gbps,SSD 硬盘和 Striped HDD 阵列的使用也使得磁盘 IO 也有较大提升。而 CPU 的主频却没有多少提升,CPU 核数的增长也不如前两者迅速。

此外在 Spark 已经对 IO 做过很多的优化(如列存储以及 IO 剪枝可以减少 IO的数据量,优化的 Shuffle 改善了 IO 和网络的传输效率),再继续进行优化提升空间并不大。

而随着序列化以及 Hash 的广泛使用,现在 CPU 反而成为了一个瓶颈。

内存方面,使用 Java 原生的堆内存管理方式很容易产生 OOM 问题,并伴随着较大的 GC 负担,进一步降低了 CPU 的利用率。

基于上述观察 Spark 在 1.4 中启动了 Tungsten 项目,并在 1.5 中完成第一阶段的优化

这些优化包括下面三个方面:


  • 内存管理和二进制格式处理
  • 缓存友好的计算
  • 代码生成

内存管理和二进制格式处理
避免以原生格式存储 Java 对象(使用二进制的存储格式),减少 GC 负担
压缩内存数据格式,减少内存占用以及可能的溢写。使用更准确的内存的统计而不是依赖启发规则管理内存。

对于那些已知数据格式运算( DataFrame 和 SQL ),直接使用二进制的运算,避免序列化和反序列化开销。

缓存友好的计算
更加快的排序以及 Hash,优化 Aggregation, Join 以及 Shuffle 操作。

代码生成
更快的表达式计算以及 DataFrame/SQL 运算(这是代码生成的主要应用场景,主要是为了降低进行表达式评估中 JVM 的各种开销,如虚函数调用,分支预测,原始类型的对象装箱开销以及内存消耗)更快的序列化。

相关的每个版本所做的优化如下:

大数据pyspark写多表查询 大数据spark案例_初始化_09


Tungsten 项目并不是完全是一个通用的优化技术,其中很多优化利用了 DataFrame 模型所提供的丰富的语义信息(因此 DataFrame 和 Spark SQL 查询能够享受该项目所来的大量的好处),同样未来也会改进 RDD API 来为底层优化提供更多的信息支持。

Spark 在 Hulu 的实践

Hulu 是一家在线付费视频网站,每天都有大量的用户观看行为数据产生,这些数据会由 Hulu 的大数据平台进行存储以及处理。推荐团队需要从这些数据中挖掘出单个用户感兴趣的内容并推荐给对应的观众,广告团队需要根据用户的观看记录以及行为给其推荐的最合适广告,而数据团队则需要分析所有数据的各个维度并为公司的策略制定提供有效支持。

他们的所有工作都是在 Hulu 的大数据平台上完成的,该平台由 HDFS/Yarn, HBase, Hive, Cassandera 以及的 Presto,Spark 等组成。Spark 是运行在 Yarn上,由 Yarn 来管理资源并进行任务调度。

Spark 则主要有两类应用:Streaming 应用以及短时 Job。

Streaming 应用中各个设备前端将用户的行为日志输入到 Kafka 中,然后由 Spark Streaming 来进行处理,输出结果到 Cassandera, HBase 以及 HDFS 中。短时 Job 并不像 Streaming 应用一样一直运行,而是由用户或者定时脚本触发,一般运行时间从几分钟到十几个小时不等。

此外为了方便 PM 类型的用户更便捷的使用 Spark,我们也搭建了 Apache Zeppelin 这种交互式可视化执行环境。对于非 Python/Scala/Java/R 用户(例如某些用户想在 NodeJS 中提交 Spark 任务),我们也提供 REST 的 Spark-JobServer 来方便用户提交作业。

Hulu 从 0.9 版本就开始将 Spark 应用于线上作业,内部经历了 1.1.1, 1.2.0, 1.4.0 等诸多版本,目前内部使用的最新版本是基于社区 1.5.1 进行改造的。

在之前的版本中我们遇到的很多的问题也添加了不少新功能,大部分修改都已经包含在最新版本里面,我就不再这里赘述了。这节里我主要想讲的是社区里所没有的,但是我们认为还比较重要的一些修改。

较多的迭代触发 StackOverflow 的问题

在某些机器学习算法里面需要进行比较多轮的迭代,当迭代的次数超过一定次数时候应用程序就会发生 StackOverflow 而崩溃。这个次数限制并不会很大,几百次迭代就可能发生栈溢出。大家可以利用一小段代码来进行一个简单的测试:

大数据pyspark写多表查询 大数据spark案例_API_10

产生上述错误的原因在于 Driver 将 RDD 任务发送给 Executor 执行的时候需要将 RDD 的信息序列化后广播到对应的 Executor 上。而 RDD 在序列的时候需要递归将其依赖的 RDD 序列化,这样在出现长 lineage 的 RDD 的时候就可能因为线程的栈帧内存不够,抛出 StackOverflow 异常。

解决方法也比较直接,就是将递归改为迭代,把原来需要递归保存在线程栈帧的序列化 RDD 挪到堆区进行保存。具体的做法是将 RDD 的依赖关系分离出来,变成两个映射表: rddId->List of depId 以及depId -> Dependency。Driver 端然后将 RDD 以及这些映射序列化为字节数组广播出去,Executor 端接收到广播消息后重新将映射组装成为原始的依赖。

这个过程中要改动 RDD 核心 Task 接口,需要经过严格的测试。但是在做这种优化之后,迭代个一两千次都没有什么问题。

Streaming 延迟数据接收机制( Receive-Base )

在 Receive-Base 的 Spark Streaming 的架构中, 主要有两个角色 Driver 和 Executor。

在 Executor 中运行着 Receiver, Receiver 的主要作用是从外部接收数据并缓存到本地内存中,同时 Receiver 回向 Driver 汇报自己所接收的数据块,Driver 定期产生新的任务并分发到各个 Executor 去处理这些数据。

在应用启动的时候,Driver 会首先将 Receiver 处理程序调度到各个 Executor 上让其初始化。一旦 Receiver 初始化完毕,它就开始源源不断的接收数据,并且需要 Driver 定期调度任务来消耗这些数据。

但是在某些场景下, Executor 处理端还并没有准备好,无法开始处理数据。

这时候在 Receiver 端就会发生内存积压,随着积压的数据越来越多,大部分数据都会撑过新生代回收年龄进入老年代,进一步给 GC 带来严重的压力,这个时候也就离应用程序崩溃不远了。

在 Hulu 的 Spark Streaming 处理中,需要加载并初始化很多机器学习的模型,这些模型的初始化非常费时间,长的可能需要半个小时才能初始化完毕。在此期间 Receiver 不能接收数据,否者内存将会被消耗殆尽。

Hulu 中的解决方法是在每个 Executor 接收任何任务之前先进行执行一个用户定义的初始化任务,初始化任务中可以执行一些独立的用户代码。我们在新增了一个接口,让用户可以设置自定义的初始化任务。

如下代码所示:

大数据pyspark写多表查询 大数据spark案例_API_11


实现上需要更改 Spark 的任务调度器,先将每个 Executor 设置为未初始化状态,除了初始化任务之外调度器不会给未初始化状态的 Executor 分配其他任务。等 Executor 运行完初始化任务,调度器更新 Executor 的状态为已初始化,这样的 Executor 就可以分配给其他正常任务了,包括初始化 Receiver 的任务。

其他注意事项

Spark 允许用户设置 spark.executor.userClassPathFirst,这可以部分缓解用户代码库和 Spark 系统代码库冲突的问题。

但是在实践过程中我们发现,大并发情况加载相同的类有可能发生死锁的情况(我们的一个场景下有 1/10 几率复现该问题)。

其问题在于 Spark 所新增加的 ChildFirstURLClassLoader 的实现引入了并发死锁的问题。

Java 7 中的 ClassLoader 本身提供细粒度的类加载并发锁,可以做到为每个 classname 设置一个锁,但是使用该细粒度的类加载锁有一个条件,用户自己实现的 ClassLoader 必须在自身静态初始化方法中将自己注册到 ClassLoader 中。

然而在 Scala 语言中并没有类的静态初始化方法,只有一个伴生对象的初始化方法。但是伴生对象和类对象的类型并不完全一致。

因此 Scala 在 ChildFirstURLClassLoader 中模仿 Java 的 ClassLoader 实现了自己的细粒度的类加载锁,然而这段代码却无法达到预期目的,最终还是会降级到 ClassLoader 级别的锁,并且在某些场景下还会触发死锁,解决方法是去除对应的细粒度锁代码。

Spark 未来的发展趋势

Spark 1.6 即将发布,其中最重要的特性有两个 [SPARK-10000] 统一内存管理以及 [SPARK-9999] DataSet API。当然还有很多其他的改进,由于篇幅关系,下面主要介绍上两个。

统一内存管理

在 1.5 以及之前存在两个独立的内存管理:执行时内存管理以及存储内存管理,前者是在对 Shuffle, Join, Sort, Aggregation 等计算的时候所用到的内存,后者是缓存以及广播变量时用的内存。

可以通过 spark.storage.memoryFraction 来指定两部分的大小,默认存储占据 60% 的堆内存。这种方式分配的内存都是静态的,需要手动调优以避免 spill,且没有一个合理的默认值可以覆盖到所有的应用场景。

在 1.6 中这两个部分内存管理被统一起来了,当执行时内存超过给自己分配的大小时可以临时向存储时内存借用空间,临时借用的内存可以在任何时候被回收,反之亦然。更进一步可以设置存储内存的最低量,系统保证这部分量不会被剔除。

DataSet API

RDD API 存储的是原始的 JVM 对象,提供丰富的函数式运算符,使用起来很灵活,但是由于缺乏类型信息很难对它的执行过程优化。DataFrame API 存储的是 Row 对象,提供了基于表达式的运算以及逻辑查询计划,很方便做优化,并且执行起来速度很快,但是却不如 RDD API 灵活。

DataSet API 则充分结合了二者的优势,既允许用户很方便的操作领域对象又拥有 SQL 执行引擎的高性能表现。

本质上来说 DataSet API 相当于 RDD + Encoder, Encoder 可以将原始的 JVM对象高效的转化为二进制格式,使得可以后续对其进行更多的处理。目前是实现为 Catalyst 的逻辑计划,这样就能够充分利用现有的 Tungsten 的优化成果。

DataSet API 需要达到如下几点目标:

  • 快速:Encoder 需要至少和现有的 Kryo 或者 Java 序列一样快。
  • 类型安全:在操作这些对象的时候需要尽可能提供编译时的类型安全,如果编译期无法知晓类型,在发生 Schema 不匹配的时候需要快速失败。
  • 对象模型支持:默认需要提供原子类型,case 类,tuple, POJOs, JavaBeans 的 Encoder。
  • Java 兼容性:需要提供一个简单的 API 来兼容 Scala 和 Java,尽可能使用这些 API,如果实在不能使用这些API也需要提供重载的版本。
  • DataFrame 的互操作:用户需要能够无缝的在 DataSet API 和 DataFrame API之间做转换。

目前 DataSet API 和 DataFrame API 还是独立的两个 API,未来 DataFrame 有可能继承自 DataSet[Row]。

最后再来看一下整体的架构:

大数据pyspark写多表查询 大数据spark案例_数据_12

1、在 hulu, streaming 跑在多少个节点上?Zeppelin 和 sparknotebook.io 各有什么优劣、是如何选型的?

hulu 的 Spark Streaming 运行在 YARN 上,规模是几百个节点。我们当前主要用的是 Zeppelin,sparknotebook.ion 目前还没有试用

2、我们用的是 hive on spark 模式,因为 hive 是统一入口,上面已经有 mr 和 tez,请问对比 spark sql 各自优缺点?还有就到对比一下 spark shuffle 和 yarn自带 shuffle(on yarn 模式)的优缺点?

底层的存储引擎不一样,相比于性能方面 spark 和 tez 不相上下,但是稳定性方面 spark 更胜一筹。spark shuffle 提供了三种实现,分别是 hash-based,sort-based 和 tungsten-sort, 而 mapreduce shuffle 知识 sort-based,在灵活度上,spark 更高,且个别之处,spark 有深度优化。

3、能否简单说说 spark 在图片计算方面的应用?

是指图像处理方面吗,这方面 Spark 并没有专门的组件来处理。图片方面的应用比较少,至少在 hulu 没有。

4、Tungsten 项目目前成熟吗?或者说贵司有线上应用没?

Tungsten 项目还处于开发阶段(阶段二),不建议在线上使用。

5、请问使用 Spark streaming 在 YARN 上和其他任务共同运行,稳定性如何?YARN 有没有做 CPU 级别的隔离?我们在 YARN 上运行的任务,运行几天就会挂掉,通常都是 OOM,但是从程序看,并没有使用过多内存。

如果 YARN 上还会混合运行 mapreduce 和 tez 等应用,则会对 Spark streaming 存在资源竞争,造成性能不稳定,可以使用 label-based scheduling 对一些节点打标签,专门运行 Spark streaming。总体上说,spark streaming 在 YARN 上运行比较稳定。YARN 对 CPU 有隔离,使用的 cgroups。 如果是 OOM 挂掉,可能程序存在内存泄露,不知道你们用的什么版本,建议使用 jprofile 定位一下内存效率之处。

6、能否简单对比下 Storm 和 Spark 的优劣?如何技术选型?

Storm 是实时流式数据处理,面向行处理,单条延时比较低。Spark 是近实时流式处理,面向 vp 处理,吞吐量比较高。如果应用对实时性要求比较高建议试用 Storm, 否则大家可以考虑利用 Spark 的丰富的数据操作能力。