算子

用户通过算子能将一个或多个 DataStream 转换成新的 DataStream,在应用程序中可以将多个数据转换算子合并成一个复杂的数据流拓扑。

这部分内容将描述 Flink DataStream API 中基本的数据转换 API,数据转换后各种数据分区方式,以及算子的链接策略。

数据流转换

Map

DataStream → DataStream

输入一个元素同时输出一个元素。下面是将输入流中元素数值加倍的 map function:

Java

  1. DataStream<Integer> dataStream = //...
  2. dataStream.map(new MapFunction<Integer, Integer>() {
  3. @Override
  4. public Integer map(Integer value) throws Exception {
  5. return 2 * value;
  6. }
  7. });

Scala

  1. dataStream.map { x => x * 2 }

Python

  1. data_stream = env.from_collection(collection=[1, 2, 3, 4, 5])
  2. data_stream.map(lambda x: 2 * x, output_type=Types.INT())

FlatMap

DataStream → DataStream

输入一个元素同时产生零个、一个或多个元素。下面是将句子拆分为单词的 flatmap function:

Java

  1. dataStream.flatMap(new FlatMapFunction<String, String>() {
  2. @Override
  3. public void flatMap(String value, Collector<String> out)
  4. throws Exception {
  5. for(String word: value.split(" ")){
  6. out.collect(word);
  7. }
  8. }
  9. });

Scala

  1. dataStream.flatMap { str => str.split(" ") }

Python

  1. data_stream = env.from_collection(collection=['hello apache flink', 'streaming compute'])
  2. data_stream.flat_map(lambda x: x.split(' '), output_type=Types.STRING())

Filter

DataStream → DataStream

为每个元素执行一个布尔 function,并保留那些 function 输出值为 true 的元素。下面是过滤掉零值的 filter:

Java

  1. dataStream.filter(new FilterFunction<Integer>() {
  2. @Override
  3. public boolean filter(Integer value) throws Exception {
  4. return value != 0;
  5. }
  6. });

Scala

  1. dataStream.filter { _ != 0 }

Python

  1. data_stream = env.from_collection(collection=[0, 1, 2, 3, 4, 5])
  2. data_stream.filter(lambda x: x != 0)

KeyBy

DataStream → KeyedStream

在逻辑上将流划分为不相交的分区。具有相同 key 的记录都分配到同一个分区。在内部, keyBy() 是通过哈希分区实现的。有多种指定 key的方式。

Java

  1. dataStream.keyBy(value -> value.getSomeKey());
  2. dataStream.keyBy(value -> value.f0);

Scala

  1. dataStream.keyBy(_.someKey)
  2. dataStream.keyBy(_._1)

Python

  1. data_stream = env.from_collection(collection=[(1, 'a'), (2, 'a'), (3, 'b')])
  2. data_stream.key_by(lambda x: x[1], key_type=Types.STRING()) // Key by the result of KeySelector

以下情况,一个类不能作为 key

  1. 它是一种 POJO 类,但没有重写 hashCode() 方法而是依赖于 Object.hashCode() 实现。
  2. 它是任意类的数组。

Reduce

KeyedStream → DataStream

在相同 key 的数据流上“滚动”执行 reduce。将当前元素与最后一次 reduce 得到的值组合然后输出新值。

下面是创建局部求和流的 reduce function:

Java

  1. keyedStream.reduce(new ReduceFunction<Integer>() {
  2. @Override
  3. public Integer reduce(Integer value1, Integer value2)
  4. throws Exception {
  5. return value1 + value2;
  6. }
  7. });

Scala

  1. keyedStream.reduce { _ + _ }

Python

  1. data_stream = env.from_collection(collection=[(1, 'a'), (2, 'a'), (3, 'a'), (4, 'b')], type_info=Types.TUPLE([Types.INT(), Types.STRING()]))
  2. data_stream.key_by(lambda x: x[1]).reduce(lambda a, b: (a[0] + b[0], b[1]))

Window

KeyedStream → WindowedStream

可以在已经分区的 KeyedStreams 上定义 Window。Window 根据某些特征(例如,最近 5 秒内到达的数据)对每个 key Stream 中的数据进行分组。请参阅 windows 获取有关 window 的完整说明。

Java

  1. dataStream
  2. .keyBy(value -> value.f0)
  3. .window(TumblingEventTimeWindows.of(Time.seconds(5)));

Scala

  1. dataStream
  2. .keyBy(_._1)
  3. .window(TumblingEventTimeWindows.of(Time.seconds(5)))

