Scala模式匹配

Scala模式匹配

模式匹配是Scala中非常有特色，非常强大的一种功能。模式匹配，其实类似于Java中的swich case语法，即对一个值进行条件判断，然后针对不同的条件，进行不同的处理。

但是Scala的模式匹配的功能比Java的swich case语法的功能要强大地多，Java的swich case语法只能对值进行匹配。但是Scala的模式匹配除了可以对值进行匹配之外，还可以对类型进行匹配、对Array和List的元素情况进行匹配、对case class进行匹配、甚至对有值或没值（Option）进行匹配。

而且对于Spark来说，Scala的模式匹配功能也是极其重要的，在spark源码中大量地使用了模式匹配功能。因此为了更好地编写Scala程序，并且更加通畅地看懂Spark的源码，学好模式匹配都是非常重要的。

1、模式匹配的基础语法

1.1 对变量的值进行模式匹配

Scala是没有Java中的switch case语法的，相对应的，Scala提供了更加强大的match case语法，即模式匹配，类替代switch case，match case也被称为模式匹配

Scala的match case与Java的switch case最大的不同点在于，Java的switch case仅能匹配变量的值，比1、2、3等；而Scala的match case可以匹配各种情况，比如变量的类型、集合的元素、有值或无值

match case的语法如下：变量 match { case 值 => 代码 }。如果值为下划线，则代表了不满足以上所有情况下的默认情况如何处理。此外，match case中，只要一个case分支满足并处理了，就不会继续判断下一个case分支了。（与Java不同，java的switch case需要用break阻止）

match case语法最基本的应用，就是对变量的值进行模式匹配
案例：

def method1(x:Int):String={
  if (x==1){
    "one"
  }else if (x==2){
    "two"
  }else{
    "many"
  }
}

//模式匹配--简化代码
def match1(x:Int):String = x match {
  case 1 =>"one"
  case 2 =>"two"
  case _ =>"many"
}

def main(args: Array[String]): Unit = {
      println(match1(2))
}

1.2 模式匹配中使用if守卫

Scala的模式匹配语法，有一个特点在于，可以在case后的条件判断中，不仅仅只是提供一个值，而是可以在值后面再加一个if守卫，进行双重过滤

def match2(name:String,score:String)=score match {
  case "A" => println("excellent")
  case "B" => println("good")
  case "C" => println("just so so")
  case _ if name == "qian" => println(name + " you are good boy,come on!") //if守卫
  case _ => println("you need work harder")
}
match2("qian","D")

1.3 在模式匹配中进行变量赋值

Scala的模式匹配语法，有一个特点在于，可以将模式匹配的默认情况，下划线’_'替换为一个变量名，此时模式匹配语法就会将要匹配的值赋值给这个变量，从而可以在后面的处理语句中使用要匹配的值

为什么有这种语法？？思考一下。因为只要使用用case匹配到的值，是不是我们就知道这个只啦！！在这个case的处理语句中，是不是就直接可以使用写程序时就已知的值！

但是对于下划线_这种情况，所有不满足前面的case的值，都会进入_这种默认情况进行处理，此时如果我们在处理语句中需要拿到具体的值进行处理呢？那就需要使用这种在模式匹配中进行变量赋值的语法！！

def match3(name:String,score:String)=score match {
  case "A" => println("excellent")
  case "B" => println("good")
  case "C" => println("just so so")
  case _ if name == "qian" => println(name + " you are good boy,come on!") //if守卫
  case _score => println("you need work harder,your score only："+_score) //变量赋值
}
match3("zs","E")

2、对类型进行模式匹配

Scala的模式匹配一个强大之处就在于，可以直接匹配类型，而不是值！！！这点是java的switch case绝对做不到的。

理论知识：对类型如何进行匹配？其他语法与匹配值其实是一样的，但是匹配类型的话，就是要用“case 变量: 类型 => 代码”这种语法，而不是匹配值的“case 值 => 代码”这种语法。

接下来我们来看一个不同数据类型的模式匹配案例：

