spark-streaming中DS的創建方法(用於測試)

由Socket方式構建

package com.zhengkw.stu.day01

import org.apache.spark.streaming.dstream.{DStream, ReceiverInputDStream}
import org.apache.spark.streaming.{Seconds, StreamingContext}
import org.apache.spark.{SparkConf, SparkContext}

/**
 * @ClassName:SoketWordCount
 * @author: zhengkw
 * @description:
 * @date: 20/05/16上午 10:40
 * @version:1.0
 * @since: jdk 1.8 scala 2.11.8
 */
object SoketWordCount {
  def main(args: Array[String]): Unit = {
    val conf: SparkConf = new SparkConf().setAppName("SoketWordCount").setMaster("local[2]")
    //獲取上下文
    val ssc: StreamingContext = new StreamingContext(conf, Seconds(3))

    val sourceStream: ReceiverInputDStream[String] = ssc.socketTextStream("hadoop102", 9999)
    // 2. 從數據源讀取數據, 得到 DStream  (RDD, DataSet, DataFrame)
    val resultStream: DStream[(String, Int)] = sourceStream.flatMap(_.split(" ")).map((_, 1)).reduceByKey(_ + _)
    //打印前N個
    resultStream.print(100)
    // 4. 啓動流
    ssc.start()
    // 5. 阻止程序退出
    ssc.awaitTermination()

  }
}

由RDD[T]隊列構建

package com.zhengkw.stu.day01

import org.apache.spark.SparkConf
import org.apache.spark.rdd.RDD
import org.apache.spark.streaming.dstream.{DStream, InputDStream}
import org.apache.spark.streaming.{Seconds, StreamingContext}

import scala.collection.mutable

/**
 * @ClassName:RDDQueue
 * @author: zhengkw
 * @description: rdd隊列得到DS
 * @date: 20/05/16上午 11:36
 * @version:1.0
 * @since: jdk 1.8 scala 2.11.8
 */
object RDDQueue {
  def main(args: Array[String]): Unit = {
    val conf: SparkConf = new SparkConf().setMaster("local[2]").setAppName("RDDQueue")
    val ssc: StreamingContext = new StreamingContext(conf, Seconds(3))
    //創建一個隊列 
    val queue: mutable.Queue[RDD[Int]] = mutable.Queue[RDD[Int]]()
    //獲取DS
    //oneAtATime Whether only one RDD should be consumed from the queue in every interval 這裏即一次取多個!
    val stream: InputDStream[Int] = ssc.queueStream(queue, oneAtATime = false)
    //聚合DS
    val value: DStream[Int] = stream.reduce(_ + _)
    //不傳參數打印10條!
    value.print
    ssc.start()

    //循環添加隊列
    while (true) {
      println(queue.size)
      val rdd = ssc.sparkContext.parallelize(1 to 100)
      queue.enqueue(rdd)
      // Thread.sleep(2000)
    }
    ssc.awaitTermination()
  }
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章