Python

  1. data_stream.key_by(lambda x: x[1]).window(TumblingEventTimeWindows.of(Time.seconds(5)))

WindowAll

DataStream → AllWindowedStream

可以在普通 DataStream 上定义 Window。 Window 根据某些特征(例如,最近 5 秒内到达的数据)对所有流事件进行分组。请参阅windows获取有关 window 的完整说明。

这适用于非并行转换的大多数场景。所有记录都将收集到 windowAll 算子对应的一个任务中。

Java

  1. dataStream
  2. .windowAll(TumblingEventTimeWindows.of(Time.seconds(5)));

Scala

  1. dataStream
  2. .windowAll(TumblingEventTimeWindows.of(Time.seconds(5)))

Python

  1. data_stream.window_all(TumblingEventTimeWindows.of(Time.seconds(5)))

Window Apply

WindowedStream → DataStream

AllWindowedStream → DataStream

将通用 function 应用于整个窗口。下面是一个手动对窗口内元素求和的 function。

如果你使用 windowAll 转换,则需要改用 AllWindowFunction

Java

  1. windowedStream.apply(new WindowFunction<Tuple2<String,Integer>, Integer, Tuple, Window>() {
  2. public void apply (Tuple tuple,
  3. Window window,
  4. Iterable<Tuple2<String, Integer>> values,
  5. Collector<Integer> out) throws Exception {
  6. int sum = 0;
  7. for (value t: values) {
  8. sum += t.f1;
  9. }
  10. out.collect (new Integer(sum));
  11. }
  12. });
  13. // 在 non-keyed 窗口流上应用 AllWindowFunction
  14. allWindowedStream.apply (new AllWindowFunction<Tuple2<String,Integer>, Integer, Window>() {
  15. public void apply (Window window,
  16. Iterable<Tuple2<String, Integer>> values,
  17. Collector<Integer> out) throws Exception {
  18. int sum = 0;
  19. for (value t: values) {
  20. sum += t.f1;
  21. }
  22. out.collect (new Integer(sum));
  23. }
  24. });

Scala

  1. windowedStream.apply { WindowFunction }
  2. // 在 non-keyed 窗口流上应用 AllWindowFunction
  3. allWindowedStream.apply { AllWindowFunction }

Python

  1. class MyWindowFunction(WindowFunction[tuple, int, int, TimeWindow]):
  2. def apply(self, key: int, window: TimeWindow, inputs: Iterable[tuple]) -> Iterable[int]:
  3. sum = 0
  4. for input in inputs:
  5. sum += input[1]
  6. yield sum
  7. class MyAllWindowFunction(AllWindowFunction[tuple, int, TimeWindow]):
  8. def apply(self, window: TimeWindow, inputs: Iterable[tuple]) -> Iterable[int]:
  9. sum = 0
  10. for input in inputs:
  11. sum += input[1]
  12. yield sum
  13. windowed_stream.apply(MyWindowFunction())
  14. # 在 non-keyed 窗口流上应用 AllWindowFunction
  15. all_windowed_stream.apply(MyAllWindowFunction())

WindowReduce

WindowedStream → DataStream

对窗口应用 reduce function 并返回 reduce 后的值。

Java

  1. windowedStream.reduce (new ReduceFunction<Tuple2<String,Integer>>() {
  2. public Tuple2<String, Integer> reduce(Tuple2<String, Integer> value1, Tuple2<String, Integer> value2) throws Exception {
  3. return new Tuple2<String,Integer>(value1.f0, value1.f1 + value2.f1);
  4. }
  5. });

Scala

  1. windowedStream.reduce { _ + _ }

Python

  1. class MyReduceFunction(ReduceFunction):
  2. def reduce(self, value1, value2):
  3. return value1[0], value1[1] + value2[1]
  4. windowed_stream.reduce(MyReduceFunction())

Union

DataStream* → DataStream

将两个或多个数据流联合来创建一个包含所有流中数据的新流。注意:如果一个数据流和自身进行联合,这个流中的每个数据将在合并后的流中出现两次。

Java

  1. dataStream.union(otherStream1, otherStream2, ...);

Scala

  1. dataStream.union(otherStream1, otherStream2, ...)

Python

  1. data_stream.union(otherStream1, otherStream2, ...)

Window Join

DataStream,DataStream → DataStream

根据指定的 key 和窗口 join 两个数据流。

