在实际操作中,sortBy算子往往不能满足数据多种排序的需求,这就需要我们使用自定义排序来实现,以下是实现简单的自定义排序的两种方法,起到抛砖引玉的作用。
第一种方法:(Ordered:自定义一个函数)
package cn.allengao.IpSearch
import org.apache.spark.{SparkConf, SparkContext}
/**
* class_name:
* package:
* describe: 自定义排序(首先比较攻击值,值高的排前面,如果攻击值相等,比较防守值,攻击值相同时,防守值高的排前面)
* creat_user: Allen Gao
* creat_date: 2018/2/2
* creat_time: 11:01
**/
object CustomSort {
def main(args: Array[String]): Unit = {
val conf = new SparkConf().setAppName("CustomSort").setMaster("local[*]")
val sc = new SparkContext(conf)
//球员信息,姓名name,攻击值Attack,防守值defense。
// (首先比较攻击值,值高的排前面,如果攻击值相等,比较防守值,攻击值相同时,防守值高的排前面)
val playerInfo = sc.parallelize(Array(("Kobe", 98, 95),("O'Neal" , 98, 96),("Harden",95 ,95)))
val res = playerInfo.sortBy(x => Player(x._2, x._3),false)
println(res.collect().toBuffer)
//执行结果:ArrayBuffer((O'Neal,98,96), (Kobe,98,95), (Harden,95,95))
sc.stop()
}
}
//第一种方式使用case class 模式匹配(Ordered 需要一个函数)
case class Player(val attack: Int, val defense: Int) extends Ordered[Player] with Serializable {
override def compare(that: Player) :Int = {
if (this.attack == that.attack) {
this.defense - that.defense
} else {
this.attack - that.attack
}
}
}
第二种方法:(Ordering:implicit隐式转换值)
package cn.allengao.IpSearch
import org.apache.spark.{SparkConf, SparkContext}
/**
* class_name:
* package:
* describe: 自定义排序(首先比较攻击值,值高的排前面,如果攻击值相等,比较防守值,攻击值相同时,防守值高的排前面)
* creat_user: Allen Gao
* creat_date: 2018/2/2
* creat_time: 11:01
**/
object MySort {
implicit val playerOrdering = new Ordering[Player_1] {
// implicit object PlayerOrdering extends Ordering[Player] {
override def compare(x: Player_1, y: Player_1): Int = {
if (x.attack > y.attack) 1
else if (x.attack == y.attack) {
if (x.defense > y.defense) 1 else -1
} else -1
}
}
}
object CustomSort_1 {
def main(args: Array[String]): Unit = {
val conf = new SparkConf().setAppName("CustomSort").setMaster("local[*]")
val sc = new SparkContext(conf)
//球员信息,姓名name,攻击值Attack,防守值defense。
// (首先比较攻击值,值高的排前面,如果攻击值相等,比较防守值,攻击值相同时,防守值高的排前面)
val playerInfo = sc.parallelize(Array(("Kobe", 98, 95), ("O'Neal", 98, 96), ("Harden", 95, 95)))
import MySort._
val res = playerInfo.sortBy(x => Player_1(x._2, x._3), false)
println(res.collect().toBuffer)
//执行结果:ArrayBuffer((O'Neal,98,96), (Kobe,98,95), (Harden,95,95))
sc.stop()
}
}
//第二种方式:使用隐式转换(Ordering需要一个隐式转换值)
case class Player_1(attack: Int, defense: Int) extends Serializable