目录
不会点
DataStream API
编辑
执行环境
创建执行环境
执行模式
触发程序执行
源算子
准备基础类型
从集合中读取数据
从文件读取数据
从 Socket 读取数据
从 Kafka 读取数据 (没学过)
从数据生成器读取数据
Flink 支持的数据类型
转换算子
基本转换算子
map
flatMap
filter
聚合算子
keyBy(准备工作)
sum/min-max/minBy-maxBy
reduce
UDF
函数类
富函数类(多个open、close)
物理分区算子
随机分区
轮询分配(Round-Robin)
重缩放(Rescale)
广播(Broadcast)
全局分区(并行度转1)
自定义分区
分流
合流
输出算子
连接到外部系统
输出到文件
输出到kafka(没学过
MySQL
DIY
索引:getRuntimeContext().getIndexOfThisSubtask()
子任务名: getRuntimeContext().getTaskNameWithSubtasks()
索引: 1
子任务名: Map -> Sink: Print to Std. Out (2/2)#0
Flink 程序可以在各种上下文环境中运行:我们可以在本地 JVM 中执行程序,也可以提交到远程集群上运行。
不同的环境,代码的提交运行的过程会有所不同。这就要求我们在提交作业执行计算时,首先必须获取当前 Flink 的运行环境,从而建立起与Flink 框架之间的联系
我们要获取的执行环境,是 StreamExecutionEnvironment 类的对象,这是所有 Flink 程序的基础。在代码中创建执行环境的方式,就是调用这个类的静态方法,具体有以下三种
1)getExecutionEnvironment
最简单的方式,就是直接调用 getExecutionEnvironment 方法。它会根据当前运行的上下文直接得到正确的结果:如果程序是独立运行的,就返回一个本地执行环境;如果是创建了jar 包,然后从命令行调用它并提交到集群执行,那么就返回集群的执行环境。也就是说,这个方法会根据当前运行的方式,自行决定该返回什么样的运行环境
2)createLocalEnvironment
这个方法返回一个本地执行环境。可以在调用时传入一个参数,指定默认的并行度;如果不传入,则默认并行度就是本地的CPU 核心数
3)createRemoteEnvironment
这个方法返回集群执行环境。需要在调用时指定 JobManager 的主机名和端口号,并指定要在集群中运行的 Jar 包。
/**
* @param host The host name or address of the master (JobManager), where the program should be
* executed.
* @param port The port of the master (JobManager), where the program should be executed.
* @param jarFiles The JAR files with code that needs to be shipped to the cluster. If the
* program uses user-defined functions, user-defined input formats, or any libraries, those
* must be provided in the JAR files.
* @return A remote environment that executes the program on a cluster.
*/
public static StreamExecutionEnvironment createRemoteEnvironment(
String host, int port, String... jarFiles) {
return new RemoteStreamEnvironment(host, port, jarFiles);
}
DataStream API 执行模式包括:流执行模式、批执行模式和自动模式。
从 Flink 1.12 开始,官方推荐的做法是直接使用 DataStream API,在提交任务时通过将执行模式设为BATCH 来进行批处理。不建议使用DataSet API。
流批一体
需要注意的是,写完输出(sink)操作并不代表程序已经结束。因为当main()方法被调用时,其实只是定义了作业的每个执行操作,然后添加到数据流图中;这时并没有真正处理数据——因为数据可能还没来。Flink 是由事件驱动的,只有等到数据到来,才会触发真正的计算,这也被称为“延迟执行”或“懒执行”。
所以我们需要显式地调用执行环境的 execute()方法,来触发程序执行。execute()方法将一直等待作业完成,然后返回一个执行结果(JobExecutionResult)。
env.execute();
Flink 可以从各种来源获取数据,然后构建DataStream 进行转换处理。一般将数据的输入来源称为数据源(data source),而读取数据的算子就是源算子(source operator)。所以,source 就是我们整个处理程序的输入端。
从 Flink1.12 开始,主要使用流批统一的新 Source 架构:
DataStreamSourcestream = env.fromSource(…)
使用WaterSensor 作为数据模型。
import java.time.LocalDate;
import java.util.Objects;
public class WaterSensor {
public String id; // id
public Long ts; // 时间戳
public Integer vc; // 水位记录
public WaterSensor(){
}
public WaterSensor(String id, Long ts, Integer vc) {
this.id = id;
this.ts = ts;
this.vc = vc;
}
public String getId() {
return id;
}
public void setId(String id) {
this.id = id;
}
public Long getTs() {
return ts;
}
public void setTs(Long ts) {
this.ts = ts;
}
public Integer getVc() {
return vc;
}
public void setVc(Integer vc) {
this.vc = vc;
}
@Override
public int hashCode() {
return super.hashCode();
}
@Override
public boolean equals(Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
WaterSensor that = (WaterSensor) o;
return Objects.equals(id, that.id) &&
Objects.equals(ts, that.ts) &&
Objects.equals(vc, that.vc);
}
@Override
public String toString() {
return "WaterSensor{" +
"id='" + id + '\'' +
", ts=" + ts +
", vc=" + vc +
'}';
}
}
类是公有(public)的
有一个无参的构造方法
所有属性都是公有(public)的
所有属性的类型都是可以序列化的Flink 会把这样的类作为一种特殊的 POJO(Plain Ordinary Java Object 简单的Java 对象,实际就是普通 JavaBeans)数据类型来对待,方便数据的解析和序列化。另外我们在类中还重写了 toString 方法,主要是为了测试输出显示更清晰。
我们这里自定义的 POJO 类会在后面的代码中频繁使用,所以在后面的代码中碰到,把
这里的POJO 类导入就好了。
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env =
StreamExecutionEnvironment.getExecutionEnvironment();
List data = Arrays.asList(1, 22, 3);
DataStreamSource ds1 = env.fromCollection(data);
ds1.print();
DataStreamSource ds2 = env.fromElements(1,23,44);
ds2.print();
env.execute();
}
org.apache.flink
flink-connector-files
${flink.version}
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env =
StreamExecutionEnvironment.getExecutionEnvironment();
FileSource fileSource =
FileSource.forRecordStreamFormat(new TextLineInputFormat(), new Path("data/goodnight.txt")).build();
env.fromSource(fileSource, WatermarkStrategy.noWatermarks(),"any")
.print();
env.execute();
}
// TODO 准备环境
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
// TODO read file
// String path = "data/goodnight.txt";
DataStreamSource hadoop1 = env.socketTextStream("hadoop1", 7777);
org.apache.flink
flink-connector-kafka
${flink.version}
public class SourceKafka {
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
KafkaSource kafkaSource =
KafkaSource.builder()
.setBootstrapServers("hadoop102:9092")
.setTopics("topic_1")
.setGroupId("atguigu")
.setStartingOffsets(OffsetsInitializer.latest())
.setValueOnlyDeserializer(new SimpleStringSchema())
.build();
DataStreamSource stream = env.fromSource(kafkaSource,
WatermarkStrategy.noWatermarks(), "kafka-source");
stream.print("Kafka");
env.execute();
}
}
Flink 从 1.11 开始提供了一个内置的 DataGen 连接器,主要是用于生成一些随机数,用于在没有数据源的时候,进行流任务的测试以及性能测试等。1.17 提供了新的 Source 写法,需要导入依赖:
org.apache.flink
flink-connector-datagen
${flink.version}
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
env.setParallelism(3);
// recordsPerCheckpoint has to be greater or equal to parallelism.
// Either decrease the parallelism or increase the number of recordsPerCheckpoint.
/**
* GeneratorFun接口重写map,输入类型Long
* Long 生成的最大值,从1自增
* 限速 每秒几条
* 返回类型
*/
DataGeneratorSource source = new DataGeneratorSource<>(
new GeneratorFunction() {
@Override
public String map(Long aLong) throws Exception {
return "Number: " + aLong;
}
},
10,
RateLimiterStrategy.perSecond(2),
Types.STRING
// recordsPerCheckpoint has to be greater or equal to parallelism.
// Either decrease the parallelism or increase the number of recordsPerCheckpoint.
);
env.fromSource(source, WatermarkStrategy.noWatermarks(), "DataGen").print();
env.execute();
}
Flink 使用“类型信息”(TypeInformation)来统一表示数据类型。TypeInformation 类是 Flink 中所有类型描述符的基类。它涵盖了类型的一些基本属性,并为每个数据类型生成特定 的序列化器、反序列化器和比较器。
Flink 支持所有的 Java 类和 Scala 类。不过如果没有按照上面 POJO 类型的要求来定义, 就会被 Flink 当作泛型类来处理。Flink 会把泛型类型当作黑盒,无法获取它们内部的属性; 它们也不是由 Flink 本身序列化的,而是由 Kryo 序列化的。 在这些类型中,元组类型和 POJO 类型最为灵活,因为它们支持创建复杂类型。而相比 之下,POJO 还支持在键(key)的定义中直接使用字段名,这会让我们的代码可读性大大增 加。所以,在项目实践中,往往会将流处理程序中的元素类型定为 Flink 的 POJO 类型。
Flink 还具有一个类型提取系统,可以分析函数的输入和返回类型,自动获取类型信息, 从而获得对应的序列化器和反序列化器。但是,由于 Java 中泛型擦除的存在,在某些特殊情 况下(比如 Lambda 表达式中),自动提取的信息是不够精细的,只有显式地 告诉系统当前的返回类型,才能正确地解析出完整数据。
.map(word -> Tuple2.of(word, 1L))
.returns(Types.TUPLE(Types.STRING, Types.LONG));
泛型擦除
Java 的泛型擦除(Generic Type Erasure)是一种编译器优化和类型安全机制,它发生在编译阶段,而不是运行时。泛型擦除的主要目的是在支持泛型的同时保持与 Java 早期版本的向后兼容性,并避免在运行时引入额外的开销。
泛型擦除的关键点包括:
类型擦除:在编译时,Java 泛型的类型信息被擦除,这意味着编译后的字节码中不再包含泛型类型信息。例如,
List
和List
在运行时都被视为List
。类型参数擦除:泛型类型的类型参数(例如,
)也会被擦除,编译后的字节码中不再包含类型参数信息。这些类型参数被擦除为它们的上边界或 Object 类型。
类型强制转换:在运行时,泛型类型的实例通常会被强制转换为适当的类型,以满足编译时的类型检查。这可能导致运行时的 ClassCastException 异常,如果类型不匹配。
虽然泛型擦除是 Java 泛型的一个基本特性,但它也导致了一些泛型编程的限制和复杂性,需要开发人员注意处理泛型类型的类型安全性和边界情况
public class t1_map {
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
DataStreamSource stream = env.fromElements(
new WaterSensor("sensor_1", 1L, 1),
new WaterSensor("sensor_2", 2L, 2)
);
// 方式一:传入匿名类,实现 MapFunction
stream.map(new MapFunction() {
@Override
public String map(WaterSensor e) throws Exception {
return e.id;
}
}).print();
// 方式二:
// stream.map((MapFunction) e -> e.id).print();
SingleOutputStreamOperator map = stream.map(s -> s.getId());
// 方式三:传入 MapFunction 的实现类
// stream.map(new UserMap()).print();
env.execute();
}
public static class UserMap implements MapFunction {
@Override
public String map(WaterSensor e) throws Exception {
return e.id;
}
}
}
先按照某种规则对数据进行打散拆分,可以产生 0 到多个元素,再对拆分后的元素做转换处理
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
DataStreamSource stream = env.fromElements(
new WaterSensor("sensor_1", 1L, 1),
new WaterSensor("sensor_1", 2L, 2),
new WaterSensor("sensor_2", 2L, 2),
new WaterSensor("sensor_3", 3L, 3)
);
stream.flatMap(new MyFlatMap()).print();
env.execute();
}
public static class MyFlatMap implements FlatMapFunction {
@Override
public void flatMap(WaterSensor value, Collector out) throws Exception {
if (value.id.equals("sensor_1")) {
out.collect(String.valueOf(value.vc));
}
else if (value.id.equals("sensor_2")) {
out.collect(String.valueOf(value.ts));
out.collect(String.valueOf(value.vc));
}
}
}
进行filter 转换之后的新数据流的数据类型与原数据流是相同的。filter 转换需要传入的参
数需要实现 FilterFunction 接口,而 FilterFunction 内要实现 filter()方法,就相当于一个返回布
尔类型的条件表达式。
public class t2_filter {
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
DataStreamSource stream = env.fromElements(
new WaterSensor("sensor_1", 1L, 1),
new WaterSensor("sensor_1", 2L, 2),
new WaterSensor("sensor_2", 2L, 2),
new WaterSensor("sensor_3", 3L, 3)
);
// 方式一:传入匿名类实现FilterFunction
stream.filter(new FilterFunction() {
@Override
public boolean filter(WaterSensor e) throws Exception {
return e.id.equals("sensor_1");
}
}).print();
stream.filter((FilterFunction) e -> e.id.equals("sensor_1")).print();
stream.filter(e -> e.id.equals("sensor_1")).print();
// 方式二:传入 FilterFunction 实现类
// stream.filter(new UserFilter()).print();
env.execute();
}
public static class UserFilter implements FilterFunction {
@Override
public boolean filter(WaterSensor e) throws Exception {
return e.id.equals("sensor_1");
}
}
}
计算的结果不仅依赖当前数据,还跟之前的数据有关,相当于要把所有数据聚在一起进
行汇总合并——这就是所谓的“聚合”(Aggregation),类似于MapReduce 中的reduce 操作。
keyBy 是聚合前必须要用到的一个算子。keyBy 通过指定键(key),可以将一条流从逻辑上划分成不同的分区(partitions)。这里所说的分区,其实就是并行处理的子任务。
所有具有相同的key 的数据,都将被发往同一个分区。
对于Flink 而言,DataStream 是没有直接进行聚合的API 的。因为我们对海量数据做聚合
肯定要进行分区并行处理,这样才能提高效率。所以在Flink 中,要做聚合,需要先进行分区;这个操作就是通过 keyBy 来完成的。在内部,是通过计算 key 的哈希值(hash code),对分区数进行取模运算来实现的。所以
这里 key 如果是POJO 的话,必须要重写 hashCode()方法。
以id 作为key 做一个分区操作,代码实现如下
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
DataStreamSource stream = env.fromElements(
new WaterSensor("sensor_1", 1L, 1),
new WaterSensor("sensor_1", 2L, 2),
new WaterSensor("sensor_2", 2L, 2),
new WaterSensor("sensor_3", 3L, 3)
);
// 方式一:使用 Lambda 表达式
KeyedStream keyedStream = stream.keyBy(e -> e.id);
// 方式二:使用匿名类实现KeySelector
KeyedStream keyedStream1 = stream.keyBy(new KeySelector() {
@Override
public String getKey(WaterSensor e) throws Exception {
return e.id;
}
});
env.execute();
}
需要注意的是,keyBy 得到的结果将不再是 DataStream,而是会将 DataStream 转换为KeyedStream。KeyedStream 可以认为是“分区流”或者“键控流”,它是对 DataStream 按照key 的一个逻辑分区,所以泛型有两个类型:除去当前流中的元素类型外,还需要指定key 的类型。
只有基于它才可以做后续的聚合操作(比如sum,reduce)。
min(别的字段就第一个不变了)
minBy(别的字段会对应起来)
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
env.setParallelism(1);
DataStreamSource stream = env.fromElements(
new WaterSensor("sensor_1", 1L, 1),
new WaterSensor("sensor_1", 2L, 2),
new WaterSensor("sensor_2", 3L, 6),
new WaterSensor("sensor_2", 2L, 9),
new WaterSensor("sensor_2", 3L, 4),
new WaterSensor("sensor_2", 3L, 5),
new WaterSensor("sensor_3", 8L, 3),
new WaterSensor("sensor_3", 7L, 7),
new WaterSensor("sensor_3", 6L, 1)
);
// 方式一:使用 Lambda 表达式
KeyedStream KS = stream.keyBy(e -> e.id);
// // 方式二:使用匿名类实现KeySelector
// KeyedStream keyedStream1 = stream.keyBy(new KeySelector() {
// @Override
// public String getKey(WaterSensor e) throws Exception {
// return e.id;
// }
// });
// KS.sum(2).print(); // Cannot reference field by position on PojoType
// KS.sum("vc").print();
/* out
WaterSensor{id='sensor_1', ts=1, vc=1}
WaterSensor{id='sensor_1', ts=1, vc=3}
WaterSensor{id='sensor_2', ts=3, vc=6}
WaterSensor{id='sensor_2', ts=3, vc=11}
WaterSensor{id='sensor_2', ts=3, vc=15}
WaterSensor{id='sensor_2', ts=3, vc=17}
WaterSensor{id='sensor_3', ts=3, vc=3}
WaterSensor{id='sensor_3', ts=3, vc=6}
WaterSensor{id='sensor_3', ts=3, vc=9}
*/
// KS.min("vc").print();
/* 一条一条读进来,判断目前哪个最小,另外的值居然都是第一条的!!!!
WaterSensor{id='sensor_1', ts=1, vc=1}
WaterSensor{id='sensor_1', ts=1, vc=1}
WaterSensor{id='sensor_2', ts=3, vc=6}
WaterSensor{id='sensor_2', ts=3, vc=6}
WaterSensor{id='sensor_2', ts=3, vc=4}
WaterSensor{id='sensor_2', ts=3, vc=4}
WaterSensor{id='sensor_3', ts=8, vc=3}
WaterSensor{id='sensor_3', ts=8, vc=3}
WaterSensor{id='sensor_3', ts=8, vc=1}
*/
KS.minBy("vc").print();
/* 合理的,别的值都是对应的
WaterSensor{id='sensor_1', ts=1, vc=1}
WaterSensor{id='sensor_1', ts=1, vc=1}
WaterSensor{id='sensor_2', ts=3, vc=6}
WaterSensor{id='sensor_2', ts=3, vc=6}
WaterSensor{id='sensor_2', ts=3, vc=4}
WaterSensor{id='sensor_2', ts=3, vc=4}
WaterSensor{id='sensor_3', ts=8, vc=3}
WaterSensor{id='sensor_3', ts=8, vc=3}
WaterSensor{id='sensor_3', ts=6, vc=1}
*/
env.execute();
}
reduce 可以对已有的数据进行归约处理,把每一个新输入的数据和当前已经归约出来的值,再做一个聚合计算。
调用 KeyedStream 的 reduce 方法时,需要传入一个参数,实现 ReduceFunction 接口。接
口在源码中的定义如下:public interface ReduceFunction
extends Function, Serializable {
T reduce(T value1, T value2) throws Exception;
}
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
env.setParallelism(1);
DataStreamSource stream = env.fromElements(
new WaterSensor("sensor_1", 1L, 1),
new WaterSensor("sensor_1", 2L, 2),
new WaterSensor("sensor_2", 3L, 6),
new WaterSensor("sensor_2", 2L, 9),
new WaterSensor("sensor_2", 3L, 4),
new WaterSensor("sensor_2", 3L, 5),
new WaterSensor("sensor_3", 8L, 3),
new WaterSensor("sensor_3", 7L, 7),
new WaterSensor("sensor_3", 6L, 1)
);
KeyedStream KS = stream.keyBy(e -> e.id);
KS.reduce(new ReduceFunction() {
@Override
public WaterSensor reduce(WaterSensor value1, WaterSensor value2) throws Exception {
System.out.println("value1: "+value1);
System.out.println("value2: "+value2);
return new WaterSensor(value1.id, value2.ts, value1.vc+value2.vc);
}
}).print();
/**
* WaterSensor{id='sensor_1', ts=1, vc=1} 第一条不进入reduce
* value1: WaterSensor{id='sensor_1', ts=1, vc=1}
* value2: WaterSensor{id='sensor_1', ts=2, vc=2}
* WaterSensor{id='sensor_1', ts=2, vc=3}
*
* WaterSensor{id='sensor_2', ts=3, vc=6} 第一条不进入reduce
* value1: WaterSensor{id='sensor_2', ts=3, vc=6}
* value2: WaterSensor{id='sensor_2', ts=2, vc=9}
* WaterSensor{id='sensor_2', ts=2, vc=15}
* value1: WaterSensor{id='sensor_2', ts=2, vc=15}
* value2: WaterSensor{id='sensor_2', ts=3, vc=4}
* WaterSensor{id='sensor_2', ts=3, vc=19}
* value1: WaterSensor{id='sensor_2', ts=3, vc=19}
* value2: WaterSensor{id='sensor_2', ts=3, vc=5}
* WaterSensor{id='sensor_2', ts=3, vc=24}
*
* WaterSensor{id='sensor_3', ts=8, vc=3} 第一条不进入reduce
* value1: WaterSensor{id='sensor_3', ts=8, vc=3}
* value2: WaterSensor{id='sensor_3', ts=7, vc=7}
* WaterSensor{id='sensor_3', ts=7, vc=10}
* value1: WaterSensor{id='sensor_3', ts=7, vc=10}
* value2: WaterSensor{id='sensor_3', ts=6, vc=1}
* WaterSensor{id='sensor_3', ts=6, vc=11}
*/
env.execute();
}
public class t6_UDF {
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
env.setParallelism(1);
DataStreamSource stream = env.fromElements(
new WaterSensor("sensor_1", 1L, 1),
new WaterSensor("sensor_1", 2L, 2),
new WaterSensor("sensor_2", 3L, 6),
new WaterSensor("sensor_2", 2L, 9),
new WaterSensor("sensor_2", 3L, 4),
new WaterSensor("sensor_2", 3L, 5),
new WaterSensor("sensor_3", 8L, 3),
new WaterSensor("sensor_3", 7L, 7),
new WaterSensor("sensor_3", 6L, 1)
);
// 方式一 自定义函数实现
// SingleOutputStreamOperator filter = stream.filter(new UserFilter()).print();
// 通过匿名类来实现 FilterFunction 接口:
stream.filter(new FilterFunction() {
@Override
public boolean filter(WaterSensor value) throws Exception {
return value.id.equals("sensor_1");
}
}).print();
// lambda
stream.filter(value -> value.id.equals("sensor_1")).print();
env.execute();
}
public static class UserFilter implements FilterFunction {
@Override
public boolean filter(WaterSensor e) throws Exception {
return e.id.equals("sensor_1");
}
}
}
所有的 Flink 函数类都有其Rich 版本。富函数类一般是以抽象类的形式出现的。例如:RichMapFunction、RichFilterFunction、RichReduceFunction 等。
Rich Function 有生命周期的概念。典型的生命周期方法有:
⚫ open()方法,是 Rich Function 的初始化方法,也就是会开启一个算子的生命周期,当一个算子的实际工作方法例如 map()或者 filter()方法被调用之前,open()会首先被调用。
⚫ close()方法,是生命周期中的最后一个调用的方法,类似于结束方法。一般用来做一些清理工作。
这里的生命周期方法,对于一个并行子任务来说只会调用一次;而对应的,实际工作方法,例如 RichMapFunction 中的 map(),在每条数据到来后都会触发一次调用。
来看一个例子说明:
public class t6_UDF_rich {
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
env.setParallelism(2);
env.fromElements(1,2,3,4)
.map(new RichMapFunction() {
@Override
public void open(Configuration parameters) throws Exception {
super.open(parameters);
System.out.println(" 索引: " + getRuntimeContext().getIndexOfThisSubtask() + " 的任务的生命周期开始, " +
"子任务名: "+ getRuntimeContext().getTaskNameWithSubtasks()+"调用open");
}
@Override
public Integer map(Integer integer) throws Exception {
return integer * 10;
}
@Override
public void close() throws Exception {
super.close();
System.out.println(" 索引: " + getRuntimeContext().getIndexOfThisSubtask() + " 的任务的生命周期结束, " +
"子任务名: "+ getRuntimeContext().getTaskNameWithSubtasks()+"调用close");
}
})
.print();
env.execute();
}
}
// 索引: 0 的任务的生命周期开始, 子任务名: Map -> Sink: Print to Std. Out (1/2)#0调用open
// 索引: 1 的任务的生命周期开始, 子任务名: Map -> Sink: Print to Std. Out (2/2)#0调用open
// 1> 10
// 2> 20
// 2> 40
// 1> 30
// 索引: 0 的任务的生命周期结束, 子任务名: Map -> Sink: Print to Std. Out调用close
// 索引: 1 的任务的生命周期结束, 子任务名: Map -> Sink: Print to Std. Out调用close
常见的物理分区策略有:随机分配(Random)、轮询分配(Round-Robin)、重缩放(Rescale)和广播(Broadcast)。
最简单的重分区方式就是直接“洗牌”。通过调用 DataStream 的.shuffle()方法,将数据随
机地分配到下游算子的并行任务中去。
轮询,简单来说就是“发牌”,按照先后顺序将数据做依次分发。通过调用 DataStream.rebalance()方法,就可以实现轮询重分区。rebalance 使用的是Round-Robin 负载均衡算法,可以将输入流数据平均分配到下游的并行任务中去。
重缩放分区和轮询分区非常相似。
重分区是将数据流重新分区为新的分区数,通常涉及改变分区策略或将数据重新组织以匹配不同的并行度需求。
- 重分区通常用于将数据重新分发到更多或更少的并行任务,以实现负载均衡或更好的并行性。
- 重分区可能会引入数据洗牌和网络传输的开销。
这种方式其实不应该叫做“重分区”,因为经过广播之后,数据会在不同的分区都保留一份,可能进行重复处理。可以通过调用 DataStream 的 broadcast()方法,将输入数据复制并发送到下游算子的所有并行任务中去。
全局分区也是一种特殊的分区方式。这种做法非常极端,通过调用.global()方法,会将所有的输入流数据都发送到下游算子的第一个并行子任务中去。这就相当于强行让下游任务并行度变成了1,所以使用这个操作需要非常谨慎,可能对程序造成很大的压力。
当 Flink 提供的所有分区策略都不能满足用户的需求时,我们可以通过使用partitionCustom()方法来自定义分区策略。
public class t7_DIYpartition {
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
env.setParallelism(2);
DataStreamSource stream = env.fromElements(1, 2, 3, 4, 5, 6, 7, 8);
stream.partitionCustom(new MyPartitioner(), value -> value).print();
env.execute();
}
public static class MyPartitioner implements Partitioner {
@Override
public int partition(Integer key, int numPartitions) {
if (key < 5) {
return 0;
} else{
return 1;
}
// 这里如果分三个区就报错了,可能因为并行度?
}
}
}
所谓“分流”,就是将一条数据流拆分成完全独立的两条、甚至多条流。也就是基于一个DataStream,定义一些筛选条件,将符合条件的数据拣选出来放到对应的流里。
就这么个理
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
env.setParallelism(2);
DataStreamSource stream = env.fromElements(1, 2, 3, 4, 5, 6, 7, 8);
SingleOutputStreamOperator ds1 = stream.filter(x -> x % 2 == 0);
SingleOutputStreamOperator ds2 = stream.filter(x -> x % 2 != 0);
ds1.print("ds1");
ds2.print("ds2");
env.execute();
}
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
env.setParallelism(1);
DataStreamSource stream = env.fromElements(new WaterSensor("sensor_1", 1L, 1),
new WaterSensor("sensor_1", 2L, 2),
new WaterSensor("sensor_2", 3L, 6),
new WaterSensor("sensor_2", 2L, 9),
new WaterSensor("sensor_2", 3L, 4),
new WaterSensor("sensor_2", 3L, 5),
new WaterSensor("sensor_3", 8L, 3),
new WaterSensor("sensor_3", 7L, 7),
new WaterSensor("sensor_3", 6L, 1));
SingleOutputStreamOperator process = stream.process(new ProcessFunction() {
@Override
public void processElement(WaterSensor value, ProcessFunction.Context ctx, Collector out) throws Exception {
if (value.getId().equals("sensor_1")) {
ctx.output(new OutputTag("s1", Types.POJO(WaterSensor.class)), value);
} else if (value.getId().equals("sensor_2")) {
ctx.output(new OutputTag("s2", Types.POJO(WaterSensor.class)), value);
} else {
out.collect(value);
}
}
});
process.print();
process.getSideOutput(new OutputTag("s1", Types.POJO(WaterSensor.class))).print("s1");
process.getSideOutput(new OutputTag("s2", Types.POJO(WaterSensor.class))).print("s2");
env.execute();
/*
s1> WaterSensor{id='sensor_1', ts=1, vc=1}
s1> WaterSensor{id='sensor_1', ts=2, vc=2}
s2> WaterSensor{id='sensor_2', ts=3, vc=6}
s2> WaterSensor{id='sensor_2', ts=2, vc=9}
s2> WaterSensor{id='sensor_2', ts=3, vc=4}
s2> WaterSensor{id='sensor_2', ts=3, vc=5}
WaterSensor{id='sensor_3', ts=8, vc=3}
WaterSensor{id='sensor_3', ts=7, vc=7}
WaterSensor{id='sensor_3', ts=6, vc=1}
*/
}
在实际应用中,我们经常会遇到来源不同的多条流,需要将它们的数据进行联合处理。 所以 Flink 中合流的操作会更加普遍,对应的 API 也更加丰富。
联合(Union)
最简单的合流操作,就是直接将多条流合在一起,叫作流的“联合”(union)。联合操作 要求必须流中的数据类型必须相同,合并之后的新流会包括所有流中的元素,数据类型不变。
stream1.union(stream2, stream3, ...)
SideOutputDataStream s1 = process.getSideOutput(new OutputTag("s1", Types.POJO(WaterSensor.class)));
SideOutputDataStream s2 = process.getSideOutput(new OutputTag("s2", Types.POJO(WaterSensor.class)));
DataStream union = process.union(s1, s2);
union.print();
连接(Connect)
流的联合虽然简单,不过受限于数据类型不能改变,灵活性大打折扣,所以实际应用较少出现。除了联合(union),Flink 还提供了另外一种方便的合流操作——连接(connect)。
为了处理更加灵活,连接操作允许流的数据类型不同。 但我们知道一个DataStream中的数据只能有唯一的类型, 所以连接得到的并不是DataStream,而是一个“连接流”。
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
env.setParallelism(1);
DataStreamSource stream1 = env.fromElements(11, 22, 33);
DataStreamSource stream2 = env.fromElements("aaa", "bbb", "ccc");
// 好像只能connect两个
ConnectedStreams connect = stream1.connect(stream2);
// 流1类型 流2类型 输出类型, map完又变成了dataStream
connect.map(new CoMapFunction() {
@Override
public String map1(Integer value) throws Exception {
return value.toString()+"str";
}
@Override
public String map2(String value) throws Exception {
return value;
}
}).print();
/**
* aaa
* 11
* bbb
* 22
* ccc
* 33
*/
env.execute();
}
由于需要“一国两制”,因此调用.map()方法时传入的不再是一个简单的 MapFunction,而是一个 CoMapFunction,表示分别对两条流中的数据执行 map 操作。这个接 口有三个类型参数,依次表示第一条流、第二条流,以及合并后的流中的数据类型。需要实 现的方法也非常直白:.map1()就是对第一条流中数据的 map 操作,.map2()则是针对第二条流。
与 CoMapFunction 类似,如果是调用.map()就需要传入一个 CoMapFunction,需要实现map1()、map2()两个方法;而调用.process()时,传入的则是一个 CoProcessFunction。它也是 “处理函数”家族中的一员,用法非常相似。它需要实现的就是 processElement1()、 processElement2()两个方法,在每个数据到来时,会根据来源的流调用其中的一个方法进行处 理。
Flink 的 DataStream API 专门提供了向外部写入数据的方法:addSink。与 addSource 类似,addSink 方法对应着一个“Sink”算子,主要就是用来实现与外部系统连接、并将数据提交写入的;Flink 程序中所有对外的输出操作,一般都是利用 Sink 算子完成的。
Flink1.12 开始,同样重构了 Sink 架构,stream.sinkTo(…) ,当然,Sink 多数情况下同样并不需要我们自己实现。之前我们一直在使用的 print 方法其实就是一种 Sink,它表示将数据流写入标准控制台打印输出。Flink 官方为我们提供了一部分的框架的Sink 连接器。如下图所示,列出了 Flink 官方目前支持的第三方系统连接器:
Flink 专门提供了一个流式文件系统的连接器:FileSink,为批处理和流处理提供了一个统一的Sink,它可以将分区文件写入 Flink 支持的文件系统。
FileSink 支持行编码(Row-encoded)和批量编码(Bulk-encoded)格式。这两种不同的方式都有各自的构建器(builder),可以直接调用 FileSink 的静态方法:
⚫ 行编码: FileSink.forRowFormat(basePath,rowEncoder)。
⚫ 批量编码: FileSink.forBulkFormat(basePath,bulkWriterFactory)。
org.apache.flink
flink-connector-jdbc
3.1.0-1.17
CREATE TABLE `ws` (
`id` varchar(100) NOT NULL,
`ts` bigint(20) DEFAULT NULL,
`vc` int(11) DEFAULT NULL,
) ENGINE=InnoDB DEFAULT CHARSET=utf8 ;
public static void main(String[] args) throws Exception {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
env.setParallelism(2);
SingleOutputStreamOperator sensorDS = env
.fromElements(
new WaterSensor("sensor_1", 2L, 2),
new WaterSensor("sensor_2", 3L, 6),
new WaterSensor("sensor_2", 2L, 9),
new WaterSensor("sensor_2", 3L, 4),
new WaterSensor("sensor_2", 3L, 5),
new WaterSensor("sensor_3", 8L, 3),
new WaterSensor("sensor_3", 7L, 7),
new WaterSensor("sensor_3", 6L, 1)
);
/**
* TODO 写入mysql
* 1、只能用老的sink 写法: addsink
* 2、JDBCSink 的 4 个参数:
* 第一个参数: 执行的 sql,一般就是 insert into
* 第二个参数: 预编译 sql, 对占位符填充值
* * 第三个参数: 执行选项 ---》 攒批、重试
* * 第四个参数: 连接选项 ---》 url、用户名、密码
*
*/
SinkFunction jdbcSink = JdbcSink.sink(
"insert into ws values(?,?,?)",
new JdbcStatementBuilder() {
@Override
public void accept(PreparedStatement preparedStatement, WaterSensor waterSensor) throws SQLException {
//每收到一条 WaterSensor,如何去填充占位符
preparedStatement.setString(1,
waterSensor.getId());
preparedStatement.setLong(2,
waterSensor.getTs());
preparedStatement.setInt(3,
waterSensor.getVc());
}
},
JdbcExecutionOptions.builder()
.withMaxRetries(3) // 重试次数
.withBatchSize(100) // 批次的大小:条数
.withBatchIntervalMs(3000) // 批次的时间
.build(),
new JdbcConnectionOptions.JdbcConnectionOptionsBuilder()
.withUrl("jdbc:mysql://localhost:3306/firstscheam?serverTimezone=Asia/Shanghai&useUnicode=true&characterEncoding=UTF-8")
.withUsername("root")
.withPassword("123456")
.withConnectionCheckTimeoutSeconds(60) // 重试的超时时间
.build()
);
sensorDS.addSink(jdbcSink);
env.execute();
}
最好用提供好的
如果我们想将数据存储到我们自己的存储设备中,而Flink 并没有提供可以直接使用的连接器,就只能自定义 Sink 进行输出了。
stream.addSink(new MySinkFunction
());
在实现 SinkFunction 的时候,需要重写的一个关键方法 invoke(),在这个方法中我们就可以实现将流里的数据发送出去的逻辑。
这种方式比较通用,对于任何外部存储系统都有效;不过自定义 Sink 想要实现状态一致性并不容易,所以一般只在没有其它选择时使用。实际项目中用到的外部连接器Flink 官方基本都已实现,而且在不断地扩充,因此自定义的场景并不常见。