您的位置:首页 > 运维架构

spark streaming监控HDFS文件目录

2018-03-16 16:59 671 查看
集群环境:CDH5.8.0 / spark1.6.0 / scala2.10.4

基于Scala的基本使用方式如下:
package com.egridcloud.sparkstreaming

import org.apache.hadoop.conf.Configuration
import org.apache.hadoop.fs.Path
import org.apache.hadoop.io.LongWritable
import org.apache.hadoop.io.Text
import org.apache.hadoop.mapreduce.lib.input.TextInputFormat
import org.apache.spark.SparkConf
import org.apache.spark.serializer.KryoSerializer
import org.apache.spark.streaming.dstream.{DStream, InputDStream}
import org.apache.spark.streaming.{ Durations, StreamingContext}
/**
* Created by LHX on 2018/3/7 下午 8:06.
* 监控文件夹,实现单词统计,结果保存到HDFS
*/
object SparkStreamingFile {
def main(args: Array[String]): Unit = {
val classes: Array[Class[_]] = Array[Class[_]](classOf[LongWritable], classOf[Text])
val conf = new SparkConf().setAppName("sparkstreamingfile")//.setMaster("local[2]")
conf.set("spark.streaming.fileStream.minRememberDuration", "2592000s")
conf.set("spark.serialize", classOf[KryoSerializer].getName())
conf.registerKryoClasses(classes)
// 设置批次间隔时间
val streamingContext = new StreamingContext(conf, Durations.seconds(30))
// val inputPath = "C:/tmp/sparkstreamingfile"
val inputPath = args(0)
// val outputPath = "C:/tmp/sparkstreamingfile_save/"
val outputPath=args(1)
val hadoopConf = new Configuration()
val fileStream: InputDStream[(LongWritable, Text)] = streamingContext.fileStream[LongWritable,Text,TextInputFormat](inputPath, (path: Path) => {println(path.getName);path.getName.endsWith(".csv")}, false, hadoopConf)
//遍历每一行,用“,”分割
val flatMap: DStream[String] = fileStream.flatMap(_._2.toString.split(","))
//将每个单词标记 为1
val mapToPair: DStream[(String, Int)] = flatMap.map((_,1))
//将相同单词标记 累加
val reducerByKey: DStream[(String, Int)] = mapToPair.reduceByKey(_ + _)
reducerByKey.foreachRDD((a,b)=> println(s"count time:${b},${a.collect().toList}"))
//结果输出到HDFS
// reducerByKey.saveAsTextFiles(outputPath, "suffix")
reducerByKey.saveAsTextFiles(outputPath)

//是否触发job取决于设置的Duration时间间隔
streamingContext.start()
//等待程序结束
streamingContext.awaitTermination()
}
}打包上传集群,指定输入输出路径运行,往输入目录添加文件即可。
完整项目参考:GitHub
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