Apache Flink是一个开源的分布式流处理框架,旨在提供高效、可扩展、容错的流式数据处理技术,支持实时流处理和批处理,并提供了Java、Scala、Python等语言的API。
import org.apache.flink.api.common.functions.FlatMapFunction;
import org.apache.flink.api.java.DataSet;
import org.apache.flink.api.java.ExecutionEnvironment;
import org.apache.flink.api.java.tuple.Tuple2;
import org.apache.flink.api.java.utils.ParameterTool;
import org.apache.flink.util.Collector;
public class WordCount {
public static void main(String[] args) throws Exception {
ParameterTool params = ParameterTool.fromArgs(args);
ExecutionEnvironment env = ExecutionEnvironment.getExecutionEnvironment();
env.getConfig().setGlobalJobParameters(params);
DataSet<String> text = env.readTextFile(params.get("input"));
DataSet<Tuple2<String, Integer>> wordCounts = text
.flatMap(new LineSplitter())
.groupBy(0)
.sum(1);
if (params.has("output")) {
wordCounts.writeAsCsv(params.get("output"), "\n", " ");
env.execute("WordCount Example");
} else {
System.out.println("Printing result to stdout. Use --output to specify output path.");
wordCounts.print();
}
}
}
class LineSplitter implements FlatMapFunction<String, Tuple2<String, Integer>> {
@Override
public void flatMap(String line, Collector<Tuple2<String, Integer>> out) throws Exception {
for (String word : line.split(" ")) {
out.collect(new Tuple2<>(word, 1));
}
}
}
依赖:
<dependency>
<groupId>org.apache.flinkgroupId>
<artifactId>flink-javaartifactId>
<version>1.13.5version>
<scope>providedscope>
dependency>
import org.apache.flink.api.common.functions.FlatMapFunction;
import org.apache.flink.api.java.tuple.Tuple2;
import org.apache.flink.streaming.api.datastream.DataStream;
import org.apache.flink.streaming.api.environment.StreamExecutionEnvironment;
import org.apache.flink.streaming.api.windowing.time.Time;
import org.apache.flink.util.Collector;
public class WindowWordCount {
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
DataStream<Tuple2<String, Integer>> dataStream = env
.socketTextStream("localhost", 9999)
.flatMap(new Splitter())
.keyBy(0)
.timeWindow(Time.seconds(5))
.sum(1);
dataStream.print();
env.execute("Window WordCount");
}
public static class Splitter implements FlatMapFunction<String, Tuple2<String, Integer>> {
@Override
public void flatMap(String sentence, Collector<Tuple2<String, Integer>> out) throws Exception {
for (String word : sentence.split(" ")) {
out.collect(new Tuple2<>(word, 1));
}
}
}
}
依赖:
<dependency>
<groupId>org.apache.flinkgroupId>
<artifactId>flink-streaming-java_2.12artifactId>
<version>1.13.5version>
<scope>providedscope>
dependency>
其中2.12是Scala版本。
官方文档:https://nightlies.apache.org/flink/flink-docs-stable/docs/learn-flink/overview/
很多数据都是流(stream)式的,分为有界流(bounded stream)和无界流(unbounded stream)。
在Flink中,应用由数据流(streaming dataflows)组成,数据流由用户定义的算子(operator)构成。数据流可抽象为有向图,以一个或多个源(source)算子开始、一个或多个汇(sink)算子结束。
Flink程序本质上就是并行和分布式的。在执行中,流有一个或多个分区(partition),每个算子有一个或多个子任务(subtask)。算子子任务是相互独立的,在不同线程中执行,可能位于不同机器上。
算子子任务的数量称为算子的并行度(parallelism)。同一个程序的不同算子可以有不同的并行度。
流可以在两个算子之间以一对一或重新分配的方式传输数据:
Flink的DataStream API支持任何可序列化的类型作为流元素,包括:
POJO类型需要满足以下条件:
transient
字段要么是公有且非final
的,要么具有公有getter和setter方法且遵循Java bean的命名规则例如下一节中的Person
类。
下面的示例输入人员信息的(有界)流,过滤出成年人,并打印出来。
AdultFilter.java
import org.apache.flink.streaming.api.datastream.DataStream;
import org.apache.flink.streaming.api.environment.StreamExecutionEnvironment;
public class AdultFilter {
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
DataStream<Person> flintstones = env.fromElements(
new Person("Fred", 35),
new Person("Wilma", 35),
new Person("Pebbles", 2)
);
DataStream<Person> adults = flintstones.filter(person -> person.age >= 18);
adults.print();
env.execute();
}
}
Person.java
public class Person {
public String name;
public Integer age;
public Person() {}
public Person(String name, Integer age) {
this.name = name;
this.age = age;
}
@Override
public String toString() {
return name + ": age " + age.toString();
}
}
该示例中的数据流结构如下:
s o u r c e → f l i n t s t o n e s f i l t e r → a d u l t s s i n k source \xrightarrow{flintstones} filter \xrightarrow{adults} sink sourceflintstonesfilteradultssink
其中source算子是fromElements()
,表示从指定的元素创建一个流,得到流flintstones
;之后经过filter()
算子过滤出成年人,得到流adults
;最后使用sink算子print()
打印元素。
程序运行结果:
1> Fred: age 35
2> Wilma: age 35
其中 “1>” 和 “2>” 表示子任务(即线程)id。
每个Flink应用需要一个执行环境,流处理应用需要StreamExecutionEnvironment
,即上面示例中的env
。
DataStream API调用构成一个数据流附加到执行环境,当调用env.execute()
时,数据流将被打包并发送到作业管理器(JobManager),JobManager将作业并行化切片并分发到任务管理器(TaskManager)执行。作业的每个并行切片将在一个任务槽(task slot)中执行。
注意:调用execute()
时应用才真正开始运行。
source算子(StreamExecutionEnvironment
类的方法)
fromElements()
:从指定元素创建流fromCollection()
:从指定集合创建流fromSequence()
:从整数区间创建流readTextFile()
:读取文本文件,每行作为一个元素socketTextStream()
:从套接字读取数据,使用指定的分隔符addSource()
:使用自定义source函数,见DataStream Connectors转换算子(DataStream
及其子类的方法)
map()
:元素一对一映射flatMap()
:元素一对n映射filter()
:按指定的条件过滤元素keyBy()
:按指定的key分组reduce()
:对已分组的流进行聚合union()
:合并多个流sink算子(DataStream
及其子类的方法)
print()
:将每个元素打印到标准输出流writeAsText()
:写入文本文件,每个元素占一行writeAsCsv()
:写入CSV文件writeToSocket()
:写入套接字addSink()
:使用自定义sink函数算子和流类型的转换关系如下图所示:
完整列表参考:DataStream API
https://github.com/apache/flink-training/tree/master/ride-cleansing
Flink的一种非常常见的应用场景是实现ETL (extract-transform-load)管道,即从一个或多个数据源获取数据,进行一些转换和/或信息补充,并将结果保存起来。本节将介绍如何使用Flink的DataStream API来实现这类应用。
算子map()
和flatMap()
是用于实现无状态转换的基本操作。
map()
算子接收一个MapFunction接口参数,对元素进行一对一转换,即每个元素对应恰好一个结果。由于MapFunction
是函数式接口,因此可以使用Lambda表达式。
例如,下面的代码将每个数字变为2倍:
DataStream<Long> doubled = env.fromSequence(1, 5).map(x -> 2 * x);
结果为[2, 4, 6, 8, 10]。
下面的代码将每个单词映射到其长度:
DataStream<Integer> wordLengths = env.fromElements("to be or not to be that is the question".split(" "))
.map(String::length);
结果为[2, 2, 2, 3, 2, 2, 4, 2, 3, 8]。
flatMap()
算子接收一个FlatMapFunction接口参数,对元素进行一对n转换,即每个元素映射到零个或多个结果。
例如,下面的代码将文本行分割为单词:
DataStream<String> words = env.fromElements("to be or not to be", "that is the question")
.flatMap(new LineSplitter());
class LineSplitter implements FlatMapFunction<String, String> {
@Override
public void flatMap(String line, Collector<String> out) throws Exception {
for (String word : line.split(" ")) {
out.collect(word);
}
}
}
结果为[“to”, “be”, “or”, “not”, “to”, “be”, “that”, “is”, “the”, “question”]。
下面的代码只保留奇数:
DataStream<Long> oddNumbers = env.fromSequence(1, 5).flatMap(new OddNumber());
class OddNumber implements FlatMapFunction<Long, Long> {
@Override
public void flatMap(Long x, Collector<Long> out) throws Exception {
if (x % 2 == 1) {
out.collect(x);
}
}
}
结果为[1, 3, 5](也可以用filter()
实现)。
将一个流按照某个字段分组通常是十分有用的,类似于SQL的GROUP BY
语句。在Flink中可以用keyBy()
算子实现。
keyBy()
算子接收一个KeySelector接口参数(元组也可以使用字段索引,POJO也可以使用字段名),返回KeyedStream
。接口KeySelector
的抽象方法从一个元素提取出用于分组的key。
例如,下面的代码将人员信息按年龄分组:
KeyedStream<Person, Integer> keyedByAge = people.keyBy(p -> p.age);
KeySelector
提取的key不限于元素的字段,可以按任何方式计算得到key,只要key是确定的,并且实现了hashCode()
和equals()
方法即可。例如,随机数、数组和枚举不能作为key,但元组和POJO可以作为复合key。
keyBy()
操作会将流重新分区,这个开销是很大的,因为涉及网络通信、序列化和反序列化,如3.1.1节图所示。
分组后的流可以对每个组进行聚合操作。例如,下面的代码输出每组相同姓名的人员中年龄的最大值:
DataStream<Person> people = env.fromElements(
new Person("Alice", 24),
new Person("Alice", 18),
new Person("Bob", 30),
new Person("Bob", 25),
new Person("Alice", 32));
people.keyBy(p -> p.name).max("age").print();
注意,不是每组只输出一个最大值,而是每个元素对应其所属分组中目前遇到的最大值,因此输出结果为
Alice: age 24
Alice: age 24
Bob: age 30
Bob: age 30
Alice: age 32
下面的代码对单词计数:
DataStream<Tuple2<String, Integer>> wordCounts = env.fromElements("to be or not to be".split(" "))
.map(w -> Tuple2.of(w, 1))
.returns(Types.TUPLE(Types.STRING, Types.INT));
wordCounts.keyBy(t -> t.f0).sum(1).print();
输出结果为
(to,1)
(be,1)
(or,1)
(not,1)
(to,2)
(be,2)
注:由于类型擦除,Tuple2.of()
无法提供关于其字段类型的信息,如果不写returns()
会报错 “The generic type parameters of ‘Tuple2’ are missing”,见Java Lambda Expressions。解决方法是实现MapFunction
接口、使用匿名类或者returns()
等能够显式指定类型的方式。
如上面两个例子所示,KeyedStream
的聚合操作是对每个分组分别进行聚合,并且是“滚动”聚合。例如sum()
计算的是前缀和(部分和)而不是总和,即相当于Scala的Array.scan()
而不是Array.reduce()
、Python的itertools.accumulate()
而不是functools.reduce()
,因为无限流不存在“总和”。
为了记录每个分组“当前遇到的最大值”或者“当前部分和”,Flink内部使用了状态(见3.3.3节)。当应用涉及状态时,就应该考虑状态的大小。如果key的取值空间是无限的,则Flink的状态需要的存储空间也同样是无限的。
在流处理场景中,考虑有限窗口的聚合往往比整个流聚合更有意义(见3.4节)。
KeyedStreams
提供了更通用的聚合算子reduce()
,以及求和、最大值和最小值的三个特例:
reduce()
:接收一个ReduceFunction接口参数,对每个相同key的分组内的元素进行“滚动”聚合操作,即a[0], a[1], a[2], ... -> a[0], f(a[0], a[1]), f(f(a[0], a[1]), a[2]), ...
sum()
:计算前缀和max()
:计算当前遇到的最大值min()
:计算当前遇到的最小值本节将介绍如何使用Flink的API来管理状态(state)。
Flink的filter()
、map()
、flatMap()
等算子使用的函数接口FilterFunction
、MapFunction
、FlatMapFunction
都提供了一个 “rich” 变体(实现了RichFunction接口),增加了以下方法:
open()
:算子初始化时调用一次close()
:关闭时调用一次getRuntimeContext()
:返回用于创建和访问状态的上下文对象下面的例子对事件流去重,每个key只保留第一个事件。在该应用中,使用一个名为Deduplicator
的RichFlatMapFunction
来实现去重操作。
class Event {
public String key;
public long timestamp;
public Event(String key, long timestamp) {
this.key = key;
this.timestamp = timestamp;
}
@Override
public String toString() {
return key + '@' + timestamp;
}
}
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
env.setParallelism(1);
DataStream<Event> events = env.fromElements(
new Event("a", 1), new Event("b", 2), new Event("a", 2),
new Event("c", 3), new Event("b", 3), new Event("d", 4)
);
events.keyBy(e -> e.key)
.flatMap(new Deduplicator())
.print();
env.execute();
}
程序将输出
a@1
b@2
c@3
d@4
为了实现这一功能,Deduplicator
需要记住出现过的key,这里使用状态来记录。
状态相当于算子的局部变量,可用于存储数据。对于KeyedStream
,Flink将维护一个键值对存储,即每个key分别存储一份数据。最简单的一种状态是ValueState,即每个key分别存储单个对象,在这个例子中是Boolean
类型。
Deduplicator
实现如下:
class Deduplicator extends RichFlatMapFunction<Event, Event> {
private ValueState<Boolean> keyHasBeenSeen;
@Override
public void open(Configuration conf) throws Exception {
ValueStateDescriptor<Boolean> descriptor = new ValueStateDescriptor<>("keyHasBeenSeen", Types.BOOLEAN);
keyHasBeenSeen = getRuntimeContext().getState(descriptor);
}
@Override
public void flatMap(Event event, Collector<Event> out) throws Exception {
if (keyHasBeenSeen.value() == null) {
out.collect(event);
keyHasBeenSeen.update(true);
}
}
}
Deduplicator
有两个方法:open()
使用ValueStateDescriptor
通过名字 “keyHasBeenSeen” 获取状态;flatMap()
方法获取状态的值,如果为null
则表示当前key未出现过,因此输出当前事件并将状态更新为true
。
注意:在读访问和更新状态keyHasBeenSeen
时,key并未显式出现过。但Flink运行时调用flatMap()
方法时,当前被处理的事件的key是已知的,Flink根据这个key来确定keyHasBeenSeen.value()
访问键值对存储中的哪个条目。
部署在分布式集群时,将会有很多Deduplicator
的实例,每个实例负责整个key空间中的一个子集。因此,ValueState
代表的不是一个布尔变量,而是一个分布式、分片的键值对存储。
上面的例子有一个潜在的问题:当key空间无界时状态需要的存储空间也是无界的。因此清除不再需要的key的状态是很有必要的,可以通过调用状态对象的clear()
方法来实现。
可以在ProcessFunction
中通过定时器清除状态(见3.5节),也可以通过状态的生存时间 (time-to-live, TTL)配置自动清除时间。
有时可能需要通过引入阈值、规则或其他参数来动态调整转换功能。Flink支持这种需求的模式称为连接流(connected stream),其中一个算子有两个输入流,如下图所示。
连接流也可以用于实现流的关联(join)。
在该示例中,使用控制流control
来指定要从单词流streamOfWords
中过滤掉的单词。使用connect()
算子连接两个流,之后在flatMap()
算子中使用RichCoFlatMapFunction
来实现这一功能。
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
DataStream<String> control = env.fromElements("DROP", "IGNORE").keyBy(w -> w);
DataStream<String> streamOfWords = env.fromElements("Apache", "DROP", "Flink", "IGNORE").keyBy(w -> w);
control.connect(streamOfWords)
.flatMap(new ControlFunction())
.print();
env.execute();
}
两个流只有按相同的方式分组时才能连接。
ControlFunction
在状态中存储一个布尔值,被两个流共享:
class ControlFunction extends RichCoFlatMapFunction<String, String, String> {
private ValueState<Boolean> blocked;
@Override
public void open(Configuration parameters) throws Exception {
blocked = getRuntimeContext().getState(new ValueStateDescriptor<>("blocked", Boolean.class));
}
@Override
public void flatMap1(String blockWord, Collector<String> out) throws Exception {
blocked.update(true);
}
@Override
public void flatMap2(String word, Collector<String> out) throws Exception {
if (blocked.value() == null) {
out.collect(word);
}
}
}
流control
和streamOfWords
中的单词将分别进入flatMap1()
和flatMap2()
。状态blocked
用于记录当前key(单词)是否在control
中出现过,这个状态是与key关联的,并且被两个流共享,这也是为什么两个流必须有相同的key空间。
理论上程序只会输出 “Apache” 和 “Flink” 两个单词,但实际上多次运行会发现 “DROP” 和 “IGNORE” 也有可能被输出。
注意:对于同一个key,flatMap1()
和flatMap2()
的调用顺序是无法控制的,因为这两个流是相互竞争的。对于需要保证时间和/或顺序的场景,必须将元素缓存在状态中,直到它们能够被处理。
https://github.com/apache/flink-training/tree/master/rides-and-fares
Flink支持三种时间语义:
注:Flink时间戳的单位是毫秒。
使用事件时间可以得到可复现的结果,无论什么时候计算结果都一样。实时应用可能会使用处理时间,但多次运行会得到不同的结果,使得重新分析历史数据或者测试新代码比较困难。
要使用事件时间,需要用到水印。
考虑一个简单的例子。有一个乱序到达的事件流,如下所示,其中数字表示事件发生时间:
... 23 19 22 24 21 14 17 13 12 15 9 11 7 2 4 ->
假设要对事件流排序:在没个事件到达时就处理事件,输出按时间戳排序的流。但是发现存在一些困难:
(1)第一个到达的事件的时间戳是4,但不能立即将其输出,因为更早的事件可能还未到达(站在上帝视角可以知道必须等待时间戳为2的事件到达)。因此,缓冲和延迟是必要的。
(2)如果处理不当,可能会永远等待。排序程序首先会看到时间戳为4的事件,之后看到时间戳为2的事件。接下来是否会有时间戳小于2的事件到达?可能会,也可能不会(再次站在上帝视角,可以知道永远不会看到时间戳1)。因此,必须在某个时刻决定将2作为第一个输出结果。
(3)接下来需要一种策略,来定义对于给定时间戳的事件,何时停止等待更早事件的到达——这正是水印的作用:
t时刻的水印 Watermark(t)表示t时刻之前的事件(很可能)已经全部到达。在Watermark(t)之后到达且时间戳≤t的事件称为迟到事件。例如,时间戳9:00的水印在10:00到达,则10:00之后到达、时间戳在9:00及以前(即延迟1个小时及以上)的事件是迟到的。
Flink处理事件时间依赖于水印生成器,将带有时间戳的特殊元素(水印)插入到流中。当t时刻的水印到达时停止等待时间戳≤t的事件。 在这个例子中,当时间戳≥2的水印到达时,将2输出作为已经排好序的流。
(4)如何决定水印的生成策略?每个事件都会延迟一段时间后到达,这些延迟有大有小。一种简单的方法是假定这些延迟有一个上界,Flink将这种策略称为“有界乱序”(bounded-out-of-orderness)水印。
流应用开发者需要在延迟和完整性之间权衡:如果将水印的边界时间设置得较小,则等待时间较短,但丢失的数据较多;反之得到的数据更完整,但延迟较大。
使用水印最简单的方式是用WatermarkStrategy。例如:
DataStream<Event> events = ...;
WatermarkStrategy<Event> strategy = WatermarkStrategy
.<Event>forBoundedOutOfOrderness(Duration.ofSeconds(20))
.withTimestampAssigner((event, timestamp) -> event.timestamp);
DataStream<Event> withTimestampsAndWatermarks = events.assignTimestampsAndWatermarks(strategy);
其中forBoundedOutOfOrderness()
表示使用“有界乱序”策略,即周期性地插入当前最大事件时间-20秒的水印;withTimestampAssigner()
指定如何提取事件时间。
注:“t时刻的水印到达时停止等待t时刻及以前的事件” ⇔ “如果事件时间≤当前水印时间则丢弃该事件”,这一逻辑需要自己实现,见3.5.1.2节PseudoWindow.processElement()
方法。
Flink具有非常富有表现力的窗口(window)语义。本节将介绍:
在处理无界流时,很自然地希望对流的有界子集进行聚合分析,以便回答以下问题:
用Flink进行窗口分析主要依赖两个抽象:窗口分配器(window assigner)和窗口函数(window function)。窗口分配器用于将事件分配到窗口,窗口函数用于处理窗口内的事件。
此外还有触发器(trigger)和驱逐器(evictor)。触发器确定何时调用窗口函数,驱逐器删除窗口中的元素。
在已分组的流(KeyedStream
)上使用窗口的一般形式如下:
stream
.keyBy()
.window()
.reduce|aggregate|process();
也可以在未分组的流上使用窗口,但处理不能并行化:
stream
.windowAll()
.reduce|aggregate|process();
Flink有一些内置的窗口分配器,如下图所示。
下面是如何使用这些窗口分配器的示例:
TumblingEventTimeWindows.of(Time.minutes(1))
SlidingEventTimeWindows.of(Time.minutes(1), Time.seconds(10))
EventTimeSessionWindows.withGap(Time.minutes(30))
基于时间的窗口分配器(包括会话窗口)既可以使用事件时间,也可以使用处理时间(将类名中的EventTime
改为ProcessingTime
即可)。这两种时间窗口各有利弊,使用处理时间窗口具有以下限制:
但优势是延迟较低。
注:使用基于事件时间的窗口必须先调用assignTimestampsAndWatermarks()
指定时间戳。
基于计数的窗口只有在到达一个完整批次时才会触发计算,无法设置超时,除非自定义触发器。
全局窗口分配器将(具有相同key的)所有事件分配到同一个全局窗口,这只有在自定义触发器时才有用。很多情况下使用ProcessFunction
更好(见3.5.1节)。
有三种方式处理窗口内的数据:
Iterable
,可以全量计算;ReduceFunction
或AggregateFunction
,每个事件被分配到窗口时都会调用一次;ReduceFunction
或AggregateFunction
预聚合的结果在触发窗口时提供给ProcessWindowFunction
做全量计算。下面展示第一种和第三种方式的示例,用于计算传感器在1分钟窗口内的峰值,产生一个包含元组(key, 窗口结束时间戳, 最大值)的流。
批处理示例:
class SensorReading {
long timestamp;
String key;
int value;
public SensorReading(long timestamp, String key, int value) {
this.timestamp = timestamp;
this.key = key;
this.value = value;
}
}
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
env.setParallelism(1);
DataStream<SensorReading> input = env.fromElements(
new SensorReading(0, "a", 10),
new SensorReading(0, "b", 2),
new SensorReading(30000, "b", 5),
new SensorReading(120000, "a", 15),
new SensorReading(90000, "b", 8)
);
WatermarkStrategy<SensorReading> strategy = WatermarkStrategy
.<SensorReading>noWatermarks()
.withTimestampAssigner((event, timestamp) -> event.timestamp);
input.assignTimestampsAndWatermarks(strategy)
.keyBy(x -> x.key)
.window(TumblingEventTimeWindows.of(Time.minutes(1)))
.process(new MyWastefulMax())
.print();
env.execute();
}
class MyWastefulMax extends ProcessWindowFunction<SensorReading, Tuple3<String, Long, Integer>, String, TimeWindow> {
@Override
public void process(
String key,
Context context,
Iterable<SensorReading> events,
Collector<Tuple3<String, Long, Integer>> out) {
int max = 0;
for (SensorReading event : events) {
max = Math.max(event.value, max);
}
out.collect(Tuple3.of(key, context.window().getEnd(), max));
}
}
输出结果:
(a,60000,10)
(b,60000,5)
(b,120000,8)
(a,180000,15)
注意:
ProcessWindowFunction
有一个Context参数,其windowState()
和globalState()
分别返回可以存储当前key当前窗口和当前key全局信息的状态对象。增量聚合、批处理结合示例:
public static void main(String[] args) throws Exception {
// ...
input.assignTimestampsAndWatermarks(strategy)
.keyBy(x -> x.key)
.window(TumblingEventTimeWindows.of(Time.minutes(1)))
.reduce(new MyReducingMax(), new MyWindowFunction())
.print();
env.execute();
}
class MyReducingMax implements ReduceFunction<SensorReading> {
@Override
public SensorReading reduce(SensorReading r1, SensorReading r2) {
return r1.value > r2.value ? r1 : r2;
}
}
class MyWindowFunction extends ProcessWindowFunction<SensorReading, Tuple3<String, Long, Integer>, String, TimeWindow> {
@Override
public void process(
String key,
Context context,
Iterable<SensorReading> maxReading,
Collector<Tuple3<String, Long, Integer>> out) {
SensorReading max = maxReading.iterator().next();
out.collect(Tuple3.of(key, context.window().getEnd(), max.value));
}
}
注意Iterable
只包含一个元素,即MyReducingMax
预先计算的最大值。
默认情况下,使用事件时间窗口时,迟到的事件会被丢弃,但窗口API提供了两个选项来控制这些事件。
可以使用旁路输出(side output)机制(见3.5.2节)将这些事件收集到另一个输出流。例如:
OutputTag<Event> lateTag = new OutputTag<Event>("late"){};
SingleOutputStreamOperator<Event> result = stream
.keyBy(...)
.window(...)
.sideOutputLateData(lateTag)
.process(...);
DataStream<Event> lateStream = result.getSideOutput(lateTag);
也可以执行允许的延迟间隔,延迟在这个间隔内的事件仍然会被分配到对应的窗口,每个迟到事件都会导致窗口函数被再次调用。例如:
stream
.keyBy(...)
.window(...)
.allowedLateness(Time.seconds(10))
.process(...);
https://github.com/apache/flink-training/tree/master/hourly-tips
ProcessFunction将事件处理与计时器和状态相结合,使其成为流处理应用的强大构建模块。这是使用Flink创建事件驱动应用的基础。
注:
RichFunction
接口,因此可以访问状态。ProcessFunction
要实现两个主要的回调函数:processElement()
和onTimer()
。processElement()
对于每个输入事件都会被调用,onTimer()
在计时器触发时被调用。计时器可以基于事件时间,也可以基于处理时间。两个回调函数都提供了可以获取TimerService的上下文对象,TimerService
用于管理计时器。利用ProcessFunction
可以对流执行任何操作,实现任何想要的功能。例如,使用ProcessFunction
+ 状态 + 计时器可以实现窗口功能。
在上一节的练习Hourly Tips中,使用滚动窗口来计算每小时内每个司机的小费总和:
SingleOutputStreamOperator<Tuple3<Long, Long, Float>> hourlyTips = fares
.keyBy(fare -> fare.driverId)
.window(TumblingEventTimeWindows.of(Time.hours(1)))
.process(new AddTips());
下面使用KeyedProcessFunction
实现同样的操作:
SingleOutputStreamOperator<Tuple3<Long, Long, Float>> hourlyTips = fares
.keyBy(fare -> fare.driverId)
.window(TumblingEventTimeWindows.of(Time.hours(1)))
.process(new PseudoWindow(Time.hours(1)));
class PseudoWindow extends KeyedProcessFunction<Long, TaxiFare, Tuple3<Long, Long, Float>> {
private final long durationMillis;
private transient MapState<Long, Float> sumOfTips;
public PseudoWindow(Time duration) {
this.durationMillis = duration.toMilliseconds();
}
@Override
public void open(Configuration conf) {
sumOfTips = getRuntimeContext().getMapState(new MapStateDescriptor<>("sumOfTips", Long.class, Float.class));
}
@Override
public void processElement(
TaxiFare fare,
Context ctx,
Collector<Tuple3<Long, Long, Float>> out) throws Exception {
long eventTime = fare.getEventTime();
TimerService timerService = ctx.timerService();
if (eventTime <= timerService.currentWatermark()) {
// This event is late; its window has already been triggered.
return;
}
// Round up eventTime to the end of the window containing this event.
long endOfWindow = eventTime - (eventTime % durationMillis) + durationMillis;
// Schedule a callback for when the window has been completed.
timerService.registerEventTimeTimer(endOfWindow);
// Add this fare's tip to the running total for that window.
Float sum = sumOfTips.get(endOfWindow);
if (sum == null) {
sum = 0.0f;
}
sum += fare.tip;
sumOfTips.put(endOfWindow, sum);
}
@Override
public void onTimer(
long timestamp,
OnTimerContext ctx,
Collector<Tuple3<Long, Long, Float>> out) throws Exception {
long driverId = ctx.getCurrentKey();
// Look up the result for the hour that just ended.
Float sumOfTips = this.sumOfTips.get(timestamp);
Tuple3<Long, Long, Float> result = Tuple3.of(timestamp, driverId, sumOfTips);
out.collect(result);
this.sumOfTips.remove(timestamp);
}
}
注意:
open()
方法中使用MapState
记录每个“窗口”内的小费总和,其中key是窗口结束时间,value是小费总和。另外,在KeyedProcessFunction
中,状态是与外层key关联的,在这里是司机id。processElement()
方法将迟到事件丢弃(也可以使用旁路输出,见下一节)。对于未迟到事件,首先计算其所属的“窗口”,使用窗口结束时间作为key查询状态,累加小费,并更新状态;之后调用timerService.registerEventTimeTimer()
注册基于事件时间的计时器回调函数,在给定时间戳的水印到达(即当前“窗口”结束)时将被调用。onTimer()
方法使用timestamp
(即窗口结束时间)作为key查询状态,得到当前“窗口”的小费总和,并与外层key(司机id)一起作为输出。每条记录与MapState
的对应关系如下图所示:
有时希望一个算子有多个输出流,例如用于报告:
旁路输出(side output)是实现这些功能的一种方便的方式。除了错误报告,也可以实现流的n路分割。
现在可以处理上一节中忽略的迟到事件。
旁路输出与OutputTag关联。
例如,在PseudoWindow
类中增加
public static final OutputTag<TaxiFare> lateFares = new OutputTag<TaxiFare>("lateFares") {};
注意,必须定义为匿名内部类。
在processElement()
方法中可以将迟到事件发送到旁路输出:
if (eventTime <= timerService.currentWatermark()) {
// This event is late; its window has already been triggered.
ctx.output(lateFares, fare);
} else {
...
}
在main()
方法中可以访问旁路输出流:
hourlyTips.getSideOutput(PseudoWindow.lateFares).print();
https://github.com/apache/flink-training/tree/master/long-ride-alerts
Flink管理的状态是一种分片的键值对存储,每个key对应的状态对象都保存在key所属的TaskManager本地。
Flink管理的状态存储在状态后端(state backend)。Flink有两种状态后端的实现:一种基于RocksDB,将状态存储在磁盘上;另一种基于堆,将状态存储在Java堆内存中。
访问RocksDB状态后端中的对象涉及序列化和反序列化,因此会有更大的开销,但RocksDB可存储的状态数量仅受本地磁盘大小的限制。只有RocksDB能够进行增量快照,对于状态数量大、变化慢的应用有很大好处。
所有状态后端都能异步执行快照,这意味着可以在不妨碍正在进行的流处理的情况下执行快照。
Flink会定期获取每个算子所有状态的快照,并将这些快照保存到持久化的位置,例如分布式文件系统。如果发生故障,Flink可以恢复应用的完整状态并继续处理,就如同没有出现过异常。
这些快照的存储位置由检查点存储(checkpoint storage)定义。有两种检查点存储的实现:一种保存在分布式文件系统上,另一种使用JobManager的堆内存。
注:使用StreamExecutionEnvironment.enableCheckpointing()
启用检查点功能。
Flink使用Chandy-Lamport算法的一种变体,称为异步屏障快照(asynchronous barrier snapshotting)。
当TaskManager接收到检查点协调器(JobManager的一部分)的指示开始生成检查点时,它会让所有source记录自己的偏移量,并将编号的检查点屏障(checkpoint barrier)插入到它们的流中。这些屏障表示每个检查点在流中的位置(类似于水印的概念),如下图所示。
检查点n将包含每个算子在消费了严格位于屏障n之前的所有事件后生成的状态。
当每个算子接收到屏障之后就会记录其状态。有两个输入流的算子(例如CoProcessFunction
)将进行屏障对齐,使得生成的快照包含消费了两个输入流各自的屏障之前的所有事件后生成的状态,如下图所示。
当流处理应用发生错误时,结果可能会产生丢失或重复。Flink根据应用和集群的配置可能产生以下结果:
这些语义保证可使用CheckpointingMode配置。
Flink通过回退(rewinding)和重放(replaying)源数据流来从故障中恢复。理想情况“恰好一次” 并不意味着每个事件都将被处理恰好一次,而是每个事件都将影响Flink管理的状态恰好一次。
为了实现端到端的恰好一次,即source中的每个事件都对sink生效恰好一次,必须满足以下条件: