spark 学习笔记

最近使用spark简单的处理一些实际中的场景，感觉简单实用，就记录下来了。

---

场景运用

部门用户业绩表（1000w测试数据）

用户、部门、业绩

数据加载：

val conf = new SparkConf().setAppName("spark函数").setMaster("local")val context = new SparkContext(conf)var data = context.textFile("data.txt")

场景1：求每个部门的总业绩,并从大到小排序

data.map(_.split(",")).map(p => (p(1), p(2).toInt)).reduceByKey(_ + _).sortBy(_._2, false).foreach(println) //取出部门和业绩字段，对其进行按部门汇总，并进行排序即可。

场景2：求每个部门中业绩最高的人

data.map(_.split(",")).map(p => (p(0), p(1), p(2).toInt)).groupBy(_._2).map(p => (p._2.map(p => (p._1,p._3)).toArray.maxBy(_._2),p._1)).sortBy(p => p._1._2, false).map { case ((user, gz), dept) => user + "," + dept + "," + gz }.foreach(println)//step1:先对部门进行分组//step2:找出部门中最高业绩和人//step3:排序，并去除格式

加载用户订单表
用户、月份、订单

 var data1 = context.textFile("/userdt.txt")

场景3：求每季度的订单金额

data1.map(_.split(",")).map(p => {    if (p(1).toInt <= 3) {      (1, p(2).toInt)    } else if (p(1).toInt > 3 && p(1).toInt <= 6) {      (2, p(2).toInt)    } else if (p(1).toInt > 6 && p(1).toInt <= 9) {      (3, p(2).toInt)    } else {      (4, p(2).toInt)    }  }).reduceByKey(_ + _).foreach(println)  //这里用if else 对每季度进行了分组，应该可以用match匹配的，个人对match不熟。  //分完组，按组进行汇总即可

场景4：求用户每季度订单金额

data1.map(_.split(",")).map(p => {    if (p(1).toInt <= 3) {      (p(0), 1, p(2).toInt)    } else if (p(1).toInt > 3 && p(1).toInt <= 6) {      (p(0), 2, p(2).toInt)    } else if (p(1).toInt > 6 && p(1).toInt <= 9) {      (p(0), 3, p(2).toInt)    } else {      (p(0), 4, p(2).toInt)    }  }).groupBy(p => (p._1, p._2)).map(p => (p._1, p._2.map(p => p._3).reduce(_ + _)))    .map(p => (p._1._1, p._1._2, p._2)).sortBy(p => (p._1, p._2)).foreach(println) //这里只有月份，有年也是一样，先用filter过滤出那一年，再按季度进行汇总 //根据用户和季度进行分组，再用户和季度的组对金额进行汇总 //格式化数据，并按用户和季度排序

场景5：纬度关联，求全国各省份2015年各季度销售业绩

数据表
城市信息表（城市ID,城市，省份）

订单信息表（用户，年份，月份，城市ID，业绩）

  var cy = context.textFile("city.txt").map(_.split(",")).map(p => (p(0).toInt, p(2))).collect().toMap   //把城市ID和省份转化成map形式，方便获取  var dd = context.textFile("dingdan.txt").map{p=>    var sp=p.split(",")    var quarter=sp(2).toInt    if(quarter<=3){      quarter=1    }else if(quarter>3 && quarter<=6){      quarter=2    }else if(quarter>6 && quarter<=9){      quarter=3    }else{      quarter=4    }    ( sp(1),quarter, sp(3), sp(4))  }.filter(_._1.toInt.equals(2015))  //对月份按季度进行分组，并过滤出2015的订单 var data2 = dd.map(p => (p._1, p._2, cy.getOrElse(p._3.toInt,"其他"),p._4)).groupBy(p=>(p._1,p._2,p._3)) .map(p=>(p._1,p._2.map(_._4.toInt).reduce(_+_))).sortBy(p=>(p._1._3,p._1._1,p._1._2)).map{case ((year,quarter,area),num)=>area+","+year+","+quarter+","+num}.foreach(println)//step1:两表关联，订单的城市ID通过城市纬度表转换成省份//step2:对年份、季度、省份进行联合分组//step3:对订单金额进行合计，并按省份、年份、季度进行排序，格式化后输出

通过sqlContext（隐式转换的方式）来处理场景，这里只举一个例子

场景：求每个部门中业绩最高的人

val sqlContext=new SQLContext(context)  import  sqlContext.implicits._  //隐式转换case class  temptable(id:String,dept:String,sar:Int)var sql=data.map(_.split(",")).map(p=>temptable(p(0),p(1),p(2).toInt)).toDF()  sql.registerTempTable("deptinfo")val jh= sqlContext.sql("select a.id,a.sar,a.dept from deptinfo a INNER JOIN (select max(sar) aaa,dept from deptinfo group by dept)b on a.sar=b.aaa and a.dept=b.dept    order by a.sar ")jh.foreach(println)//把内容转换成临时表，并通过sql直接编写