updateStateByKey案例(Scala版本)

来源:互联网 发布:廉租住房软件 编辑:程序博客网 时间:2024/06/18 01:57
package SparkStreaming

import org.apache.spark.SparkConf
import org.apache.spark.streaming.{Seconds, StreamingContext}

/**
  * Created by tg on 11/4/16.
  */
object updateStateByKeyPro {
  def main(args: Array[String]): Unit = {
    val conf=new SparkConf().setAppName("updateStateByKeyPro")
             .setMaster("local[2]")
    val ssc=new StreamingContext(conf,Seconds(10))


    //开启checkpoint
    ssc.checkpoint("hdfs://tgmaster:9000/in/ch")


    //连接nc(netcat)服务,接收数据源,产生Dtream 对象
    val lines=ssc.socketTextStream("localhost",9999)
    
    //分隔单词,并将分隔后的每个单词出现次数记录为1
    val pairs=lines.flatMap(_.split(" "))
                .map(word=>(word,1))
    //调用updateStateByKey算子,统计单词在全局中出现的次数
    val result=pairs.updateStateByKey((values:Seq[Int],state:Option[Int])=>{
      //创建一个变量,用于记录单词出现次数
      var newValue=state.getOrElse(0) //getOrElse相当于if....else.....
      for(value <- values){
        newValue +=value //将单词出现次数累计相加
      }
      Option(newValue)
    })
    //直接输出结果
    result.print()


    ssc.start() //开启实时计算
    ssc.awaitTermination() //等待应用停止


  }


}
0 0
原创粉丝点击