Java

  1. dataStream.join(otherStream)
  2. .where(<key selector>).equalTo(<key selector>)
  3. .window(TumblingEventTimeWindows.of(Time.seconds(3)))
  4. .apply (new JoinFunction () {...});

Scala

  1. dataStream.join(otherStream)
  2. .where(<key selector>).equalTo(<key selector>)
  3. .window(TumblingEventTimeWindows.of(Time.seconds(3)))
  4. .apply { ... }

Python

Python 中尚不支持此特性。

Interval Join

KeyedStream,KeyedStream → DataStream

根据 key 相等并且满足指定的时间范围内(e1.timestamp + lowerBound <= e2.timestamp <= e1.timestamp + upperBound)的条件将分别属于两个 keyed stream 的元素 e1 和 e2 Join 在一起。

Java

  1. // this will join the two streams so that
  2. // key1 == key2 && leftTs - 2 < rightTs < leftTs + 2
  3. keyedStream.intervalJoin(otherKeyedStream)
  4. .between(Time.milliseconds(-2), Time.milliseconds(2)) // lower and upper bound
  5. .upperBoundExclusive(true) // optional
  6. .lowerBoundExclusive(true) // optional
  7. .process(new IntervalJoinFunction() {...});

Scala

  1. // this will join the two streams so that
  2. // key1 == key2 && leftTs - 2 < rightTs < leftTs + 2
  3. keyedStream.intervalJoin(otherKeyedStream)
  4. .between(Time.milliseconds(-2), Time.milliseconds(2))
  5. // lower and upper bound
  6. .upperBoundExclusive(true) // optional
  7. .lowerBoundExclusive(true) // optional
  8. .process(new IntervalJoinFunction() {...})

Python

Python 中尚不支持此特性。

Window CoGroup

DataStream,DataStream → DataStream

根据指定的 key 和窗口将两个数据流组合在一起。

Java

  1. dataStream.coGroup(otherStream)
  2. .where(0).equalTo(1)
  3. .window(TumblingEventTimeWindows.of(Time.seconds(3)))
  4. .apply (new CoGroupFunction () {...});

Scala

  1. dataStream.coGroup(otherStream)
  2. .where(0).equalTo(1)
  3. .window(TumblingEventTimeWindows.of(Time.seconds(3)))
  4. .apply {}

Python

Python 中尚不支持此特性。

Connect

DataStream,DataStream → ConnectedStream

“连接” 两个数据流并保留各自的类型。connect 允许在两个流的处理逻辑之间共享状态。

Java

  1. DataStream<Integer> someStream = //...
  2. DataStream<String> otherStream = //...
  3. ConnectedStreams<Integer, String> connectedStreams = someStream.connect(otherStream);

Scala

  1. someStream : DataStream[Int] = ...
  2. otherStream : DataStream[String] = ...
  3. val connectedStreams = someStream.connect(otherStream)

Python

  1. stream_1 = ...
  2. stream_2 = ...
  3. connected_streams = stream_1.connect(stream_2)

CoMap, CoFlatMap

ConnectedStream → DataStream

类似于在连接的数据流上进行 map 和 flatMap。

Java

  1. connectedStreams.map(new CoMapFunction<Integer, String, Boolean>() {
  2. @Override
  3. public Boolean map1(Integer value) {
  4. return true;
  5. }
  6. @Override
  7. public Boolean map2(String value) {
  8. return false;
  9. }
  10. });
  11. connectedStreams.flatMap(new CoFlatMapFunction<Integer, String, String>() {
  12. @Override
  13. public void flatMap1(Integer value, Collector<String> out) {
  14. out.collect(value.toString());
  15. }
  16. @Override
  17. public void flatMap2(String value, Collector<String> out) {
  18. for (String word: value.split(" ")) {
  19. out.collect(word);
  20. }
  21. }
  22. });

Scala

  1. connectedStreams.map(
  2. (_ : Int) => true,
  3. (_ : String) => false
  4. )
  5. connectedStreams.flatMap(
  6. (_ : Int) => true,
  7. (_ : String) => false
  8. )

Python

  1. class MyCoMapFunction(CoMapFunction):
  2. def map1(self, value):
  3. return value[0] + 1, value[1]
  4. def map2(self, value):
  5. return value[0], value[1] + 'flink'
  6. class MyCoFlatMapFunction(CoFlatMapFunction):
  7. def flat_map1(self, value)
  8. for i in range(value[0]):
  9. yield i
  10. def flat_map2(self, value):
  11. yield value[0] + 1
  12. connectedStreams.map(MyCoMapFunction())
  13. connectedStreams.flat_map(MyCoFlatMapFunction())

