Spark Streaming Receiver学习

本文深入解析了Spark Streaming中Receiver的工作原理,包括如何监听端口、处理数据流和与BlockManager交互的过程。同时对比了Receiver-based与KafkaReceiver-based在数据处理和数据丢失预防上的差异,特别强调了KafkaReceiver-based能够提供Exactly-Once语义的优势。

Receiver工作机制

以org.apache.spark.streaming.dstream.SocketReceiver为例。

一个数据流有一个receiver,在executor上启动。

Receiver监听端口,拉取数据,调用ReceiverSupervisor和BlockGenerator将数据加入BlockGenerator的缓存。在加入缓存前,调用限流器(RateLimiter),以阻塞当前线程,直到限流完成。这里使用GuavaRateLimiter。Receiver和ReceiverSupervisor是包含关系;SocketReceiver不支持WriteAheadLog.

BlockGenerator的守护线程定时(spark.streaming.blockInterval) 将缓存数据加入data block并发送到blockManager,并将blockId,数据条数等通过RPC告知Driver。

批次时间到,按照Data Locality原则优先在含有数据的executor上执行开始数据处理。

 

Kafka Receiver-based VS DirectStream

Receiver-based 默认情况下可能丢数据,需要配置WriteAheadLog,保证At-Least-Once语义。由于Receiver commitOffset和数据处理逻辑不同步,不能100%保证Eactly-Once语义。

DirectStream保证kafka消费Exactly-Once语义。通过在KafkaRDD里拉取处理数据和自己管理offset。

### 创建使用自定义 Receiver 的过程 为了在 Spark Streaming 应用程序中创建并使用自定义接收器,需遵循特定的方法来构建 `CustomReceiver` 类,并将其集成到流处理框架内。 #### 定义自定义接收器类 首先,需要继承 `Receiver` 抽象类,并重写其方法以实现具体的功能逻辑。下面是一个简单的例子: ```scala import org.apache.spark.streaming.receiver.Receiver import java.net.{ServerSocket, Socket} class CustomReceiver(host: String, port: Int) extends Receiver[String](StorageLevel.MEMORY_AND_DISK_2) { def onStart() { // 启动线程执行 receive 方法 new Thread("Socket Receiver") { override def run() { receive() } }.start() } def onStop() { // 停止接收数据的操作 } /** Create a socket connection and receive data until receiver is stopped */ private def receive() { var socket: Socket = null try { // 连接到服务器套接字 val serverSocket = new ServerSocket(port) LogInfo(s"Connecting to $host:$port") socket = serverSocket.accept() LogInfo(s"Connected to $host:$port") // 将收到的数据发送给存储引擎 val input = new BufferedReader(new InputStreamReader(socket.getInputStream(), StandardCharsets.UTF_8)) var line: String = "" while (!isStopped && (line = input.readLine()) != null) { store(line) } LogInfo("Stopped receiving") onStop() } catch { case e: Throwable => restart(s"Error connecting or reading from socket", e) } finally { if (socket != null) socket.close() } } } ``` 这段代码展示了如何建立一个基于 TCP 协议的自定义接收器[^1]。 #### 配置 StreamingContext 启动应用 一旦有了自定义接收器之后,在主函数里配置好 `StreamingContext` 并调用 `receiverStream()` 来注册这个新的接收器实例即可开始工作流程。 ```scala val conf = new SparkConf().setMaster("local[*]").setAppName("CustomReceiverApp") // 设置日志级别为 ERROR 减少控制台输出干扰 conf.set("spark.logConf","false").set("spark.driver.bindAddress","localhost") val ssc = new StreamingContext(conf, Seconds(1)) // 使用自定义接收器获取实时数据流 val customReceiverStream = ssc.receiverStream(new CustomReceiver("localhost", 9999)) // 对接收到的数据做进一步操作比如单词计数等... customReceiverStream.flatMap(_.split(" ")) .map((_, 1)) .reduceByKey(_ + _) .print() ssc.start() // 开始接受数据 ssc.awaitTermination() // 等待终止信号 ``` 此部分实现了通过 `StreamingContext` 初始化以及利用之前定义好的 `CustomReceiver` 获取来自指定主机与端口的数据流,并进行了基本的文字分割平铺映射转换成键值对形式以便后续聚合计算。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值