一概述

Structured Streaming 是 Spark Streaming 的进化版, 如果了解了 Spark 的各方面的进化过程, 有助于理解 Structured Streaming 的使命和作用

1 Spark 编程模型的进化过程

过程

编程模型 RDD 的优点和缺陷 2011
编程模型 DataFrame 的优点和缺陷 2013
编程模型 Dataset 的优点和缺陷 2015

RDD 的优点

面向对象的操作方式
可以处理任何类型的数据

RDD 的缺点

运行速度比较慢, 执行过程没有优化
API 比较僵硬, 对结构化数据的访问和操作没有优化

DataFrame 的优点

针对结构化数据高度优化, 可以通过列名访问和转换数据
增加 Catalyst 优化器, 执行过程是优化的, 避免了因为开发者的原因影响效率

DataFrame 的缺点

只能操作结构化数据
只有无类型的 API, 也就是只能针对列和 SQL 操作数据, API 依然僵硬

Dataset 的优点

结合了 RDD 和 DataFrame 的 API, 既可以操作结构化数据, 也可以操作非结构化数据
既有有类型的 API 也有无类型的 API, 灵活选择

2 Spark 的序列化的进化过程

Spark 中的序列化过程决定了数据如何存储, 是性能优化一个非常重要的着眼点, Spark 的进化并不只是针对编程模型提供的 API, 在大数据处理中, 也必须要考虑性能

过程

序列化和反序列化是什么
Spark 中什么地方用到序列化和反序列化
RDD 的序列化和反序列化如何实现
Dataset 的序列化和反序列化如何实现

1在java中序列化

public class JavaSerializable implements Serializable {
  NonSerializable ns = new NonSerializable();
}

public class NonSerializable {

}

public static void main(String[] args) throws IOException {
  // 序列化
  JavaSerializable serializable = new JavaSerializable();
  ObjectOutputStream objectOutputStream = new ObjectOutputStream(new FileOutputStream("/tmp/obj.ser"));
  // 这里会抛出一个 "java.io.NotSerializableException: cn.itcast.NonSerializable" 异常
  objectOutputStream.writeObject(serializable);
  objectOutputStream.flush();
  objectOutputStream.close();

  // 反序列化
  FileInputStream fileInputStream = new FileInputStream("/tmp/obj.ser");
  ObjectInputStream objectOutputStream = new ObjectInputStream(fileInputStream);
  JavaSerializable serializable1 = objectOutputStream.readObject();
}

序列化是什么

序列化的作用就是可以将对象的内容变成二进制, 存入文件中保存
反序列化指的是将保存下来的二进制对象数据恢复成对象

序列化对对象的要求

对象必须实现 Serializable 接口
对象中的所有属性必须都要可以被序列化, 如果出现无法被序列化的属性, 则序列化失败

限制

对象被序列化后, 生成的二进制文件中, 包含了很多环境信息, 如对象头, 对象中的属性字段等, 所以内容相对较大
因为数据量大, 所以序列化和反序列化的过程比较慢

序列化的应用场景

持久化对象数据
网络中不能传输 Java 对象, 只能将其序列化后传输二进制数据

2 在 `Spark` 中的序列化和反序列化的应用场景

1Task 分发 Task 是一个对象, 想在网络中传输对象就必须要先序列化

2RDD 缓存

RDD 中处理的是对象, 例如说字符串, Person 对象等
如果缓存 RDD 中的数据, 就需要缓存这些对象
对象是不能存在文件中的, 必须要将对象序列化后, 将二进制数据存入文件

3广播变量

广播变量会分发到不同的机器上, 这个过程中需要使用网络, 对象在网络中传输就必须先被序列化

4Shuffle 过程

Shuffle 过程是由 Reducer 从 Mapper 中拉取数据, 这里面涉及到两个需要序列化对象的原因

RDD 中的数据对象需要在 Mapper 端落盘缓存, 等待拉取
Mapper 和 Reducer 要传输数据对象

5Spark Streaming 的 Receiver

Spark Streaming 中获取数据的组件叫做 Receiver, 获取到的数据也是对象形式, 在获取到以后需要落盘暂存, 就需要对数据对象进行序列化

6算子引用外部对象

class Unserializable(i: Int)

rdd.map(i => new Unserializable(i))
   .collect
   .foreach(println)

在 Map 算子的函数中, 传入了一个 Unserializable 的对象
Map 算子的函数是会在整个集群中运行的, 那 Unserializable 对象就需要跟随 Map 算子的函数被传输到不同的节点上
如果 Unserializable 不能被序列化, 则会报错

3 rdd的序列化

RDD 的序列化

RDD 的序列化只能使用 Java 序列化器, 或者 Kryo 序列化器

为什么?

RDD 中存放的是数据对象, 要保留所有的数据就必须要对对象的元信息进行保存, 例如对象头之类的
保存一整个对象, 内存占用和效率会比较低一些

Kryo 是什么

Kryo 是 Spark 引入的一个外部的序列化工具, 可以增快 RDD 的运行速度
因为 Kryo 序列化后的对象更小, 序列化和反序列化的速度非常快
在 RDD 中使用 Kryo 的过程如下

val conf = new SparkConf()
  .setMaster("local[2]")
  .setAppName("KyroTest")

conf.set("spark.serializer", "org.apache.spark.serializer.KryoSerializer")
conf.registerKryoClasses(Array(classOf[Person]))

val sc = new SparkContext(conf)

rdd.map(arr => Person(arr(0), arr(1), arr(2)))

4 `DataFrame` 和 `Dataset` 中的序列化

RDD 中无法感知数据的组成, 无法感知数据结构, 只能以对象的形式处理数据

1`DataFrame` 和 `Dataset` 的特点

DataFrame 和 Dataset 是为结构化数据优化的
在 DataFrame 和 Dataset 中, 数据和数据的 Schema 是分开存储的

DataFrame 中没有数据对象这个概念, 所有的数据都以行的形式存在于 Row 对象中, Row 中记录了每行数据的结构, 包括列名, 类型等

Dataset 中上层可以提供有类型的 API, 用以操作数据, 但是在内部, 无论是什么类型的数据对象 Dataset 都使用一个叫做 InternalRow 的类型的对象存储数据

5 优化

1 元信息独立

RDD 不保存数据的元信息, 所以只能使用 Java Serializer 或者 Kyro Serializer 保存 整个对象
DataFrame 和 Dataset 中保存了数据的元信息, 所以可以把元信息独立出来分开保存

3 一个 DataFrame 或者一个 Dataset 中, 元信息只需要保存一份, 序列化的时候, 元信息不需要参与

4在反序列化 ( InternalRow → Object ) 时加入 Schema 信息即可

元信息不再参与序列化, 意味着数据存储量的减少, 和效率的增加

2 使用堆外内存

DataFrame 和 Dataset 不再序列化元信息, 所以内存使用大大减少. 同时新的序列化方式还将数据存入堆外内存中, 从而避免 GC 的开销.
堆外内存又叫做 Unsafe, 之所以叫不安全的, 因为不能使用 Java 的垃圾回收机制, 需要自己负责对象的创建和回收, 性能很好, 但是不建议普通开发者使用, 毕竟不安全

6总结

当需要将对象缓存下来的时候, 或者在网络中传输的时候, 要把对象转成二进制, 在使用的时候再将二进制转为对象, 这个过程叫做序列化和反序列化
在 Spark 中有很多场景需要存储对象, 或者在网络中传输对象
1. Task 分发的时候, 需要将任务序列化, 分发到不同的 Executor 中执行
2. 缓存 RDD 的时候, 需要保存 RDD 中的数据
3. 广播变量的时候, 需要将变量序列化, 在集群中广播
4. RDD 的 Shuffle 过程中 Map 和 Reducer 之间需要交换数据
5. 算子中如果引入了外部的变量, 这个外部的变量也需要被序列化
RDD 因为不保留数据的元信息, 所以必须要序列化整个对象, 常见的方式是 Java 的序列化器, 和 Kyro 序列化器
Dataset 和 DataFrame 中保留数据的元信息, 所以可以不再使用 Java 的序列化器和 Kyro 序列化器, 使用 Spark 特有的序列化协议, 生成 UnsafeInternalRow 用以保存数据, 这样不仅能减少数据量, 也能减少序列化和反序列化的开销, 其速度大概能达到 RDD 的序列化的 20 倍左右

