spark保存数据到hbase_Spark读取Hbase的数据

val conf = HBaseConfiguration.create()

conf.addResource(new Path("/opt/cloudera/parcels/CDH-5.4.4-1.cdh5.4.4.p0.4/lib/hbase/conf/hbase-site.xml"))

conf.addResource(new Path("/opt/cloudera/parcels/CDH-5.4.4-1.cdh5.4.4.p0.4/lib/hadoop/etc/hadoop/core-site.xml"))

conf.set(TableInputFormat.INPUT_TABLE, "FLOW")

//添加过滤条件，年龄大于 18 岁

//val scan = new Scan()

//conf.set(TableInputFormat.SCAN, convertScanToString(scan))

/*

scan.setFilter(new SingleColumnValueFilter("basic".getBytes, "age".getBytes,

CompareOp.GREATER_OR_EQUAL, Bytes.toBytes(18)))

*/

val usersRDD = sc.newAPIHadoopRDD(conf, classOf[TableInputFormat],

classOf[org.apache.hadoop.hbase.io.ImmutableBytesWritable],

classOf[org.apache.hadoop.hbase.client.Result])

val data1 = usersRDD.count()

val sf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSSSS")

println("data length:" + data1)

var map = HashMap[String, HashMap[String, collection.mutable.ArrayBuffer[Double]]]()

usersRDD.collect().map {

case (_, result) =>

val key = Bytes.toInt(result.getRow)

println("Key:" + key)

val ip = Bytes.toString(result.getValue("F".getBytes, "SADDR".getBytes))

val port = Bytes.toString(result.getValue("F".getBytes, "SPORT".getBytes))

val startTimeLong = Bytes.toString(result.getValue("F".getBytes, "STIME".getBytes))

val endTimeLong = Bytes.toString(result.getValue("F".getBytes, "LTIME".getBytes))

val protocol = Bytes.toString(result.getValue("F".getBytes, "PROTO".getBytes))

val sumTime = Bytes.toString(result.getValue("F".getBytes, "DUR".getBytes))

val sum = Bytes.toString(result.getValue("F".getBytes, "DBYTES".getBytes)).toDouble

println("ip:" + ip + ",port:" + port + ",startTime:" + startTimeLong + ",endTime:" + endTimeLong + ",protocol:" + protocol + ",sum:" + sum)

//ip+port+udp，14:02 14:07 List

//ip+port+tcp，15:02 15:07 List

val startTimeDate = sf.parse(startTimeLong)

val endTimeLongDate = sf.parse(endTimeLong)

val startHours = startTimeDate.getHours

val startMinutes = startTimeDate.getMinutes

val endHours = endTimeLongDate.getHours

val endMinutes = endTimeLongDate.getMinutes

val key1 = ip + "_" + port + "_" + protocol

println("key1:" + key1)

val key2 = startHours + ":" + startMinutes + "_" + endHours + ":" + endMinutes

println("key2:" + key2)

val tmpMap = map.get(key1)

if (!tmpMap.isEmpty) {

println("--------------------map is not null:" + tmpMap.size + "--------------------")

val sumArray = tmpMap.get.get(key2)

if (!sumArray.isEmpty) {

sumArray.get += sum

}

} else {

println("--------------------map is null--------------------")

//如果当前Key不存在的话，是一个全新的Ip

val sumArray = collection.mutable.ArrayBuffer[Double]()

sumArray += sum

val secondMap = HashMap[String, collection.mutable.ArrayBuffer[Double]]()

secondMap += (key2 -> sumArray)

map += (key1 -> secondMap)

}

map

println("map size-----------------:" + map.size)

}

println("map size:" + map.size)

map.map(e => {

println("--------------------Statistics start --------------------")

val resultKey1 = e._1

val resultVal1 = e._2

println("resultKey1:" + resultKey1)

resultVal1.foreach(f => {

val resultKey2 = f._1

val resultVal2 = f._2

println("resultKey2:" + resultKey2)

println("-----------------resultVal2:" + resultVal2.length)

resultVal2.map(f=>{

println("------------------------f:"+f)

})

val dataArray = resultVal2.map(f => Vectors.dense(f))

val summary: MultivariateStatisticalSummary = Statistics.colStats(sc.parallelize(dataArray))

//

println("--------------------mean:" + summary.mean + " --------------------")

println("--------------------variance:" + summary.variance + " --------------------")

println("--------------------mean apply 0:" + summary.mean.toArray.apply(0) + " --------------------")

println("--------------------variance apply 0:" + summary.variance.apply(0) + " --------------------")

val upbase = summary.mean.toArray.apply(0) + 1.960 * Math.sqrt(summary.variance.apply(0))

val downbase = summary.mean.toArray.apply(0) - 1.960 * Math.sqrt(summary.variance.apply(0))

println("------------------- " + upbase + " ---------- " + downbase)

val df = new DecimalFormat(".##")

val upbaseString = df.format(upbase)

val downbaseString = df.format(downbase)

//resultMap.put(key, value)

val result3 = HashMap[Double, Double]()

//result3 +=(upbase -> downbase)

println("ip port:" + resultKey1 + ",time:" + resultKey2 + ",upbase:" + upbase + ",downbase:" + downbase)

})

})

println("--------------------baseLine end --------------------")

sc.stop()