我们先从wordcount程序说起
下面是一段使用 Flink 实现的 WordCount 代码
import org.apache.flink.streaming.api.scala._
object WordCount {
def main(args: Array[String]): Unit = {
val env =
StreamExecutionEnvironment.getExecutionEnvironment
val text = env.readTextFile("d://1.txt")
val counts : DataStream[(String,Int)] = text
.flatMap(_.toLowerCase.split(" "))
.filter(_.nonEmpty)
.map((_,1))
.keyBy(0)
.sum(1)
counts.print()
env.execute("WordCount")
}
}
整个 flink 程序一共有五步,分别是:创建 Flink 执行环境、创建或加载数据、对数据集进行转换操作、指定计算结果输出位置、调用execute方法触发执行。
下面依次来讲这五个步骤(分两篇文章讲完)
(1)Execution Environment
运行 Flink 程序第一步就是要获取相应的执行环境,决定程序在什么地方执行(本地或者集群上),同时不同的运行环境决定了应用的类型,批量处理作业(ExecutionEnvironment)还是流式处理作业(StreamExecutionEnvionment)。
自动选择环境,如果是在本地执行,则创建本地执行环境,如果是在集群执行,则创建集群执行环境
StreamExecutionEnvironment.getExecutionEnvironment
指定并行度,并创建本地执行环境
StreamExecutionEnvironment.createLocalEnvironment(5)
批处理也是一样:
自动选择环境
ExecutionEnvironment.getExecutionEnvironment
指定并行度,创建本地环境
ExecutionEnvironment.createLocalEnvironment(5)
(2)初始化数据
Flink 提供了不同的数据接口完成数据的初始化,将数据转换为 DataStream 或 DataSet 数据集。
如下的代码,把本地的文本文件读取为 DataStream
val text:DataStream[String] = env.readTextFile("d://1.txt")
Flink 提供了多种数据连接器,用来读取外部数据源数据,比如 kafka,es,hdfs 等等,后续会重点讲。
(3)执行转换操作
数据转换,就是把从输入数据得到的 DataStream 转换成其他数据。
Flink 内置了很多的算子,比如 map、flatMap、filter、keyBy等,我们只需要定义每个算子的逻辑即可。
比如我们的wordcount的代码:
flatMap算子,输入是数组,输出是元素。就是把数组压扁成一个个元素。这里先把每行变成小写,然后按空格切分,输入是一行数据,输出是多个切分后的 单词
.flatMap(_.toLowerCase.split(" "))
filter 过滤算子,留下满足条件的。这里过滤掉空的单词
.filter (_.nonEmpty)
map 算子,一对一转换,输入是一个单词,输出是一个元组(单词,1)
.map((_,1))
按照指定 key 对数据重分区
.keyBy(0)
执行求和操作
.sum(1)
这里,我们只需要传入相应的 Lambda 表达式,就能完成 Function 的定义
特殊情况下,用户也可以通过实行 Function 接口来完成定义数据处理逻辑。然后将定义好的 Function 应用在对应的算子中即可
(1)通过创建 Class 实行 Function 接口
val counts : DataStream[(String,Int)] = text.map(new MyMapFunction)
class MyMapFunction extends MapFunction[String,String] {
override def map(value: String): String = value.toUpperCase
}
如下我们自己定义了转换大写的 Function,调用 map 的时候,实例化一个传进去即可
(2)创建匿名类实现 Function 接口
val counts : DataStream[(String,Int)] = text.map(
new MapFunction[String,String] {
override def map(value: String): String = value.toUpperCase()
})
这里我们没有创建类,而是 new 了一个接口,实现了接口实现的方法
(3)通过实现 RichFunction 接口
RichFunction 接口有
open 方法
close方法
getRuntimeContext 方法
getIterationRuntimeContext 方法
setRuntimeContext方法