前言
Scala有一个十分强大的模式匹配机制,可以应用到很多场合:如switch语句、类型检查等。并且Scala还提供了样例类,对模式匹配惊醒了优化,可以快速进行匹配。
1. 匹配字符串
import scala.util.Random
object CaseDemo01 extends App{
val arr = Array("hadoop","zookeeper","spark")
val name = arr(Random.nextInt(arr.length))
name match{
case "hadoop" => println("大数据分布式存储和计算框架...")
case "zookeeper" => println("大数据分布式协调服务框架...")
case "spark" => println("大数据分布式内存计算框架...")
case _ => println("我不认识你...")
}
}
2. 匹配类型
import scala.util.Random
object CaseDemo02 extends App{
val arr = Array("hello",1,2.0,CaseDemo)
val v = arr(Random.nextInt(4))
println(v)
v match{
case x:Int => println("Int" + x)
case y:Double if(y >= 0) => println("Double" + y)
case z:String => println("String" + z)
case _ => throw new Exception("not match exception")
}
}
3. 匹配数组、元组、集合
object CaseDemo03 extends App{
val arr = Array(1, 3, 5)
arr match {
case Array(1, x, y) => println(x + " " + y)
case Array(0) => println("only 0")
case Array(0, _*) => println("0 ...")
case _ => println("something else")
}
val lst = List(3, -1)
lst match {
case 0 :: Nil => println("only 0")
case x :: y :: Nil => println(s"x: $x y: $y")
case 0 :: tail => println("0 ...")
case _ => println("something else")
}
val tup = (1, 3, 7)
tup match {
case (1, x, y) => println(s"1, $x , $y")
case (_, z, 5) => println(z)
case _ => println("else")
}
}
注意:在Scala中列表要么为空(Nil表示空列表),要么是一个head元素加上一个tail列表。9 :: List(5, 2) :: 操作符是将给定的头和尾创建一个新的列表
注意::: 操作符是右结合的,如9 :: 5 :: 2 :: Nil相当于 9 :: (5 :: (2 :: Nil))
4. 样例类
在Scala中样例类是一种特殊的类,可用于模式匹配。
定义形式:
case class 类型,是多例的,后面要跟构造参数。
case object 类型,是单例的。
import scala.util.Random
case class SubmitTask(id:String,name:String)
case class HeartBeat(time:Long)
case object CheckTimeOutTask
object CaseDemo04 extends App{
val arr = Array(CheckTimeOutTask,HeartBeat(12333),SubmitTask("0001","task-0001"))
arr(Random.nextInt(arr.length) match){
case SubmitTask(id,name) =>{
println(s"$id,$name")
}
case HeartBeat(time) => {
println(time)
}
case CheckTimeOutTask => {
println("check")
}
}
}
5. Option类型
在Scala中Option类型用样例类来表示可能存在或者可能不存在的值(Option的子类有Some和None)。Some包装了某个值,None表示没有值。
object OptionDemo{
def main(args:Array[String]){
val map = Map("a" -> 1,"b" -> 2)
val v = map.get("b") match{
case Some(i) => i
case None => 0
}
println(v)
//更好的方式
val v1 = map.getOrElse("c",0)
println(v1)
}
}
6. 偏函数
被包在花括号内没有match的一组case语句是一个偏函数,它是PartialFunction[A, B]的一个实例,A代表输入参数类型,B代表返回结果类型,常用作输入模式匹配,偏函数最大的特点就是它只接受和处理其参数定义域的一个子集。
object PartialFuncDemo {
val func1: PartialFunction[String, Int] = {
case "one" => 1
case "two" => 2
case _ => -1
}
def func2(num: String) : Int = num match {
case "one" => 1
case "two" => 2
case _ => -1
}
def main(args: Array[String]) {
println(func1("one"))
println(func2("one"))
}
}