def match2(x:Any):Unit = x match {
  case 1 => println("输入的值为1")
  case i:Int => println("输入的值类型为Int类型，值为："+i)
  case s:String => println("输入的值类型为String类型，值为："+s)
  case _ => println("我不知道是啥")
}
def main(args: Array[String]): Unit = {
        match2(17)
        match2("abc")
}

3、对Array和List的元素进行模式匹配

对Array进行模式匹配，分别可以匹配带有指定元素的数组、带有指定个数元素的数组、以某元素打头的数组

对List进行模式匹配，与Array类似，但是需要使用List特有的::操作符

案例：对朋友打招呼

```scala
//对Array进行模式匹配
def match5(arr:Array[String])=arr match {
  case Array("KB09") => println("Hello KB09") //匹配一个元素
  case Array(girl1,girl2,girl3) => println("Hello "+girl1+"&"+girl2+"&"+girl3)//匹配三个元素
  case Array("KB09", _*) => println("Hello KB09*")//匹配以KB09开头，三个元素的话会被上面匹配,scala匹配机制是匹配到就停止
  case _ => println("Who are you?")
}
var arr1=Array("KB09","KB07","KB05","KB02") //Hello KB09*
var arr2=Array("KB09")  //Hello KB09
var arr3=Array("KB09","KB07","KB05") //Hello KB09&KB07&KB05
var arr4=Array("KB02","KB09","KB07","KB05") //Who are you?
match5(arr1)
match5(arr2)
match5(arr3)
match5(arr4)

//对List进行模式匹配
 def greeting(list: List[String]) {      
   list match {        
 	 case "Leo" :: Nil => println("Hi, Leo!")        
 	 case girl1 :: girl2 :: girl3 :: Nil => println("Hi, girls, nice to meet you. " + girl1 + " and " + girl2 + " and " + girl3)        
 	 case "Leo" :: tail => println("Hi, Leo, please introduce your friends to me.")        
 	 case _ => println("hey, who are you?")      
   }    
 }    
 greeting(List("Leo","jack","poly","herry"))  

4、Option与模式匹配

Scala有一种特殊的类型，叫做Option。Option有两种值，一种是Some，表示有值，一种是None，表示没有值。

Option通常会用于模式匹配中，用于判断某个变量是有值还是没有值，这比null来的更加简洁明了

Option的用法必须掌握，因为Spark源码中大量地使用了Option，比如Some(a)、None这种语法，因此必须看得懂Option模式匹配，才能够读懂spark源码。

案例：成绩查询

val grades = Map("张三" -> "A", "李四" -> "B", "王五" -> "C")
def getGrade(name:String)={
  val grade = grades.get(name)
  grade match {
    case Some(grade) => println(name+"的成绩为：" +grade)
    case None => println("没有"+name+"对应的成绩")
  }
}
getGrade("李四")
getGrade("赵六")

5、case class与模式匹配

Scala中提供了一种特殊的类，用case class进行声明，中文也可以称作样例类。case class其实有点类似于Java中的JavaBean的概念。即只定义field，并且由Scala编译时自动提供getter和setter方法，但是没有method。

case class的主构造函数接收的参数通常不需要使用var或val修饰，Scala自动就会使用val修饰（但是如果你自己使用var修饰，那么还是会按照var来）

Scala自动为case class定义了伴生对象，也就是object，并且定义了apply()方法，该方法接收主构造函数中相同的参数，并返回case class对象

案例：

//定义样例类
case class Teacher(name:String,age:Int)

def match3(teacher:Teacher):Unit = teacher match {
  case Teacher("gree",18) => println("hello gree")
  case Teacher("wenwen",21) => println("hello wenwen")
  case x:Teacher =>{
    println("hello "+x.name)
  }
}

def main(args: Array[String]): Unit = {
    val t=Teacher("gree",19)
    println(t)
    val t1=Teacher("gree",18)
    if (t.equals(t1)) {
      println("同一个人")
    }else{
      println("不是同一个人")
    }

参考博客：https://www.cnblogs.com/linxizhifeng/p/9269756.html