1. 程式人生 > >Spark實現二次排序

Spark實現二次排序

1、HDFS檔案說明 

檔案為普通的文字檔案,無壓縮,\001分割,共3列,一次為province_id,city_id,city_uv

需要按照province_id升序,city_uv降序操作


2、程式碼

var data = sc.textFile("/home/hdfs/test_second")

var rdd1=data.map(_.split("\001")).map(fields=>(fields(0).toInt,s"${fields(1)},${fields(2)}"))
var rdd2 = rdd1.groupByKey.sortByKey()
var rdd3 = rdd2.flatMap(x=>{
var r = x._2.toList.sortWith(_.split(",")(1).toInt>_.split(",")(1).toInt)
for(e <- r) yield s"${x._1},${e}"

}).collect.foreach(println)

結果如下:---------------------------------------------

如果想把province_id按照字串排序,需要加上,修改預設的數字排序為字串排序

implicit val sortIntegersByString = new Ordering[Int]{
override def compare(a: Int, b: Int) =

a.toString.compare(b.toString)}

結果


3、使用dataframe實現



var data = sc.textFile("/home/hdfs//test_second")
case class Element(province_id: Integer, city_id: Integer, uv: Integer)
import org.apache.spark.sql.hive.HiveContext
val hiveContext = new org.apache.spark.sql.hive.HiveContext(sc)
import hiveContext.implicits._
val df = data.map(_.split("\001")).map(x=>Element(x(0).toInt,x(1).toInt,x(2).toInt)).toDF()
df.registerTempTable("element")
hiveContext.sql("select * from element order by province_id desc ,uv desc ").collect().foreach(println)