Iterate

DataStream → IterativeStream → ConnectedStream

通过将一个算子的输出重定向到某个之前的算子来在流中创建“反馈”循环。这对于定义持续更新模型的算法特别有用。下面的代码从一个流开始,并不断地应用迭代自身。大于 0 的元素被发送回反馈通道,其余元素被转发到下游。

Java

  1. IterativeStream<Long> iteration = initialStream.iterate();
  2. DataStream<Long> iterationBody = iteration.map (/*do something*/);
  3. DataStream<Long> feedback = iterationBody.filter(new FilterFunction<Long>(){
  4. @Override
  5. public boolean filter(Long value) throws Exception {
  6. return value > 0;
  7. }
  8. });
  9. iteration.closeWith(feedback);
  10. DataStream<Long> output = iterationBody.filter(new FilterFunction<Long>(){
  11. @Override
  12. public boolean filter(Long value) throws Exception {
  13. return value <= 0;
  14. }
  15. });

Scala

  1. initialStream.iterate {
  2. iteration => {
  3. val iterationBody = iteration.map {/*do something*/}
  4. (iterationBody.filter(_ > 0), iterationBody.filter(_ <= 0))
  5. }
  6. }

Python

Python 中尚不支持此特性。

Cache

DataStream → CachedDataStream

把算子的结果缓存起来。目前只支持批执行模式下运行的作业。算子的结果在算子第一次执行的时候会被缓存起来,之后的 作业中会复用该算子缓存的结果。如果算子的结果丢失了,它会被原来的算子重新计算并缓存。

Java

  1. DataStream<Integer> dataStream = //...
  2. CachedDataStream<Integer> cachedDataStream = dataStream.cache();
  3. cachedDataStream.print(); // Do anything with the cachedDataStream
  4. ...
  5. env.execute(); // Execute and create cache.
  6. cachedDataStream.print(); // Consume cached result.
  7. env.execute();

Scala

  1. val dataStream : DataStream[Int] = //...
  2. val cachedDataStream = dataStream.cache()
  3. cachedDataStream.print() // Do anything with the cachedDataStream
  4. ...
  5. env.execute() // Execute and create cache.
  6. cachedDataStream.print() // Consume cached result.
  7. env.execute()

Python

  1. data_stream = ... # DataStream
  2. cached_data_stream = data_stream.cache()
  3. cached_data_stream.print()
  4. # ...
  5. env.execute() # Execute and create cache.
  6. cached_data_stream.print() # Consume cached result.
  7. env.execute()

物理分区

Flink 也提供以下方法让用户根据需要在数据转换完成后对数据分区进行更细粒度的配置。

自定义分区

DataStream → DataStream

使用用户定义的 Partitioner 为每个元素选择目标任务。

Java

  1. dataStream.partitionCustom(partitioner, "someKey");
  2. dataStream.partitionCustom(partitioner, 0);

Scala

  1. dataStream.partitionCustom(partitioner, "someKey")
  2. dataStream.partitionCustom(partitioner, 0)

Python

  1. data_stream = env.from_collection(collection=[(2, 'a'), (2, 'a'), (3, 'b')])
  2. data_stream.partition_custom(lambda key, num_partition: key % partition, lambda x: x[0])

随机分区

DataStream → DataStream

将元素随机地均匀划分到分区。

Java

  1. dataStream.shuffle();

Scala

  1. dataStream.shuffle()

Python

  1. data_stream.shuffle()

Rescaling

DataStream → DataStream

将元素以 Round-robin 轮询的方式分发到下游算子。如果你想实现数据管道,这将很有用,例如,想将数据源多个并发实例的数据分发到多个下游 map 来实现负载分配,但又不想像 rebalance() 那样引起完全重新平衡。该算子将只会到本地数据传输而不是网络数据传输,这取决于其它配置值,例如 TaskManager 的 slot 数量。

上游算子将元素发往哪些下游的算子实例集合同时取决于上游和下游算子的并行度。例如,如果上游算子并行度为 2,下游算子的并发度为 6, 那么上游算子的其中一个并行实例将数据分发到下游算子的三个并行实例, 另外一个上游算子的并行实例则将数据分发到下游算子的另外三个并行实例中。再如,当下游算子的并行度为2,而上游算子的并行度为 6 的时候,那么上游算子中的三个并行实例将会分发数据至下游算子的其中一个并行实例,而另外三个上游算子的并行实例则将数据分发至另下游算子的另外一个并行实例。