3 Spark Streaming 和 Structured Streaming

1 `Spark Streaming` 时代

Spark Streaming 其实就是 RDD 的 API 的流式工具, 其本质还是 RDD, 存储和执行过程依然类似 RDD

2 `Structured Streaming` 时代

Structured Streaming 其实就是 Dataset 的 API 的流式工具, API 和 Dataset 保持高度一致

3 `Spark Streaming` 和 `Structured Streaming`

Structured Streaming 相比于 Spark Streaming 的进步就类似于 Dataset 相比于 RDD 的进步
另外还有一点, Structured Streaming 已经支持了连续流模型, 也就是类似于 Flink 那样的实时流, 而不是小批量, 但在使用的时候仍然有限制, 大部分情况还是应该采用小批量模式

在 2.2.0 以后 Structured Streaming 被标注为稳定版本, 意味着以后的 Spark 流式开发不应该在采用 Spark Streaming了

二小试牛刀

需求

编写一个流式计算的应用, 不断的接收外部系统的消息
对消息中的单词进行词频统计
统计全局的结果

Socket Server(Netcat nc) 等待 Structured Streaming 程序连接
Structured Streaming 程序启动, 连接 Socket Server, 等待 Socket Server 发送数据
Socket Server 发送数据, Structured Streaming 程序接收数据
Structured Streaming 程序接收到数据后处理数据
数据处理后, 生成对应的结果集, 在控制台打印

pom 文件

