模式匹配和样例类、匹配字符串、匹配类型、匹配数组/元组/集合、样例类、Option类型、偏函数 15

桃扇骨 2022-01-30 11:53 88阅读 0赞

# 前言 #

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"))
      }
    }