当算子的并行度不是彼此的倍数时,一个或多个下游算子将从上游算子获取到不同数量的输入。

请参阅下图来可视化地感知上述示例中的连接模式:

Checkpoint barriers in data streams

Java

  1. dataStream.rescale();

Scala

  1. dataStream.rescale()

Python

  1. data_stream.rescale()

广播

DataStream → DataStream

将元素广播到每个分区 。

Java

  1. dataStream.broadcast();

Scala

  1. dataStream.broadcast()

Python

  1. data_stream.broadcast()

算子链和资源组

将两个算子链接在一起能使得它们在同一个线程中执行,从而提升性能。Flink 默认会将能链接的算子尽可能地进行链接(例如, 两个 map 转换操作)。此外, Flink 还提供了对链接更细粒度控制的 API 以满足更多需求:

如果想对整个作业禁用算子链,可以调用 StreamExecutionEnvironment.disableOperatorChaining()。下列方法还提供了更细粒度的控制。需要注意的是,这些方法只能在 DataStream 转换操作后才能被调用,因为它们只对前一次数据转换生效。例如,可以 someStream.map(...).startNewChain() 这样调用,而不能 someStream.startNewChain() 这样。

一个资源组对应着 Flink 中的一个 slot 槽,更多细节请看 slots 。 你可以根据需要手动地将各个算子隔离到不同的 slot 中。

创建新链

基于当前算子创建一个新的算子链。
后面两个 map 将被链接起来,而 filter 和第一个 map 不会链接在一起。

Java

  1. someStream.filter(...).map(...).startNewChain().map(...);

Scala

  1. someStream.filter(...).map(...).startNewChain().map(...)

Python

  1. some_stream.filter(...).map(...).start_new_chain().map(...)

禁止链接

禁止和 map 算子链接在一起。

Java

  1. someStream.map(...).disableChaining();

Scala

  1. someStream.map(...).disableChaining()

Python

  1. some_stream.map(...).disable_chaining()

配置 Slot 共享组

为某个算子设置 slot 共享组。Flink 会将同一个 slot 共享组的算子放在同一个 slot 中,而将不在同一 slot 共享组的算子保留在其它 slot 中。这可用于隔离 slot 。如果所有输入算子都属于同一个 slot 共享组,那么 slot 共享组从将继承输入算子所在的 slot。slot 共享组的默认名称是 “default”,可以调用 slotSharingGroup(“default”) 来显式地将算子放入该组。

Java

  1. someStream.filter(...).slotSharingGroup("name");

Scala

  1. someStream.filter(...).slotSharingGroup("name")

Python

  1. some_stream.filter(...).slot_sharing_group("name")

名字和描述

Flink里的算子和作业节点会有一个名字和一个描述。名字和描述。名字和描述都是用来介绍一个算子或者节点是在做什么操作,但是他们会被用在不同地方。

名字会用在用户界面、线程名、日志、指标等场景。节点的名字会根据节点中算子的名字来构建。 名字需要尽可能的简洁,避免对外部系统产生大的压力。

描述主要用在执行计划展示,以及用户界面展示。节点的描述同样是根据节点中算子的描述来构建。 描述可以包括详细的算子行为的信息,以便我们在运行时进行debug分析。

Java

  1. someStream.filter(...).name("filter").setDescription("x in (1, 2, 3, 4) and y > 1");

Scala

  1. someStream.filter(...).name("filter").setDescription("x in (1, 2, 3, 4) and y > 1")

Python

  1. some_stream.filter(...).name("filter").set_description("x in (1, 2, 3, 4) and y > 1")

节点的描述默认是按照一个多行的树形结构来构建的,用户可以通过把pipeline.vertex-description-mode设为CASCADING, 实现将描述改为老版本的单行递归模式。

Flink SQL框架生成的算子默认会有一个由算子的类型以及id构成的名字,以及一个带有详细信息的描述。 用户可以通过将table.exec.simplify-operator-name-enabled设为false,将名字改为和以前的版本一样的详细描述。

当一个作业的拓扑很复杂时,用户可以把pipeline.vertex-name-include-index-prefix设为true,在节点的名字前增加一个拓扑序的前缀,这样就可以很容易根据指标以及日志的信息快速找到拓扑图中对应节点。