<properties>
       <scala.version>2.11.8</scala.version>
       <spark.version>2.2.0</spark.version>
       <slf4j.version>1.7.16</slf4j.version>
       <log4j.version>1.2.17</log4j.version>
   </properties>

   <dependencies>
       <dependency>
           <groupId>org.scala-lang</groupId>
           <artifactId>scala-library</artifactId>
           <version>${scala.version}</version>
       </dependency>
       <dependency>
           <groupId>org.apache.spark</groupId>
           <artifactId>spark-core_2.11</artifactId>
           <version>${spark.version}</version>
       </dependency>
       <dependency>
           <groupId>mysql</groupId>
           <artifactId>mysql-connector-java</artifactId>
           <version>5.1.47</version>
       </dependency>

       <dependency>
           <groupId>org.apache.spark</groupId>
           <artifactId>spark-streaming_2.11</artifactId>
           <version>2.2.0</version>
       </dependency>

       <dependency>
           <groupId>org.apache.spark</groupId>
           <artifactId>spark-sql-kafka-0-10_2.11</artifactId>
           <version>2.2.0</version>
       </dependency>
       <dependency>
           <groupId>org.apache.spark</groupId>
           <artifactId>spark-sql_2.11</artifactId>
           <version>${spark.version}</version>
       </dependency>
       <dependency>
           <groupId>org.apache.spark</groupId>
           <artifactId>spark-hive_2.11</artifactId>
           <version>${spark.version}</version>
       </dependency>
       <dependency>
           <groupId>org.apache.hadoop</groupId>
           <artifactId>hadoop-client</artifactId>
           <version>2.6.0</version>
       </dependency>

       <dependency>
           <groupId>org.slf4j</groupId>
           <artifactId>jcl-over-slf4j</artifactId>
           <version>${slf4j.version}</version>
       </dependency>
       <dependency>
           <groupId>org.slf4j</groupId>
           <artifactId>slf4j-api</artifactId>
           <version>${slf4j.version}</version>
       </dependency>
       <dependency>
           <groupId>org.slf4j</groupId>
           <artifactId>slf4j-log4j12</artifactId>
           <version>${slf4j.version}</version>
       </dependency>
       <dependency>
           <groupId>log4j</groupId>
           <artifactId>log4j</artifactId>
           <version>${log4j.version}</version>
       </dependency>
       <dependency>
           <groupId>junit</groupId>
           <artifactId>junit</artifactId>
           <version>4.10</version>
           <scope>provided</scope>
       </dependency>
   </dependencies>

   <build>
       <sourceDirectory>src/main/scala</sourceDirectory>
       <testSourceDirectory>src/test/scala</testSourceDirectory>
       <plugins>

           <plugin>
               <groupId>org.apache.maven.plugins</groupId>
               <artifactId>maven-compiler-plugin</artifactId>
               <version>3.0</version>
               <configuration>
                   <source>1.8</source>
                   <target>1.8</target>
                   <encoding>UTF-8</encoding>
               </configuration>
           </plugin>

           <plugin>
               <groupId>net.alchim31.maven</groupId>
               <artifactId>scala-maven-plugin</artifactId>
               <version>3.2.0</version>
               <executions>
                   <execution>
                       <goals>
                           <goal>compile</goal>
                           <goal>testCompile</goal>
                       </goals>
                       <configuration>
                           <args>
                               <arg>-dependencyfile</arg>
                               <arg>${project.build.directory}/.scala_dependencies</arg>
                           </args>
                       </configuration>
                   </execution>
               </executions>
           </plugin>
           <!--打包插件  因为默认的打包不包含maven的依赖-->
           <plugin>
               <groupId>org.apache.maven.plugins</groupId>
               <artifactId>maven-shade-plugin</artifactId>
               <version>3.1.1</version>
               <executions>
                   <execution>
                       <phase>package</phase>
                       <goals>
                           <goal>shade</goal>
                       </goals>
                       <configuration>
                           <filters>
                               <filter>
                                   <artifact>*:*</artifact>
                                   <excludes>
                                       <exclude>META-INF/*.SF</exclude>
                                       <exclude>META-INF/*.DSA</exclude>
                                       <exclude>META-INF/*.RSA</exclude>
                                   </excludes>
                               </filter>
                           </filters>
                           <transformers>
                               <transformer implementation="org.apache.maven.plugins.shade.resource.ManifestResourceTransformer">
                                   <mainClass></mainClass>
                               </transformer>
                           </transformers>
                       </configuration>
                   </execution>
               </executions>
           </plugin>
       </plugins>
   </build>


@Test
def rmDemo(): Unit = {
  //创建SparkSession
  val session: SparkSession = SparkSession.builder()
    .master("local[3]")
    .appName("demo")
    .getOrCreate()
    //调整 Log 级别, 避免过多的 Log 影响视线
  session.sparkContext.setLogLevel("error")
  import session.implicits._
  //读取外部数据源 转为dataset
  val source: Dataset[String] = session.readStream
    .format("socket")
    .option("host", "node03")
    .option("port", 9999)
    .load()
    .as[String] //默认 readStream 会返回 DataFrame, 但是词频统计更适合使用 Dataset 的有类型 API


  //统计词频
  val wc: Dataset[(String, Long)] = source.flatMap(_.split(" "))
    .map((_, 1))
    .groupByKey(_._1)
    .count()

  //输出结果
  wc.writeStream
    .outputMode(OutputMode.Complete()) //	统计全局结果, 而不是一个批次
    .format("console") //	将结果输出到控制台
    .start() //	开始运行流式应用
    .awaitTermination() // 阻塞主线程, 在子线程中不断获取数据
}

总结

Structured Streaming 中的编程步骤依然是先读, 后处理, 最后落地
Structured Streaming 中的编程模型依然是 DataFrame 和 Dataset
Structured Streaming 中依然是有外部数据源读写框架的, 叫做 readStream 和 writeStream
Structured Streaming 和 SparkSQL 几乎没有区别, 唯一的区别是, readStream 读出来的是流, writeStream 是将流输出, 而 SparkSQL 中的批处理使用 read 和 write

运行

在虚拟机中运行 nc -lk 9999 程序本地运行

从结果集中可以观察到以下内容

Structured Streaming 依然是小批量的流处理
Structured Streaming 的输出是类似 DataFrame 的, 也具有 Schema, 所以也是针对结构化数据进行优化的
从输出的时间特点上来看, 是一个批次先开始, 然后收集数据, 再进行展示, 这一点和 Spark Streaming 不太一样
Structured Streaming 的 API 和运行也是针对结构化数据进行优化过的

三Stuctured Streaming 的体系和结构

1 无限扩展的表格

Structured Streaming 是一个复杂的体系, 由很多组件组成, 这些组件之间也会进行交互, 如果无法站在整体视角去观察这些组件之间的关系, 也无法理解 Structured Streaming 的全局

1`Dataset` 和流式计算

可以理解为 Spark 中的 Dataset 有两种, 一种是处理静态批量数据的 Dataset, 一种是处理动态实时流的 Dataset, 这两种 Dataset 之间的区别如下

流式的 Dataset 使用 readStream 读取外部数据源创建, 使用 writeStream 写入外部存储
批式的 Dataset 使用 read 读取外部数据源创建, 使用 write 写入外部存储

2如何使用 `Dataset` 这个编程模型表示流式计算?

可以把流式的数据想象成一个不断增长, 无限无界的表
无论是否有界, 全都使用 Dataset 这一套 API
通过这样的做法, 就能完全保证流和批的处理使用完全相同的代码, 减少这两种处理方式的差异

3`WordCount` 的原理

整个计算过程大致上分为如下三个部分
1. Source, 读取数据源
2. Query, 在流式数据上的查询
3. Result, 结果集生成
整个的过程如下
1. 随着时间段的流动, 对外部数据进行批次的划分
2. 在逻辑上, 将缓存所有的数据, 生成一张无限扩展的表, 在这张表上进行查询
3. 根据要生成的结果类型, 来选择是否生成基于整个数据集的结果

4总结

Dataset 不仅可以表达流式数据的处理, 也可以表达批量数据的处理
Dataset 之所以可以表达流式数据的处理, 因为 Dataset 可以模拟一张无限扩展的表, 外部的数据会不断的流入到其中

2 体系结构

1 体系结构

在 Structured Streaming 中负责整体流程和执行的驱动引擎叫做 StreamExecution

StreamExecution 在流上进行基于 Dataset 的查询, 也就是说, Dataset 之所以能够在流上进行查询, 是因为 StreamExecution 的调度和管理

StreamExecution 如何工作?

StreamExecution 分为三个重要的部分

Source, 从外部数据源读取数据
LogicalPlan, 逻辑计划, 在流上的查询计划
Sink, 对接外部系统, 写入结果

2 `StreamExecution` 的执行顺序

1 根据进度标记, 从 Source 获取到一个由 DataFrame 表示的批次, 这个 DataFrame 表示数据的源头

val source = spark.readStream
  .format("socket")
  .option("host", "127.0.0.1")
  .option("port", 9999)
  .load()
  .as[String]

这一点非常类似 val df = spark.read.csv() 所生成的 DataFrame, 同样都是表示源头

2 根据源头 DataFrame 生成逻辑计划

val words = source.flatMap(_.split(" "))
  .map((_, 1))
  .groupByKey(_._1)
  .count()

述代码表示的就是数据的查询, 这一个步骤将这样的查询步骤生成为逻辑执行计划

3 优化逻辑计划最终生成物理计划

这一步其实就是使用 Catalyst 对执行计划进行优化, 经历基于规则的优化和基于成本模型的优化

4 执行物理计划将表示执行结果的 DataFrame / Dataset 交给 Sink

整个物理执行计划会针对每一个批次的数据进行处理, 处理后每一个批次都会生成一个表示结果的 Dataset

Sink 可以将每一个批次的结果 Dataset 落地到外部数据源

5 执行完毕后, 汇报 Source 这个批次已经处理结束, Source 提交并记录最新的进度

3 增量查询

上图中清晰的展示了最终的结果生成是全局的结果, 而不是一个批次的结果, 但是从 StreamExecution 中可以看到, 针对流的处理是按照一个批次一个批次来处理的

那么, 最终是如何生成全局的结果集呢?

状态记录

在 Structured Streaming 中有一个全局范围的高可用 StateStore, 这个时候针对增量的查询变为如下步骤

从 StateStore 中取出上次执行完成后的状态
把上次执行的结果加入本批次, 再进行计算, 得出全局结果
将当前批次的结果放入 StateStore 中, 留待下次使用

总结

StreamExecution 是整个 Structured Streaming 的核心, 负责在流上的查询
StreamExecution 中三个重要的组成部分, 分别是 Source 负责读取每个批量的数据, Sink 负责将结果写入外部数据源, Logical Plan 负责针对每个小批量生成执行计划
StreamExecution 中使用 StateStore 来进行状态的维护

四 Source

从 HDFS 中读取数据
从 Kafka 中读取数据

1 从 `HDFS` 中读取数据

读取小文件

1编写小文件

编写 Python 小程序, 在某个目录生成大量小文件
- Python 是解释型语言, 其程序可以直接使用命令运行无需编译, 所以适合编写快速使用的程序, 很多时候也使用 Python 代替 Shell
- 使用 Python 程序创建新的文件, 并且固定的生成一段 JSON 文本写入文件
- 在真实的环境中, 数据也是一样的不断产生并且被放入 HDFS 中, 但是在真实场景下, 可能是 Flume 把小文件不断上传到 HDFS 中, 也可能是 Sqoop 增量更新不断在某个目录中上传小文件
使用 Structured Streaming 汇总数据
- HDFS 中的数据是不断的产生的, 所以也是流式的数据
- 数据集是 JSON 格式, 要有解析 JSON 的能力
- 因为数据是重复的, 要对全局的流数据进行汇总和去重, 其实真实场景下的数据清洗大部分情况下也是要去重的
使用控制台展示数据
- 最终的数据结果以表的形式呈现
- 使用控制台展示数据意味着不需要在修改展示数据的代码, 将 Sink 部分的内容放在下一个大章节去说明
- 真实的工作中, 可能数据是要落地到 MySQL, HBase, HDFS 这样的存储系统中

步骤

Step 1: 编写 Python 脚本不断的产生数据
1. 使用 Python 创建字符串保存文件中要保存的数据
2. 创建文件并写入文件内容
3. 使用 Python 调用系统 HDFS 命令上传文件
Step 2: 编写 Structured Streaming 程序处理数据
1. 创建 SparkSession
2. 使用 SparkSession 的 readStream 读取数据源
3. 使用 Dataset 操作数据, 只需要去重
4. 使用 Dataset 的 writeStream 设置 Sink 将数据展示在控制台中
Step 3: 部署程序, 验证结果
1. 上传脚本到服务器中, 使用 python 命令运行脚本
2. 开启流计算应用, 读取 HDFS 中对应目录的数据
3. 查看运行结果

注意点

在读取 HDFS 的文件时, Source 不仅对接数据源, 也负责反序列化数据源中传过来的数据
- Source 可以从不同的数据源中读取数据, 如 Kafka, HDFS
- 数据源可能会传过来不同的数据格式, 如 JSON, Parquet
读取 HDFS 文件的这个 Source 叫做 FileStreamSource

从命名就可以看出来这个 Source 不仅支持 HDFS, 还支持本地文件读取, 亚马逊云, 阿里云等文件系统的读取, 例如: file://, s3://, oss://
基于流的 Dataset 操作和基于静态数据集的 Dataset 操作是一致的

整体流程

Python 程序产生数据到 HDFS 中
Structured Streaming 从 HDFS 中获取数据
Structured Streaming 处理数据
将数据展示在控制台

1编写Python文件

import os

for index in range(100):
    content = """
    {"name":"Michael"}
    {"name":"Andy", "age":30}
    {"name":"Justin", "age":19}
    """

    file_name = "/export/dataset/text{0}.json".format(index)

    with open(file_name, "w") as file:  
        file.write(content)

    os.system("/export/servers/hadoop/bin/hdfs dfs -mkdir -p /dataset/dataset/")
    os.system("/export/servers/hadoop/bin/hdfs dfs -put {0} /dataset/dataset/".format(file_name))

//读取hdfs上的文件
 @Test
 def hdfsSource(): Unit = {
   val spark = SparkSession.builder()
     .appName("hdfs_source")
     .master("local[3]")
     .getOrCreate()
   System.setProperty("path", "C:\\winutils")
   spark.sparkContext.setLogLevel("WARN")

   val userSchema = new StructType()
     .add("name", "string")
     .add("age", "integer")


   val source = spark
     .readStream //	指明读取的是一个流式的 Dataset
     .schema(userSchema) //指定读取到的数据的 Schema
     .json("hdfs://node03:8020/dataset/dataset") //指定目录位置, 以及数据格式

   val result = source.distinct() //去重

   result.writeStream
     .outputMode(OutputMode.Update())
     .format("console")
     .start()
     .awaitTermination()
 }

运行

运行 Python 脚本

# 进入 Python 文件被上传的位置
cd ~

# 创建放置生成文件的目录
mkdir -p /export/dataset

# 运行程序
python gen_files.py

运行spark 程序

1本地运行

2上传运行

1	spark-submit --class cn.itcast.structured.HDFSSource ./original-streaming-0.0.1.jar

总结:

Python 生成文件到 HDFS, 这一步在真实环境下, 可能是由 Flume 和 Sqoop 收集并上传至 HDFS
Structured Streaming 从 HDFS 中读取数据并处理
Structured Streaming 讲结果表展示在控制台

2 从 Kafka 中读取数据

kafka 回顾

Kafka 是一个 Pub / Sub 系统

1 一个pub 一个 sub

2多个 pub 一个sub

3 一个 pub 多个 sub

4 多个 pub 多个sub

Kafka 有一个非常重要的应用场景就是对接业务系统和数据系统, 作为一个数据管道, 其需要流通的数据量惊人, 所以 Kafka 如果要满足这种场景的话, 就一定具有以下两个特点

高吞吐量
高可靠性

topic 与partition

消息和事件经常是不同类型的, 例如用户注册是一种消息, 订单创建也是一种消息

Kafka 中使用 Topic 来组织不同类型的消息

Kafka 中的 Topic 要承受非常大的吞吐量, 所以 Topic 应该是可以分片的, 应该是分布式的

Kafka 的应用场景
- 一般的系统中, 业务系统会不止一个, 数据系统也会比较复杂
- 为了减少业务系统和数据系统之间的耦合, 要将其分开, 使用一个中间件来流转数据
- Kafka 因为其吞吐量超高, 所以适用于这种场景
Kafka 如何保证高吞吐量
- 因为消息会有很多种类, Kafka 中可以创建多个队列, 每一个队列就是一个 Topic, 可以理解为是一个主题, 存放相关的消息
- 因为 Topic 直接存放消息, 所以 Topic 必须要能够承受非常大的通量, 所以 Topic 是分布式的, 是可以分片的, 使用分布式的并行处理能力来解决高通量的问题

1 Kafka 和 Structured Streaming 整合的结构

Topic 的 Offset

Topic 是分区的, 每一个 Topic 的分区分布在不同的 Broker 上

每个分区都对应一系列的 Log 文件, 消息存在于 Log 中, 消息的 ID 就是这条消息在本分区的 Offset 偏移量

Offset 又称作为偏移量, 其实就是一个东西距离另外一个东西的距离

Kafka 中使用 Offset 命名消息, 而不是指定 ID 的原因是想表示永远自增, ID 是可以指定的, 但是 Offset只能是一个距离值, 它只会越来越大, 所以, 叫做 Offset 而不叫 ID 也是这个考虑, 消息只能追加到 Log 末尾, 只能增长不能减少

分析

Structured Streaming 中使用 Source 对接外部系统, 对接 Kafka 的 Source 叫做 KafkaSource
KafkaSource 中会使用 KafkaSourceRDD 来映射外部 Kafka 的 Topic, 两者的 Partition 一一对应

结论

Structured Streaming 会并行的从 Kafka 中获取数据

Structured Streaming 读取 Kafka 消息的三种方式

Earliest 从每个 Kafka 分区最开始处开始获取
Assign 手动指定每个 Kafka 分区中的 Offset
Latest 不再处理之前的消息, 只获取流计算启动后新产生的数据

总结

Kafka 中的消息存放在某个 Topic 的某个 Partition 中, 消息是不可变的, 只会在消息过期的时候从最早的消息开始删除, 消息的 ID 也叫做 Offset, 并且只能正增长
Structured Streaming 整合 Kafka 的时候, 会并行的通过 Offset 从所有 Topic 的 Partition 中获取数据
Structured Streaming 在从 Kafka 读取数据的时候, 可以选择从最早的地方开始读取, 也可以选择从任意位置读取, 也可以选择只读取最新的

1 使用生产者在 Kafka 的 Topic : streaming-test 中输入 JSON 数据 发送数据时必须把json转换为一行发送

{
  "devices": {
    "cameras": {
      "device_id": "awJo6rH",
      "last_event": {
        "has_sound": true,
        "has_motion": true,
        "has_person": true,
        "start_time": "2016-12-29T00:00:00.000Z",
        "end_time": "2016-12-29T18:42:00.000Z"
      }
    }
  }
}

JSON 数据本质上就是字符串, 只不过这个字符串是有结构的, 虽然有结构, 但是很难直接从字符串中取出某个值

而反序列化, 就是指把 JSON 数据转为对象, 或者转为 DataFrame, 可以直接使用某一个列或者某一个字段获取数据, 更加方便

而想要做到这件事, 必须要先根据数据格式, 编写 Schema 对象, 从而通过一些方式转为 DataFrame

 @Test
  def kafakaSource(): Unit = {
    // 1. 创建 SparkSession
    val spark = SparkSession.builder()
      .appName("hdfs_source")
      .master("local[6]")
      .getOrCreate()

    import spark.implicits._

    // 2. 读取 Kafka 数据

    val source: DataFrame = spark.readStream
      //设置为 Kafka 指定使用 KafkaSource 读取数据
      .format("kafka")
      //指定 Kafka 的 Server 地址
      .option("kafka.bootstrap.servers", "node01:9092,node02:9092,node03:9092")
      //要监听的 Topic, 可以传入多个, 传入多个 Topic 则监听多个 Topic, 也可以使用 topic-* 这样的通配符写法
      .option("subscribe", "streaming_test_1")
      //从什么位置开始获取数据, 可选值有 earliest, assign, latest
      .option("startingOffsets", "earliest") //从头开始读
      .load()

    // 3. 定义 JSON 中的类型
    /*{
 "devices": {
   "cameras": {
     "device_id": "awJo6rH",
     "last_event": {
       "has_sound": true,
       "has_motion": true,
       "has_person": true,
       "start_time": "2016-12-29T00:00:00.000Z",
       "end_time": "2016-12-29T18:42:00.000Z"
     }
   }
 }
}*/
    val eventType = new StructType() //   里层3
      .add("has_sound", BooleanType)
      .add("has_motion", BooleanType)
      .add("has_person", BooleanType)
      .add("start_time", DateType)
      .add("end_time", DateType)

    val cameraType = new StructType() //里层2
      .add("device_id", StringType)
      .add("last_event", eventType)

    val deviceType = new StructType()
      .add("cameras", cameraType) //里层1

    val schema = new StructType()
      .add("devices", deviceType) //最外层

    // 4. 解析 JSON
    // 需求: DataFrame(time, has_person)
    import org.apache.spark.sql.functions._

    val jsonOptions = Map("timestampFormat" -> "yyyy-MM-dd'T'HH:mm:ss.sss'Z'")

    val result = source.selectExpr("CAST(key AS STRING) as key", "CAST(value AS STRING) as value")
      .select(from_json('value, schema, jsonOptions).alias("parsed_value"))
      .selectExpr("parsed_value.devices.cameras.last_event.start_time", "parsed_value.devices.cameras.last_event.has_person")

    // 5. 打印数据
    result.writeStream
      .format("console")
      .outputMode(OutputMode.Append())
      .start()
      .awaitTermination()
  }

使用 Structured Streaming 来对接 Kafka 并反序列化 Kafka 中的 JSON 格式的消息, 是一个非常重要的技能

无论使用什么方式, 如果想反序列化 JSON 数据, 就必须要先追踪 JSON 数据的结构

注意点

1 业务系统如何把数据给 Kafka ?

可以主动或者被动的把数据交给 Kafka, 但是无论使用什么方式, 都在使用 Kafka 的 Client 类库来完成这件事, Kafka 的类库调用方式如下

1 2	Producer<String, String> producer = new KafkaProducer<String, String>(properties); producer.send(new ProducerRecord<String, String>("HelloWorld", msg));

其中发给 Kafka 的消息是 KV 类型的

2 使用 Structured Streaming 访问 Kafka 获取数据的时候, 需要什么东西呢?

存储当前处理过的 Kafka 的 Offset
对接多个 Kafka Topic 的时候, 要知道这条数据属于哪个 Topi

总结

Kafka 中收到的消息是 KV 类型的, 有 Key, 有 Value
Structured Streaming 对接 Kafka 的时候, 每一条 Kafka 消息不能只是 KV, 必须要有 Topic, Partition之类的信息

从 Kafka 获取的 DataFrame 格式

1	source.printSchema()

结果
root
 |-- key: binary (nullable = true)
 |-- value: binary (nullable = true)
 |-- topic: string (nullable = true)
 |-- partition: integer (nullable = true)
 |-- offset: long (nullable = true)
 |-- timestamp: timestamp (nullable = true)
 |-- timestampType: integer (nullable = true)

从 Kafka 中读取到的并不是直接是数据, 而是一个包含各种信息的表格, 其中每个字段的含义如下

Key	类型	解释
`key`	`binary`	`Kafka` 消息的 `Key`
`value`	`binary`	`Kafka` 消息的 `Value`
`topic`	`string`	本条消息所在的 `Topic`, 因为整合的时候一个 `Dataset` 可以对接多个 `Topic`, 所以有这样一个信息
`partition`	`integer`	消息的分区号
`offset`	`long`	消息在其分区的偏移量
`timestamp`	`timestamp`	消息进入 `Kafka` 的时间戳
`timestampType`	`integer`	时间戳类型

2 总结

一定要把 JSON 转为一行, 再使用 Producer 发送, 不然会出现获取多行的情况
使用 Structured Streaming 连接 Kafka 的时候, 需要配置如下三个参数
- kafka.bootstrap.servers : 指定 Kafka 的 Server 地址
- subscribe : 要监听的 Topic, 可以传入多个, 传入多个 Topic 则监听多个 Topic, 也可以使用 topic-* 这样的通配符写法
- startingOffsets : 从什么位置开始获取数据, 可选值有 earliest, assign, latest

从 Kafka 获取到的 DataFrame 的 Schema 如下

root
 |-- key: binary (nullable = true)
 |-- value: binary (nullable = true)
 |-- topic: string (nullable = true)
 |-- partition: integer (nullable = true)
 |-- offset: long (nullable = true)
 |-- timestamp: timestamp (nullable = true)
 |-- timestampType: integer (nullable = true)

3 JSON 解析

准备好 JSON 所在的列
- 问题
  
  由 Dataset 的结构可以知道 key 和 value 列的类型都是 binary 二进制, 所以要将其转为字符串, 才可进行 JSON 解析
- 解决方式
  
  source.selectExpr("CAST(key AS STRING) as key", "CAST(value AS STRING) as value")

编写 Schema 对照 JSON 的格式

Key 要对应 JSON 中的 Key
Value 的类型也要对应 JSON 中的 Value 类型

val eventType = new StructType()
  .add("has_sound", BooleanType, nullable = true)
  .add("has_motion", BooleanType, nullable = true)
  .add("has_person", BooleanType, nullable = true)
  .add("start_time", DateType, nullable = true)
  .add("end_time", DateType, nullable = true)

val camerasType = new StructType()
  .add("device_id", StringType, nullable = true)
  .add("last_event", eventType, nullable = true)

val devicesType = new StructType()
  .add("cameras", camerasType, nullable = true)

val schema = new StructType()
  .add("devices", devicesType, nullable = true)

因为 JSON 中包含 Date 类型的数据, 所以要指定时间格式化方式

1	val jsonOptions = Map("timestampFormat" -> "yyyy-MM-dd'T'HH:mm:ss.sss'Z'")

使用 from_json 这个 UDF 格式化 JSON

1	.select(from_json('value, schema, jsonOptions).alias("parsed_value"))

选择格式化过后的 JSON 中的字段

因为 JSON 被格式化过后, 已经变为了 StructType, 所以可以直接获取其中某些字段的值

1 2	.selectExpr("parsed_value.devices.cameras.last_event.has_person as has_person", "parsed_value.devices.cameras.last_event.start_time as start_time")

数据处理

统计各个时段有人的数据

1
2
3

.filter('has_person === true)
.groupBy('has_person, 'start_time)
.count()

将数据落地到控制台

result.writeStream
  .outputMode(OutputMode.Complete())
  .format("console")
  .start()
  .awaitTermination()

全部代码

import org.apache.spark.sql.SparkSession

val spark = SparkSession.builder()
  .master("local[6]")
  .appName("kafka integration")
  .getOrCreate()

import org.apache.spark.sql.streaming.OutputMode
import org.apache.spark.sql.types._

val source = spark
  .readStream
  .format("kafka")
  .option("kafka.bootstrap.servers", "node01:9092,node02:9092,node03:9092")
  .option("subscribe", "streaming-test")
  .option("startingOffsets", "earliest")
  .load()

val eventType = new StructType()
  .add("has_sound", BooleanType, nullable = true)
  .add("has_motion", BooleanType, nullable = true)
  .add("has_person", BooleanType, nullable = true)
  .add("start_time", DateType, nullable = true)
  .add("end_time", DateType, nullable = true)

val camerasType = new StructType()
  .add("device_id", StringType, nullable = true)
  .add("last_event", eventType, nullable = true)

val devicesType = new StructType()
  .add("cameras", camerasType, nullable = true)

val schema = new StructType()
  .add("devices", devicesType, nullable = true)

val jsonOptions = Map("timestampFormat" -> "yyyy-MM-dd'T'HH:mm:ss.sss'Z'")

import org.apache.spark.sql.functions._
import spark.implicits._

val result = source.selectExpr("CAST(key AS STRING) as key", "CAST(value AS STRING) as value")
    .select(from_json('value, schema, jsonOptions).alias("parsed_value"))
    .selectExpr("parsed_value.devices.cameras.last_event.has_person as has_person",
      "parsed_value.devices.cameras.last_event.start_time as start_time")
    .filter('has_person === true)
    .groupBy('has_person, 'start_time)
    .count()

result.writeStream
  .outputMode(OutputMode.Complete())
  .format("console")
  .start()
  .awaitTermination()

运行测试

进入服务器中, 启动 Kafka

启动 Kafka 的 Producer

1	bin/kafka-console-producer.sh --broker-list node01:9092,node02:9092,node03:9092 --topic streaming-test

启动 Spark shell 并拷贝代码进行测试
1
./bin/spark-shell --packages org.apache.spark:spark-sql-kafka-0-10_2.11:2.2.0
- 因为需要和 Kafka 整合, 所以在启动的时候需要加载和 Kafka 整合的包 spark-sql-kafka-0-10

五 sink

1 HDFS Sink

场景

Kafka 往往作为数据系统和业务系统之间的桥梁
数据系统一般由批量处理和流式处理两个部分组成
在 Kafka 作为整个数据平台入口的场景下, 需要使用 StructuredStreaming 接收 Kafka 的数据并放置于 HDFS上, 后续才可以进行批量处理

需求

从 Kafka 接收数据, 从给定的数据集中, 裁剪部分列, 落地于 HDFS

@Test
  def hdfsSink(): Unit = {
    System.setProperty("hadoop.home.dir", "C:\\winutil")

    // 1. 创建 SparkSession
    val spark = SparkSession.builder()
      .appName("hdfs_sink")
      .master("local[6]")
      .getOrCreate()

    import spark.implicits._

    // 2. 读取 Kafka 数据
    val source: Dataset[String] = spark.readStream
      .format("kafka")
      .option("kafka.bootstrap.servers", "node01:9092,node02:9092,node03:9092")
      .option("subscribe", "streaming_test_2")
      .option("startingOffsets", "earliest")
      .load()
      .selectExpr("CAST(value AS STRING) as value")
      .as[String]

    // 1::Toy Story (1995)::Animation|Children's|Comedy

    // 3. 处理 CSV, Dataset(String), Dataset(id, name, category)
    val result = source.map(item => {
      val arr = item.split("::")
      (arr(0).toInt, arr(1).toString, arr(2).toString)
    }).as[(Int, String, String)].toDF("id", "name", "category")

    // 4. 落地到 HDFS 中
    result.writeStream
      .format("parquet")
      .option("path", "dataset/streaming/moives/")
      .option("checkpointLocation", "checkpoint")
      .start()
      .awaitTermination()
  }

2 Kafka Sink

场景

有很多时候, ETL 过后的数据, 需要再次放入 Kafka
在 Kafka 后, 可能会有流式程序统一将数据落地到 HDFS 或者 HBase

需求

从 Kafka 中获取数据, 简单处理, 再次放入 Kafka

@Test
 def kafkaSink(): Unit = {
   System.setProperty("hadoop.home.dir", "C:\\winutil")

   // 1. 创建 SparkSession
   val spark = SparkSession.builder()
     .appName("hdfs_sink")
     .master("local[6]")
     .getOrCreate()

   import spark.implicits._

   // 2. 读取 Kafka 数据
   val source: Dataset[String] = spark.readStream
     .format("kafka")
     .option("kafka.bootstrap.servers", "node01:9092,node02:9092,node03:9092")
     .option("subscribe", "streaming_test_2")
     .option("startingOffsets", "earliest")
     .load()
     .selectExpr("CAST(value AS STRING) as value")
     .as[String]

   // 1::Toy Story (1995)::Animation|Children's|Comedy

   // 3. 处理 CSV, Dataset(String), Dataset(id, name, category)
   val result = source.map(item => {
     val arr = item.split("::")
     (arr(0).toInt, arr(1).toString, arr(2).toString)
   }).as[(Int, String, String)].toDF("id", "name", "category")

   // 4. 落地到 kafka 中
   result.writeStream
     .format("kafka")
     .outputMode(OutputMode.Append())
     .option("checkpointLocation", "checkpoint")
     .option("kafka.bootstrap.servers", "node01:9092,node02:9092,node03:9092")
     .option("topic", "streaming_test_3")
     .start()
     .awaitTermination()
 }

3 Foreach Writer

掌握 Foreach 模式理解如何扩展 Structured Streaming 的 Sink, 同时能够将数据落地到 MySQL

场景

1. 收集业务系统数据
2. 数据处理
3. 放入 OLTP 数据
4. 外部通过 ECharts 获取并处理数据
这个场景下, StructuredStreaming 就需要处理数据并放入 MySQL 或者 MongoDB, HBase 中以供 Web 程序可以获取数据, 图表的形式展示在前端

Foreach 模式::

在 Structured Streaming 中, 并未提供完整的 MySQL/JDBC 整合工具
不止 MySQL 和 JDBC, 可能会有其它的目标端需要写入
很多时候 Structured Streaming 需要对接一些第三方的系统, 例如阿里云的云存储, 亚马逊云的云存储等, 但是 Spark 无法对所有第三方都提供支持, 有时候需要自己编写
既然无法满足所有的整合需求, StructuredStreaming 提供了 Foreach, 可以拿到每一个批次的数据
通过 Foreach 拿到数据后, 可以通过自定义写入方式, 从而将数据落地到其它的系统

需求

从 Kafka 中获取数据, 处理后放入 MySQL

@Test
  def mysqlSink(): Unit = {
    System.setProperty("hadoop.home.dir", "C:\\winutil")

    // 1. 创建 SparkSession
    val spark = SparkSession.builder()
      .appName("hdfs_sink")
      .master("local[6]")
      .getOrCreate()

    import spark.implicits._

    // 2. 读取 Kafka 数据
    val source: Dataset[String] = spark.readStream
      .format("kafka")
      .option("kafka.bootstrap.servers", "node01:9092,node02:9092,node03:9092")
      .option("subscribe", "streaming_test_2")
      .option("startingOffsets", "earliest")
      .load()
      .selectExpr("CAST(value AS STRING) as value")
      .as[String]

    // 1::Toy Story (1995)::Animation|Children's|Comedy

    // 3. 处理 CSV, Dataset(String), Dataset(id, name, category)
    val result = source.map(item => {
      val arr = item.split("::")
      (arr(0).toInt, arr(1).toString, arr(2).toString)
    }).as[(Int, String, String)].toDF("id", "name", "category")

    // 4. 落地到 MySQL
    class MySQLWriter extends ForeachWriter[Row] {
      private val driver = "com.mysql.jdbc.Driver"
      private var connection: sql.Connection = _
      private val url = "jdbc::mysql://node01:3306/streaming-movies-result"
      private var statement: sql.Statement = _

      override def open(partitionId: Long, version: Long): Boolean = {
        Class.forName(driver)
        connection = DriverManager.getConnection(url)
        statement = connection.createStatement()
        true
      }

      override def process(value: Row): Unit = {
        statement.executeUpdate(s"insert into movies values(${value.get(0)}, ${value.get(1)}, ${value.get(2)})")
      }

      override def close(errorOrNull: Throwable): Unit = {
        connection.close()
      }
    }

    result.writeStream
      .foreach(new MySQLWriter)
      .start()
      .awaitTermination()
  }

4 自定义Sink

Foreach 倾向于一次处理一条数据, 如果想拿到 DataFrame 幂等的插入外部数据源, 则需要自定义 Sink

Spark 加载 Sink 流程分析

writeStream 方法中会创建一个 DataStreamWriter 对象

def writeStream: DataStreamWriter[T] = {
  if (!isStreaming) {
    logicalPlan.failAnalysis(
      "'writeStream' can be called only on streaming Dataset/DataFrame")
  }
  new DataStreamWriter[T](this)
}

在 DataStreamWriter 对象上通过 format 方法指定 Sink 的短名并记录下来
1
2
3
4
def format(source: String): DataStreamWriter[T] = {
this.source = source
this
}
3 .最终会通过 DataStreamWriter 对象上的 start 方法启动执行, 其中会通过短名创建 DataSource

val dataSource =
    DataSource(
      df.sparkSession,
      className = source, //	传入的 Sink 短名
      options = extraOptions.toMap,
      partitionColumns = normalizedParCols.getOrElse(Nil))

在创建 DataSource 的时候, 会通过一个复杂的流程创建出对应的 Source 和 Sink

1	lazy val providingClass: Class[_] = DataSource.lookupDataSource(className)

在这个复杂的创建流程中, 有一行最关键的代码, 就是通过 Java 的类加载器加载所有的 DataSourceRegister

1	val serviceLoader = ServiceLoader.load(classOf[DataSourceRegister], loader)

在 DataSourceRegister 中会创建对应的 Source 或者 Sink

trait DataSourceRegister {

  def shortName(): String       //	提供短名
}

trait StreamSourceProvider {
  def createSource(            //创建 Source
      sqlContext: SQLContext,
      metadataPath: String,
      schema: Option[StructType],
      providerName: String,
      parameters: Map[String, String]): Source
}

trait StreamSinkProvider {
  def createSink(              //创建 Sink
      sqlContext: SQLContext,
      parameters: Map[String, String],
      partitionColumns: Seq[String],
      outputMode: OutputMode): Sink
}

自定义 Sink 的方式

根据前面的流程说明, 有两点非常重要
- Spark 会自动加载所有 DataSourceRegister 的子类, 所以需要通过 DataSourceRegister 加载 Source 和 Sink
- Spark 提供了 StreamSinkProvider 用以创建 Sink, 提供必要的依赖
所以如果要创建自定义的 Sink, 需要做两件事
1. 创建一个注册器, 继承 DataSourceRegister 提供注册功能, 继承 StreamSinkProvider 获取创建 Sink 的必备依赖
2. 创建一个 Sink 子类

步骤

读取 Kafka 数据
简单处理数据
创建 Sink
创建 Sink 注册器
使用自定义 Sink

@Test
  def zdSink(): Unit = {
    import org.apache.spark.sql.SparkSession

    val spark = SparkSession.builder()
      .master("local[6]")
      .appName("kafka integration")
      .getOrCreate()

    import spark.implicits._

    val source = spark
      .readStream
      .format("kafka")
      .option("kafka.bootstrap.servers", "node01:9092,node02:9092,node03:9092")
      .option("subscribe", "streaming-bank")
      .option("startingOffsets", "earliest")
      .load()
      .selectExpr("CAST(value AS STRING)")
      .as[String]

    val result = source.map {
      item =>
        val arr = item.replace("\"", "").split(";")
        (arr(0).toInt, arr(1).toInt, arr(5).toInt)
    }
      .as[(Int, Int, Int)]
      .toDF("age", "job", "balance")

    class MySQLSink(options: Map[String, String], outputMode: OutputMode) extends Sink {

      override def addBatch(batchId: Long, data: DataFrame): Unit = {
        val userName = options.get("userName").orNull
        val password = options.get("password").orNull
        val table = options.get("table").orNull
        val jdbcUrl = options.get("jdbcUrl").orNull

        val properties = new Properties
        properties.setProperty("user", userName)
        properties.setProperty("password", password)

        data.write.mode(outputMode.toString).jdbc(jdbcUrl, table, properties)
      }
    }

    class MySQLStreamSinkProvider extends StreamSinkProvider with DataSourceRegister {

      override def createSink(sqlContext: SQLContext,
                              parameters: Map[String, String],
                              partitionColumns: Seq[String],
                              outputMode: OutputMode): Sink = {
        new MySQLSink(parameters, outputMode)
      }

      override def shortName(): String = "mysql"
    }

    result.writeStream
      .format("mysql")
      .option("username", "root")
      .option("password", "root")
      .option("table", "streaming-bank-result")
      .option("jdbcUrl", "jdbc:mysql://node01:3306/test")
      .start()
      .awaitTermination()
  }

5 Tigger

如何控制 StructuredStreaming 的处理时间

步骤

微批次处理
连续流处理

什么是微批次

并不是真正的流, 而是缓存一个批次周期的数据, 后处理这一批次的数据

通用流程

步骤

根据 Spark 提供的调试用的数据源 Rate 创建流式 DataFrame
- Rate 数据源会定期提供一个由两列 timestamp, value 组成的数据, value 是一个随机数
处理和聚合数据, 计算每个个位数和十位数各有多少条数据
- 对 value 求 log10 即可得出其位数
- 后按照位数进行分组, 最终就可以看到每个位数的数据有多少个

val spark = SparkSession.builder()
  .master("local[6]")
  .appName("socket_processor")
  .getOrCreate()

import org.apache.spark.sql.functions._
import spark.implicits._

spark.sparkContext.setLogLevel("ERROR")

val source = spark.readStream
  .format("rate")
  .load()

val result = source.select(log10('value) cast IntegerType as 'key, 'value)
    .groupBy('key)
    .agg(count('key) as 'count)
    .select('key, 'count)
    .where('key.isNotNull)
    .sort('key.asc)

默认方式划分批次

介绍

默认情况下的 Structured Streaming 程序会运行在微批次的模式下, 当一个批次结束后, 下一个批次会立即开始处理

步骤

指定落地到 Console 中, 不指定 Trigger

代码

result.writeStream
  .outputMode(OutputMode.Complete())
  .format("console")
  .start()
  .awaitTermination()

按照固定时间间隔划分批次

介绍

使用微批次处理数据, 使用用户指定的时间间隔启动批次, 如果间隔指定为 0, 则尽可能快的去处理, 一个批次紧接着一个批次

如果前一批数据提前完成, 待到批次间隔达成的时候再启动下一个批次
如果前一批数据延后完成, 下一个批次会在前面批次结束后立即启动
如果没有数据可用, 则不启动处理

步骤

通过 Trigger.ProcessingTime() 指定处理间隔

代码

result.writeStream
  .outputMode(OutputMode.Complete())
  .format("console")
  .trigger(Trigger.ProcessingTime("2 seconds"))
  .start()
  .awaitTermination()

一次性划分批次

介绍

只划分一个批次, 处理完成以后就停止 Spark 工作, 当需要启动一下 Spark 处理遗留任务的时候, 处理完就关闭集群的情况下, 这个划分方式非常实用

步骤

使用 Trigger.Once 一次性划分批次

代码

result.writeStream
  .outputMode(OutputMode.Complete())
  .format("console")
  .trigger(Trigger.Once())
  .start()
  .awaitTermination()

连续流处理

介绍

微批次会将收到的数据按照批次划分为不同的 DataFrame, 后执行 DataFrame, 所以其数据的处理延迟取决于每个 DataFrame 的处理速度, 最快也只能在一个 DataFrame 结束后立刻执行下一个, 最快可以达到 100ms 左右的端到端延迟
而连续流处理可以做到大约 1ms 的端到端数据处理延迟
连续流处理可以达到 at-least-once 的容错语义
从 Spark 2.3 版本开始支持连续流处理, 我们所采用的 2.2 版本还没有这个特性, 并且这个特性截止到 2.4 依然是实验性质, 不建议在生产环境中使用

操作

步骤

使用特殊的 Trigger 完成功能

代码

result.writeStream
  .outputMode(OutputMode.Complete())
  .format("console")
  .trigger(Trigger.Continuous("1 second"))
  .start()
  .awaitTermination()

限制

只支持 Map 类的有类型操作
只支持普通的的 SQL 类操作, 不支持聚合
Source 只支持 Kafka
Sink 只支持 Kafka, Console, Memory

6 从 Source 到 Sink 的流程

在每个 StreamExecution 的批次最开始, StreamExecution 会向 Source 询问当前 Source 的最新进度, 即最新的 offset
StreamExecution 将 Offset 放到 WAL 里
StreamExecution 从 Source 获取 start offset, end offset 区间内的数据
StreamExecution 触发计算逻辑 logicalPlan 的优化与编译
计算结果写出给 Sink
- 调用 Sink.addBatch(batchId: Long, data: DataFrame) 完成
- 此时才会由 Sink 的写入操作开始触发实际的数据获取和计算过程
在数据完整写出到 Sink 后, StreamExecution 通知 Source 批次 id 写入到 batchCommitLog, 当前批次结束

7 错误恢复和容错语义

1 端到端

Source 可能是 Kafka, HDFS
Sink 也可能是 Kafka, HDFS, MySQL 等存储服务
消息从 Source 取出, 经过 Structured Streaming 处理, 最后落地到 Sink 的过程, 叫做端到端

2 三种容错语义

1 at-most-once

在数据从 Source 到 Sink 的过程中, 出错了, Sink 可能没收到数据, 但是不会收到两次, 叫做 at-most-once
一般错误恢复的时候, 不重复计算, 则是 at-most-once

2 at-least-once

在数据从 Source 到 Sink 的过程中, 出错了, Sink 一定会收到数据, 但是可能收到两次, 叫做 at-least-once
一般错误恢复的时候, 重复计算可能完成也可能未完成的计算, 则是 at-least-once

3 exactly-once

在数据从 Source 到 Sink 的过程中, 虽然出错了, Sink 一定恰好收到应该收到的数据, 一条不重复也一条都不少, 即是 exactly-once
想做到 exactly-once 是非常困难的

3 Sink 的容错

1 故障恢复一般分为 Driver 的容错和 Task 的容错
- Driver 的容错指的是整个系统都挂掉了
- Task 的容错指的是一个任务没运行明白, 重新运行一次
2 因为 Spark 的 Executor 能够非常好的处理 Task 的容错, 所以我们主要讨论 Driver 的容错, 如果出错的时候
- 读取 WAL offsetlog 恢复出最新的 offsets
  
  当 StreamExecution 找到 Source 获取数据的时候, 会将数据的起始放在 WAL offsetlog 中, 当出错要恢复的时候, 就可以从中获取当前处理批次的数据起始, 例如 Kafka 的 Offset
- 读取 batchCommitLog 决定是否需要重做最近一个批次
  
  当 Sink 处理完批次的数据写入时, 会将当前的批次 ID 存入 batchCommitLog, 当出错的时候就可以从中取出进行到哪一个批次了, 和 WAL 对比即可得知当前批次是否处理完
- 如果有必要的话, 当前批次数据重做
  - 如果上次执行在 (5) 结束前即失效, 那么本次执行里 Sink 应该完整写出计算结果
  - 如果上次执行在 (5) 结束后才失效, 那么本次执行里 Sink 可以重新写出计算结果 (覆盖上次结果), 也可以跳过写出计算结果(因为上次执行已经完整写出过计算结果了)
- 这样即可保证每次执行的计算结果, 在 Sink 这个层面, 是 不重不丢 的, 即使中间发生过失效和恢复, 所以 Structured Streaming 可以做到 exactly-once

4 容错所需要的存储

存储
- offsetlog 和 batchCommitLog 关乎于错误恢复
- offsetlog 和 batchCommitLog 需要存储在可靠的空间里
- offsetlog 和 batchCommitLog 存储在 Checkpoint 中
- WAL 其实也存在于 Checkpoint 中

指定 Checkpoint

只有指定了 Checkpoint 路径的时候, 对应的容错功能才可以开启

aggDF
  .writeStream
  .outputMode("complete")
  .option("checkpointLocation", "path/to/HDFS/dir") //	指定 Checkpoint 的路径, 这个路径对应的目录必须是 HDFS 兼容的文件系统
  .format("memory")
  .start()

需要的外部支持

如果要做到 exactly-once, 只是 Structured Streaming 能做到还不行, 还需要 Source 和 Sink 系统的支持

Source 需要支持数据重放

当有必要的时候, Structured Streaming 需要根据 start 和 end offset 从 Source 系统中再次获取数据, 这叫做重放
Sink 需要支持幂等写入

如果需要重做整个批次的时候, Sink 要支持给定的 ID 写入数据, 这叫幂等写入, 一个 ID 对应一条数据进行写入, 如果前面已经写入, 则替换或者丢弃, 不能重复

所以 Structured Streaming 想要做到 exactly-once, 则也需要外部系统的支持, 如下

Source

`Sources`	是否可重放	原生内置支持	注解
`HDFS`	可以	已支持	包括但不限于 `Text`, `JSON`, `CSV`, `Parquet`, `ORC`
`Kafka`	可以	已支持	`Kafka 0.10.0+`
`RateStream`	可以	已支持	以一定速率产生数据
RDBMS	可以	待支持	预计后续很快会支持
Socket	不可以	已支持	主要用途是在技术会议和讲座上做 `Demo`

Sink

`Sinks`	是否幂等写入	原生内置支持	注解
`HDFS`	可以	支持	包括但不限于 `Text`, `JSON`, `CSV`, `Parquet`, `ORC`
`ForeachSink`	可以	支持	可定制度非常高的 `Sink`, 是否可以幂等取决于具体的实现
`RDBMS`	可以	待支持	预计后续很快会支持
`Kafka`	不可以	支持	`Kafka` 目前不支持幂等写入, 所以可能会有重复写入

六 . 有状态算子

1状态

1 无状态算子

2 有状态算子

有中间状态需要保存
增量查询

2 常规算子

Structured Streaming 的常规数据处理方式

需求

给定电影评分数据集 ratings.dat
筛选评分超过三分的电影
以追加模式展示数据, 以流的方式来一批数据处理一批数据, 最终每一批次展示为如下效果

+------+-------+
|Rating|MovieID|
+------+-------+
|     5|   1193|
|     4|   3408|
+------+-------+

读取文件的时候只能读取一个文件夹, 因为是流的操作, 流的场景是源源不断有新的文件读取

val source = spark.readStream
  .textFile("dataset/ratings")
  .map(line => {
    val columns = line.split("::")
    (columns(0).toInt, columns(1).toInt, columns(2).toInt, columns(3).toLong)
  })
  .toDF("UserID", "MovieID", "Rating", "Timestamp")

val result = source.select('Rating, 'MovieID)
    .where('Rating > 3)

针对静态数据集的很多转换算子, 都可以应用在流式的 Dataset 上, 例如 Map, FlatMap, Where, Select 等

3 分组算子

能够使用分组完成常见需求, 并了解如何扩展行

需求

给定电影数据集 movies.dat, 其中三列 MovieID, Title, Genres
统计每个分类下的电影数量

val source = spark.readStream
  .textFile("dataset/movies")
  .map(line => {
    val columns = line.split("::")
    (columns(0).toInt, columns(1).toString, columns(2).toString.split("\\|"))
  })
  .toDF("MovieID", "Title", "Genres")

val result = source.select(explode('Genres) as 'Genres)
    .groupBy('Genres)
    .agg(count('Genres) as 'Count)

result.writeStream
  .outputMode(OutputMode.Complete())
  .format("console")
  .queryName("genres_count")
  .start()
  .awaitTermination()

Structured Streaming 不仅支持 groupBy, 还支持 groupByKey

一 概述

1 Spark 编程模型的进化过程

2 Spark 的 序列化 的进化过程

1在java中序列化

2 在 Spark 中的序列化和反序列化的应用场景

3 rdd的序列化

4 DataFrame 和 Dataset 中的序列化

1DataFrame 和 Dataset 的特点

5 优化

1 元信息独立

2 使用堆外内存

6总结

3 Spark Streaming 和 Structured Streaming

1 Spark Streaming 时代

2 Structured Streaming 时代

3 Spark Streaming 和 Structured Streaming

二 小试牛刀

三Stuctured Streaming 的体系和结构

1 无限扩展的表格

1Dataset 和流式计算

2如何使用 Dataset 这个编程模型表示流式计算?

3WordCount 的原理

4总结

2 体系结构

1 体系结构

2 StreamExecution 的执行顺序

3 增量查询

四 Source

1 从 HDFS 中读取数据

2 从 Kafka 中读取数据

1 Kafka 和 Structured Streaming 整合的结构

2 总结

3 JSON 解析

五 sink

1 HDFS Sink

2 Kafka Sink

3 Foreach Writer

4 自定义Sink

5 Tigger

6 从 Source 到 Sink 的流程

7 错误恢复和容错语义

1 端到端

2 三种容错语义

3 Sink 的容错

4 容错所需要的存储

六 . 有状态算子

1状态

2 常规算子

3 分组算子

一概述

2 Spark 的序列化的进化过程

2 在 `Spark` 中的序列化和反序列化的应用场景

4 `DataFrame` 和 `Dataset` 中的序列化

1`DataFrame` 和 `Dataset` 的特点

1 `Spark Streaming` 时代

2 `Structured Streaming` 时代

3 `Spark Streaming` 和 `Structured Streaming`

二小试牛刀

1`Dataset` 和流式计算

2如何使用 `Dataset` 这个编程模型表示流式计算?

3`WordCount` 的原理

2 `StreamExecution` 的执行顺序

1 从 `HDFS` 中读取数据