使用Flink读取Kafka中的消息

本文将介绍如何通过Flink读取Kafka中Topic的数据。

和Spark一样,Flink内置提供了读/写Kafka Topic的Kafka连接器(Kafka Connectors)。Flink Kafka Consumer和Flink的Checkpint机制进行了整合,以此提供了exactly-once处理语义。为了实现这个语义,Flink不仅仅依赖于追踪Kafka的消费者group偏移量,而且将这些偏移量存储在其内部用于追踪。

和Spark一样,Flink和Kafka整合的相关API也没有打包进Flink包中,而是单独进行了打包;所以如果我们需要在Flink中使用到Kafka,需要将这个包引入到我们的pom.xml文件中。本文以Flink 1.0.0和Scala 2.10.x为例进行说明,我们将下面的依赖引入到pom.xml文件中:

<dependency>

<groupId>org.apache.flink</groupId>

<artifactId>flink-connector-kafka-0.8_2.10</artifactId>

<version>1.0.0</version>

</dependency>

Kafka Consumer

我们需要编写一个Kafka Consumer,通过Flink计算引擎从Kafka相应的Topic中读取数据。在Flink中,我们可以通过FlinkKafkaConsumer08来实现,这个类提供了读取一个或者多个Kafka Topic的机制。它的构造函数接收以下几个参数:

1、topic的名字,可以是String(用于读取一个Topic)List(用于读取多个Topic);

2、可以提供一个DeserializationSchema / KeyedDeserializationSchema用于反系列化Kafka中的字节数组;

3、Kafka consumer的一些配置信息,而且我们必须指定bootstrap.servers、zookeeper.connect(这个属性仅仅在Kafka 0.8中需要)和group.id属性。

好了,我们来使用FlinkKafkaConsumer08类吧,初始化如下:

val properties = new Properties();

properties.setProperty("bootstrap.servers", "www.iteblog.com:9092");

// only required for Kafka 0.8

properties.setProperty("zookeeper.connect", "www.iteblog.com:2181");

properties.setProperty("group.id", "iteblog");

val stream = env.addSource(new FlinkKafkaConsumer08[String]("iteblog",

new SimpleStringSchema(), properties))

stream.print()

上面的例子中使用到SimpleStringSchema来反系列化message,这个类是实现了DeserializationSchema接口,并重写了T deserialize(byte[] message)函数,DeserializationSchema接口仅提供了反系列化data的接口,所以如果我们需要反系列化key,我们需要使用KeyedDeserializationSchema的子类。KeyedDeserializationSchema接口提供了T deserialize(byte[] messageKey, byte[] message, String topic, int partition, long offset)方法,可以饭系列化kafka消息的data和key。

为了方便使用,Flink内部提供了一序列的schemas:TypeInformationSerializationSchema和TypeInformationKeyValueSerializationSchema,它可以根据Flink的TypeInformation信息来推断出需要选择的schemas。

kKafka Consumers和Fault Tolerance

如果我们启用了Flink的Checkpint机制,那么Flink Kafka Consumer将会从指定的Topic中消费消息,然后定期地将Kafka offsets信息、状态信息以及其他的操作信息进行Checkpint。所以,如果Flink作业出故障了,Flink将会从最新的Checkpint中恢复,并且从上一次偏移量开始读取Kafka中消费消息。

我们需要在执行环境下启用Flink Checkpint机制,如下:

val env = StreamExecutionEnvironment.getExecutionEnvironment()

env.enableCheckpointing(5000) // checkpoint every 5000 msecs

需要主要的是,Flink仅仅支持在拥有足够的处理slots情况下才能够从Checkpint恢复。Flink on YARN模式下支持动态地重启丢失的YARN containers。

如果我们没有启用Checkpint,那么Flink Kafka consumer将会定期地向Zookeeper commit偏移量。

完整的代码

package com.iteblog

import java.util.Properties

import org.apache.flink.runtime.state.filesystem.FsStateBackend

import org.apache.flink.streaming.api.scala.{StreamExecutionEnvironment, _}

import org.apache.flink.streaming.connectors.kafka.FlinkKafkaConsumer08

import org.apache.flink.streaming.util.serialization.SimpleStringSchema

/**

* Created by https://www.iteblog.com on 2016/5/3.

*/

object FlinkKafkaStreaming {

def main(args: Array[String]) {

val env = StreamExecutionEnvironment.getExecutionEnvironment

env.enableCheckpointing(5000)

val properties = new Properties()

properties.setProperty("bootstrap.servers", "www.iteblog.com:9092")

// only required for Kafka 0.8

properties.setProperty("zookeeper.connect", "www.iteblog.com:2181")

properties.setProperty("group.id", "iteblog")

val stream = env.addSource(new FlinkKafkaConsumer08[String]("iteblog",

new SimpleStringSchema(), properties))

stream.setParallelism(4).writeAsText("hdfs:///tmp/iteblog/data")

env.execute("IteblogFlinkKafkaStreaming")

}

}

运行上面的程序,将会在hdfs:///tmp/iteblog/目录下创建data文件,并将数据写入到data文件中。

相关推荐