1. 程式人生 > >Scala的模式匹配

Scala的模式匹配

1.典型的模式匹配場景

(1)匹配字串

object Test01 {
  def main(args: Array[String]): Unit = {
    val arr=Array("aa","bb","cc")
    //隨機獲取陣列的任意元素
    val index=Random.nextInt(3)
    val value=arr(index)
    //模式匹配
    value match{
      case "aa" => println("0")
      case "bb" => println("1")
      case "cc" => println("2")
      case _ => println("null")  //表示所有的都沒有匹配到執行這裡
    }
  }
}

(2)匹配型別

object Test01 {
  def main(args: Array[String]): Unit = {
    val arr=Array("aa",18,4.4,true)
    //隨機獲取陣列的任意元素
    val index=Random.nextInt(arr.length)
    val value=arr(index)
    //模式匹配
    value match{
      case x:Int if(x>3) => println("Int")
      case y:String  => println("String")
      case z:Double => println("Double")
      case flag:Boolean => println("Boolean")
      case _ => println("null")  //表示所有的都沒有匹配到執行這裡
    }
  }
}

(3)匹配陣列、元組、集合

object Test01 {
  def main(args: Array[String]): Unit = {
    //匹配陣列
    val arr3=Array(0,1,2,3,4,5)
    arr3 match{
      case Array(0,x,y) => println(x+"--"+y) //匹配以0開頭,後面兩個元素的陣列
      case Array(0) => println("only 0")     //匹配只有一個元素的陣列
      case Array(0,_*) => println("many") //匹配以0開頭的陣列,_表示任意元素,*表示一到多個元素
    }

    //匹配序列
    val lst1=List(3,1,-1)
    lst1 match{
      case 0::Nil => println("only 0")    //匹配只有一個元素0的序列
      case x::y::Nil =>println(x+"--"+y) //匹配有兩個元素的序列
      case 0::tail => println("0 is head ") //匹配以0開頭的序列
      case _ => println("nothing ")
    }

    //匹配元組
    val tuple=(2,3,4)
    tuple match {
      case (1,x,y) => println("1 is head ") //匹配以1開頭的 3個元素的tuple
      case (_,x,y) => println("many")       //匹配以任意開頭的 3個元素的tuple
      case _ => println("nothing ")
    }
  }
}

(4)匹配異常

object Test01 {
  def main(args: Array[String]): Unit = {
   try{
      println(3/0)
   }catch {
     case e:IOException  => println("IOException")
     case e:Exception  => println("Exception")
   }finally{
     println("程式結束!")
   }
  }
}

2.樣例類

 在scala中樣例類是一種特殊的類,可用於模式匹配。
特點
    - 構造器中的引數如果不被宣告為 var 的話,它預設的話是 val 型別的,但一般不推薦將構造器中的引數宣告為 var
    - 自動建立伴生物件,同時在裡面給我們實現子 apply 方法,使得我們在使用的時候可以 不直接顯示地 new 物件
    - 伴生物件中同樣會幫我們實現 unapply 方法,從而可以將 case class 應用於模式匹配 apply 方法接受引數返回物件,unapply 方法接收物件返回引數
    - 實現自己的 toString、hashCode、copy、equals 方法
    - case class 主建構函式裡面沒有修飾符,預設的是 val
例1:

object Test01 {
  def main(args: Array[String]): Unit = {
    val arr=Array(SubmitTask("1001","zs"),
      HeartBeat(10000),CheckTimeOutTask)
    arr(Random.nextInt(arr.length)) match {
      case SubmitTask(id,name) => println(id,name)  //這裡能將樣例類中的引數提取出來,是是因為有unapply方法
      case HeartBeat(time) => println(time)
      case CheckTimeOutTask => println("CheckTimeOutTask")
    }
  }
}

//多例樣例類
case class SubmitTask(id: String, name: String)
//多例樣例類
case class HeartBeat(time: Long)
//單例樣例類
case object CheckTimeOutTask

例2:
 在 Scala 中 Option 型別樣例類用來表示可能存在或也可能不存在的值(Option 的子類有 Some 和 None)。Some 包裝了某個值,None 表示沒有值。

object Test01 {
  def main(args: Array[String]): Unit = {
    val map=Map("name"->"zs","age"->18,"address"->"beijing")
    val value=map.get("name")
    val result=value match {
      case Some(i) => i
      case None => 0
    }
    println(result)

    //上述方法可以用map的方法替代
    map.getOrElse("name",0)
  }
}

3.偏函式

object Test01 {
  def main(args: Array[String]): Unit = {
    //普通方式的模式匹配
    val value = "aa"
    value match {
      case "aa" => println("0")
      case "bb" => println("1")
      case "aa" => println("0")
    }

    /**
      * 偏函式:
      * 語法:func1:PartialFunction[input,output]
      *   input:表示輸入型別
      *   output:輸出的型別
      *
      */
    def func1:PartialFunction[String,Int]={
      case "zs" =>18
      case "ls"=>18
      case "ww"=>20
      case _ =>100
    }
  }
}