Spark项目练习(实现自定义排序)

在实际操作中,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

Author: allengao

发表评论

电子邮件地址不会被公开。 必填项已用*标注