Scala模式匹配

object MatchDemo {
  /**
    * 定义偏函数用PartialFunction来表示
    * PartialFunction[T1,T2]，T1表示传入参数的数据类型，T2表示返回值的数据类型
    * 偏函数体中的case只会执行一个，如果匹配到了就不会继续往下匹配
    */
  val func: PartialFunction[String, Int] = {
    case "a" => 1
    case "b" => 2
    case _ => -1
  }

  /**
    * 匹配字符串
    */
  def matchString(str: String) = {
    str match {
      case "a" => 1
      case "b" => 2
      case _ => -1
    }
  }

  /**
    * 匹配类型
    */
  def matchType(x: Any) = {
    x match {
      case x: String => println(x.getClass + " " + x)
      case x: Int if x > 5 => println(x)
      case _ => println("unknown")
    }
  }

  /**
    * 匹配数组
    */
  def matchArray(array: Array[Int]) = {
    array match {
      case Array(1, x, y) => println("匹配以1开头，有三个元素的数组")
      case Array(0) => println("匹配只有0这个元素的数组")
      case Array(0, _*) => println("匹配以0开头，任意多个元素的数组")
      case _ => println("unknown")
    }
  }

  /**
    * 匹配序列List
    * Nil是一个空的List，定义为List[Nothing]
    */
  def matchList(list: List[Int]) = {
    list match {
      case 5 :: Nil => println("匹配只有5这个元素的序列")
      case x :: y :: Nil => println("匹配只有两个元素的序列")
      case x :: tail => println("匹配有任意多个元素的序列")
      case _ => println("unknown")
    }
  }

  /**
    * 匹配元组
    */
  def matchTuple(tuple:Any) = {
    tuple match {
      case (x, y, 6) => println("匹配只有三个元素并且以6结尾的元组")
      case (1, x, y) => println("匹配以1开头有三个元素的元组")
      case _ => println("unknown")
    }
  }

  def main(args: Array[String]): Unit = {
    val test = func("a")
    println(test)
    println(matchString("b"))
    matchType("你好")
    matchType(3)
    matchType(6)
    matchType(1.5)
    println("============================")
    matchArray(Array(1, 2, 3))
    matchArray(Array(0))
    matchArray(Array(0, 1, 2, 3))
    matchArray(Array(1, 2))
    println("=========================")
    matchList(List(5, 6, 7))
    println("=========================")
    matchTuple((1,2,6))
  }
}