首页 > 代码库 > Scala Control Structures

Scala Control Structures

2024-09-01 23:37:30   218人阅读

Scala之Control Structures

一、前言

 

　　前面学习了Scala的Numbers，接着学习Scala的Control Structures(控制结构)。

 

二、Control Structures

 

　　Scala中的控制结构与Java中的颇为类似，但也有所不同，例如，if/then/else控制结构与Java的类似，但是其可以返回值，虽然Java中有三元运算符的特殊语法，但是在Scala中使用if就可以达到同样的效果。　　

 

val x = if (a) y else z

 

　　同样，Scala的try/catch/finally 结构与Java的也类似，但是Scala在catch中时使用的模式匹配。

 

　　当使用for循环时，情况会变得更有趣，for循环的基本用法在Scala和Java中都相同，但是在Scala可以添加守卫。例如，可以使用两个for循环来读取文件中的每一行，并且对每一行的每个字符进行操作。　　

 

for (line <- source.getLines) {
    for {
        char <- line
        if char.isLetter
    } // char algorithm here ...
}

 

　　在Scala中，还可以使用更精炼的语句实现上述功能　

 

for {
    line <- source.getLines
    char <- line
    if char.isLetter
} // char algorithm here ...

 

　　在Scala中，使用for循环可以让你能够很轻易的将一个集合转化成另外一个集合

 

　　技术分享

 

　　在最基础的使用中，Scala的match表达式与Java的Switch语句类似，你可以匹配任何对象，并且从对象中解析信息。

 

　　2.1 for和foreach循环

 

　　1. 问题描述

 

　　你需要遍历集合的元素，或者操作集合中的元素，或者在已有集合上创建新的集合

 

　　2. 解决方案

 

　　有很多方式可以遍历集合，如for、while循环，还有集合的foreach、map、flatMap方法等，首先看看for循环

 

　　技术分享

 

　　当你的算法有多行时，也可以使用for循环，如下所示

 

　　技术分享

 

　　上述示例中仅仅只是遍历了集合中的元素，并且没有返回值，如新的Array，当你想要从已有集合中创建新的集合时，可以使用for/yield

 

　　技术分享

 

　　for/yield结构返回了一个新的Array。

 

　　当在for循环中需要访问计数器时，可以使用如下方法

 

　　技术分享

 

　　Scala集合也提供了zipWithIndex方法来创建循环计数器

 

　　技术分享

 

　　下面示例展示了如何通过Range来执行循环三次

 

　　技术分享

 

　　如下示例展示了如何使用守卫(在for循环中使用if语句)

 

　　技术分享

 

　　可以使用如下精炼方式遍历Map的键集合和值集合

 

　　技术分享

 

　　3. 讨论

 

　　当你在集合上使用for/yield时，会得到一个新的集合，而仅仅使用for时，则不会产生新的集合，仅仅只是利用for的副作用对集合元素进行操作。for循环可能不是遍历集合最好的方式，使用foreach、map、flatMap、collect、reduce等方法可能会更高效的解决问题。

 

　　你可以调用foreach方法来遍历每个集合元素

 

　　技术分享

 

　　可以使用匿名函数对元素进行操作

 

　　技术分享

 

　　for循环如何被翻译

 

　　　　· for循环会被编译器翻译成foreach方法

 

　　　　· 有守卫的for循环会被翻译成withFilter方法(在foreach之后)

 

　　　　· 有yield的for循环会被翻译成map方法

 

　　　　· 有yield和守卫的for循环会被翻译成withFilter(在map之后)

 

　　为了验证for循环的翻译过程，新建Main.scala文件，源代码如下　　

 

class Main {
    for (i <- 1 to 10) println(i)
}

 

　　使用scalac -Xprint:parse Main.scala命令进行编译

 

　　技术分享

 

　　可以看到for循环被翻译成了foreach方法

 

　　其他的带守卫、yield的for循环也可以采用上述方法进行查看，不再累赘

 

　　2.2 使用多计数器的for循环

 

　　1. 问题描述

 

　　你想要创建拥有多个计数器的循环，如当遍历一个多维数组

 

　　2. 解决方案

 

　　你可以使用如下方法创建多个计数器

 

　　技术分享

 

　　一种更好的方式是使用中括号

 

　　技术分享

 

　　同理，也可以使用三个计数器

 

　　技术分享

 

　　当遍历多位数组时，该方法很有效

 

　　技术分享

 

　　3. 讨论

 

　　在循环中，使用<-符号生成Range被称为生成器，可以在一个循环中使用多个生成器

 

　　2.3 使用内嵌语句的for循环

 

　　1. 问题描述

 

　　你想要在for循环中添加一个或多个条件子句，通常在筛选集合中的某些元素

 

　　2. 解决方案

 

　　可以在生成器后面添加if语句

 

　　技术分享

 

 　　或者使用中括号

 

　　技术分享

 

　　上述的if语句被称为过滤器或守卫，下面示例添加了多个守卫

 

　　技术分享

 

　　3. 讨论

 

　　在for循环中使用守卫可以是代码精炼并且提高可读性，当然也可以使用传统的方法

 

　　2.4 创建for/yield组合

 

　　1. 问题描述

 

　　你想要通过对原始集合中的每个元素应用某种算法（一个或多个守卫）来从现有集合创建新的集合

 

　　2. 解决方案

 

　　使用yield和算法来从现有集合生成新的集合

 

　　技术分享

 

　　如果算法有多行，可以采用如下形式

 

　　技术分享

 

　　通常情况下，使用yield新生的集合的类型和原始集合的类型相同，如原始集合时ArrayBuffer类型，则新生的集合也是ArrayBuffer类型

 

　　技术分享

 

　　技术分享

 

　　3. 讨论

 

　　当在for循环中使用yield时，可以将其看做如下

 

　　　　· 开始运行时，for/yield会新建一个空的集合，类型和原始集合类型相同

 

　　　　· 在for循环的每次迭代中，从输入集合的当前元素创建新的输出元素，放置在新的集合中

 

　　　　· 当for循环结束时，返回新的集合中的所有内容

 

　　没有守卫的for/yield语句就相当于调用集合的map方法

 

　　技术分享

 

　　2.5 实现break和continue

 

　　1. 问题描述

 

　　你需要使用break和continue，但是在Scala中并没有break和continue关键字

 

　　2. 解决方案

 

　　scala.util.control.Breaks提供了break功能　　

 

package com.hust.grid.leesf.breakcontinue

/**
  * Created by LEESF on 2017/1/18.
  */
import util.control.Breaks._

object BreakAndContinueDemo extends App{
  println("\n=== BREAK EXAMPLE ===")
  breakable {
    for (i <- 1 to 10) {
      println(i)
      if (i > 4) break // break out of the for loop
    }
  }
  println("\n=== CONTINUE EXAMPLE ===")
  val searchMe = "peter piper picked a peck of pickled peppers"
  var numPs = 0
  for (i <- 0 until searchMe.length) {
    breakable {
      if (searchMe.charAt(i) != ‘p‘) {
        break // break out of the ‘breakable‘, continue the outside loop
      } else {
        numPs += 1
      }
    }
  }
  println("Found " + numPs + " p‘s in the string.")

}

 

　　结果　　

 

=== BREAK EXAMPLE ===
1
2
3
4
5

=== CONTINUE EXAMPLE ===
Found 9 p‘s in the string.

 

　　值得注意的是break和breakable都不是Scala的关键字，它们是scala.util.control.Breaks包下的方法

 

private val breakException = new BreakControl

def break(): Nothing = { throw breakException }

def breakable(op: => Unit) {
    try {
        op
    } catch {
        case ex: BreakControl =>
        if (ex ne breakException) throw ex
    }
}

 

　　可以看到在调用break方法时，会抛出一个异常，此时breakable会捕捉到此异常，利用这种方式实现了break和continue的功能

 

　　3. 讨论

 

　　如果你不喜欢使用break和continue时，可以使用如下方法，如可以使用带有守卫的for循环　

 

var barrelIsFull = false
for (monkey <- monkeyCollection if !barrelIsFull) {
    addMonkeyToBarrel(monkey)
    barrelIsFull = checkIfBarrelIsFull
}

 

　　或者使用return直接返回　　

 

def sumToMax(arr: Array[Int], limit: Int): Int = {
    var sum = 0
    for (i <- arr) {
        sum += i
        if (sum > limit) return limit
    }
    sum
}

val a = Array.range(0,10)
println(sumToMax(a, 10))

 

　　2.6 使用类似三元运算符的if表达式

 

　　1. 问题描述

 

　　你需要使用if表达式来像三元运算符那样简洁的解决问题

 

　　2. 解决方案

 

　　在Scala中没有三元运算法，但是可以使用if/else表达式

 

　　技术分享

 

　　3. 讨论

 

　　在Java中可以使用?:三元运算符，在Scala可以使用if/else表达式

 

　　2.7 使用类似switch语句的match表达式

 

　　1. 问题描述

 

　　你想要创建向Java那样简单的switch语句，如星期几，哪一月

 

　　2. 解决方案

 

　　使用Scala的match表达式　

 

// i is an integer
i match {
    case 1 => println("January")
    case 2 => println("February")
    case 3 => println("March")
    case 4 => println("April")
    case 5 => println("May")
    case 6 => println("June")
    case 7 => println("July")
    case 8 => println("August")
    case 9 => println("September")
    case 10 => println("October")
    case 11 => println("November")
    case 12 => println("December")
    // catch the default with a variable so you can print it
    case whoa => println("Unexpected case: " + whoa.toString)
}

 

　　还可以直接利用match表达式的返回值　

 

val month = i match {
    case 1 => "January"
    case 2 => "February"
    case 3 => "March"
    case 4 => "April"
    case 5 => "May"
    case 6 => "June"
    case 7 => "July"
    case 8 => "August"
    case 9 => "September"
    case 10 => "October"
    case 11 => "November"
    case 12 => "December"
    case _ => "Invalid month" // the default, catch-all
}

 

　　当写match表达式时，推荐使用@switch注解，当switch不能编译成一个tableswitch或lookupswitch，该注解会提供编译时警告，将switch编译成tableswitch或lookupswitch会有更高的性能，因为结果是在分支表而不是一个决策树里，当给表达式赋值时，它可以直接跳转到结果，而不是遍历决策树　

 

import scala.annotation.switch

class SwitchDemo {
    val i = 1
    val x = (i: @switch) match {
        case 1 => "One"
        case 2 => "Two"
        case _ => "Other"
    }
}

 

　　先使用scalac SwitchDemo命令进行编译，然后使用javap SwitchDemo命令反编译

 

　　技术分享

 

　　可以看到生成了tableswitch表，当修改代码如下时　　

 

import scala.annotation.switch

class SwitchDemo {
    val i = 1
    val Two = 2 // added
    val x = (i: @switch) match {
        case 1 => "One"
        case Two => "Two" // replaced the ‘2‘
        case _ => "Other"
    }
}

 

　　同样使用scalac SwitchDemo命令进行编译

 

　　技术分享

 

　　这表明match表达式无法生成tableswitch或lookupswitch，当使用javap命令反汇编时，也找不到tableswitch

 

　　3. 讨论

 

　　match表达式非常灵活，可以匹配不同的类型　　

 

def getClassAsString(x: Any): String = x match {
    case s: String => s + " is a String"
    case i: Int => "Int"
    case f: Float => "Float"
    case l: List[_] => "List"
    case p: Person => "Person"
    case _ => "Unknown"
}

 

　　可以使用case _通配符来匹配之前的所有case无法匹配的情况，或者使用case default进行赋值

 

case default => println(default)

 

　　2.8 用一个case语句匹配多个条件

 

　　1. 问题描述

 

　　当几个匹配条件要求执行相同的业务逻辑，你想要使用业务逻辑的一个副本来匹配条件，而不是为每个case重复业务逻辑

 

　　2. 解决方案

 

　　将匹配条件放在一行使用|分割　

 

val i = 5
i match {
    case 1 | 3 | 5 | 7 | 9 => println("odd")
    case 2 | 4 | 6 | 8 | 10 => println("even")
}

 

　　同理，该语法也适用于字符串或其他类型　

 

val cmd = "stop"

cmd match {
    case "start" | "go" => println("starting")
    case "stop" | "quit" | "exit" => println("stopping")
    case _ => println("doing nothing")
}　

 

trait Command

case object Start extends Command
case object Go extends Command
case object Stop extends Command
case object Whoa extends Command

def executeCommand(cmd: Command) = cmd match {
    case Start | Go => start()
    case Stop | Whoa => stop()
}

 

　　2.9. 将match表达式的值赋值给变量

 

　　1. 描述

 

　　你想要将匹配表达式返回值其赋值给变量，或使用匹配表达式作为方法的主体

 

　　2. 解决方案

 

　　在表达式之前插入变量　　

 

val evenOrOdd = someNumber match {
    case 1 | 3 | 5 | 7 | 9 => println("odd")
    case 2 | 4 | 6 | 8 | 10 => println("even")
}

 

　　这种方法经常被用作创建短的方法或函数　

 

def isTrue(a: Any) = a match {
    case 0 | "" => false
    case _ => true
}

 

　　2.10 访问匹配表达式默认情况下的值

 

　　1. 问题描述

 

　　你想要访问默认匹配值，但是使用通配符时无法访问到值

 

　　2. 解决方案

 

　　在默认情况下指定变量名而非_通配符　

 

i match {
    case 0 => println("1")
    case 1 => println("2")
    case default => println("You gave me: " + default)
}

 

　　通过给默认匹配变量名称，你可以在语句右侧访问该变量

 

　　3. 讨论

 

　　在默认匹配时使用变量名而非通配符，若没有给出通配符时，可能抛出异常

 

　　技术分享

 

　　2.11 在匹配表达式中使用模式匹配

 

　　1. 问题描述

 

　　你需要在匹配表达式中匹配一个或多个模式，该模式可以是常量模式、变量模式、构造函数模式、序列模式、元组模式或类型模式等

 

　　2. 解决方案

 

　　为要匹配的每个模式定义一个case语句　

 

def echoWhatYouGaveMe(x: Any): String = x match {
    // constant patterns
    case 0 => "zero"
    case true => "true"
    case "hello" => "you said ‘hello‘"
    case Nil => "an empty List"
    
    // sequence patterns
    case List(0, _, _) => "a three-element list with 0 as the first element"
    case List(1, _*) => "a list beginning with 1, having any number of elements"
    case Vector(1, _*) => "a vector starting with 1, having any number of elements"
    
    // tuples
    case (a, b) => s"got $a and $b"
    case (a, b, c) => s"got $a, $b, and $c"
    
    // constructor patterns
    case Person(first, "Alexander") => s"found an Alexander, first name = $first"
    case Dog("Suka") => "found a dog named Suka"
    
    // typed patterns
    case s: String => s"you gave me this string: $s"
    case i: Int => s"thanks for the int: $i"
    case f: Float => s"thanks for the float: $f"
    case a: Array[Int] => s"an array of int: ${a.mkString(",")}"
    case as: Array[String] => s"an array of strings: ${as.mkString(",")}"    
    case d: Dog => s"dog: ${d.name}"
    case list: List[_] => s"thanks for the List: $list"
    case m: Map[_, _] => m.toString
    
    // the default wildcard pattern
    case _ => "Unknown
}

 

　　上述方法可以匹配不同的类型和模式

 

　　3. 讨论

 

　　通常使用此技术时，你的方法希望传入的是基类或特性继承的实例，然后case语句引用该基类型的子类型　　

 

import java.io.File

sealed trait RandomThing

case class RandomFile(f: File) extends RandomThing
case class RandomString(s: String) extends RandomThing

class RandomNoiseMaker {
    def makeRandomNoise(t: RandomThing) = t match {
        case RandomFile(f) => playSoundFile(f)
        case RandomString(s) => speak(s)
    }
}

 

　　sealed表示密封的特质，其子类必须全部给出。

 

　　有时你可能需要向模式添加变量，你可以用下面的一般语法：

 

　　变量名 @ 模式

 

　　这表示变量绑定模式，这种模式的含义是按照正常模式进行模式匹配，如果模式匹配成功，就像简单的变量模式一样，将变量设置为匹配对象。

 

　　当匹配List类型时，我们可以如下　

 

case list: List[_] => s"thanks for the List: $list"

 

　　当我们想要匹配以元素1作为头元素的列表时，我们可能会进行如下匹配　　

 

case list: List(1, _*) => s"thanks for the List: $list"

 

　　但此时会报错

 

技术分享

 

　　解决办法是添加变量绑定模式　　

 

case list @ List(1, _*) => s"$list"

 

　　2.12. 在匹配表达式中使用案例类

 

　　1. 问题描述

 

　　你希望在匹配表达式中匹配不同的case类（或case对象），如接收actor的消息时

 

　　2. 解决方案

 

　　根据你的需要可以使用之前的不同模式进行匹配　

 

trait Animal

case class Dog(name: String) extends Animal
case class Cat(name: String) extends Animal
case object Woodpecker extends Animal
    
object CaseClassTest extends App {
    def determineType(x: Animal): String = x match {
        case Dog(moniker) => "Got a Dog, name = " + moniker
        case _:Cat => "Got a Cat (ignoring the name)"
        case Woodpecker => "That was a Woodpecker"
        case _ => "That was something else"
    }
    
    println(determineType(new Dog("Rocky")))
    println(determineType(new Cat("Rusty the Cat")))
    println(determineType(Woodpecker))
}

 

　　2.13 将if表达式（守卫）添加到case语句中

 

　　1. 问题描述

 

　　你希望将匹配逻辑添加到匹配表达式中的case语句中，例如允许一系列数字或匹配模式，但仅当该模式与某些附加条件相匹配时

 

　　2. 解决方案

 

　　将if表达式添加至case语句中　

 

i match {
    case a if 0 to 9 contains a => println("0-9 range: " + a)
    case b if 10 to 19 contains b => println("10-19 range: " + b)
    case c if 20 to 29 contains c => println("20-29 range: " + c)
    case _ => println("Hmmm...")
}

 

　　可以匹配不同的值

 

num match {
    case x if x == 1 => println("one, a lonely number")
    case x if (x == 2 || x == 3) => println(x)
    case _ => println("some other value")
}

 

　　也可以在匹配时解析出字段信息　

 

def speak(p: Person) = p match {
    case Person(name) if name == "Fred" => println("Yubba dubba doo")
    case Person(name) if name == "Bam Bam" => println("Bam bam!")
    case _ => println("Watch the Flintstones!")
}

 

　　3. 讨论

 

　　在进行匹配时也可将if放在=>的右侧　　

 

case Person(name) =>
    if (name == "Fred") println("Yubba dubba doo")
    else if (name == "Bam Bam") println("Bam bam!")

 

　　2. 14 使用匹配表达式而非isInstanceOf

 

　　1. 问题描述

 

　　你需要编写匹配一种类型或多个不同类型的代码块

 

　　2. 解决方案

 

　　你可以使用isInstanceOf来测试对象的类型　　

 

if (x.isInstanceOf[Foo]) { do something ...}

 

　　然而并不鼓励这样做，更好的办法是使用匹配表达式　　

 

def isPerson(x: Any): Boolean = x match {
    case p: Person => true
    case _ => false
}

 

　　或者给出一个继承已知父类的子类，需要根据子类的不同做出不同的操作　　

 

trait SentientBeing
trait Animal extends SentientBeing

case class Dog(name: String) extends Animal
case class Person(name: String, age: Int) extends SentientBeing

def printInfo(x: SentientBeing) = x match {
    case Person(name, age) => // handle the Person
    case Dog(name) => // handle the Dog
}

 

　　3. 讨论

 

　　匹配表达式可以匹配多种类型，而不需要使用isInstanceOf方法

 

　　2. 15. 在匹配表达式中使用列表

 

　　1. 问题描述

 

　　列表数据结构与其他集合数据结构有点不同，它从cons cells开始构建，并以Nil作为结尾，在使用匹配表达式时你想要利用此特性，例如编写递归程序

 

　　2. 解决方案

 

　　你可以使用如下方法创建List

 

　　技术分享

 

　　当编写递归函数时，你可以利用其尾元素是Nil类型对象

 

　　技术分享

 

　　3. 讨论

 

　　在REPL中构建List时，不要忘记Nil情况

 

　　2.16 使用try/catch匹配一个或多个异常

 

　　1. 问题描述

 

　　在try/catch块中你想要匹配一个或多个异常

 

　　2. 解决方案

 

　　Scala的try/catch/finally与Java类型，但是在catch中使用的是模式匹配　

 

val s = "Foo"
try {
    val i = s.toInt
} catch {
    case e: Exception => e.printStackTrace
}

 

　　当你需要匹配多个异常时，在catch添加即可　

 

try {
    openAndReadAFile(filename)
} catch {
    case e: FileNotFoundException => println("Couldn‘t find that file.")
    case e: IOException => println("Had an IOException trying to read that file")
}

 

　　3. 讨论

 

　　Scala中的catch块可以匹配多个异常，如果你不关心发生的异常，可以使用如下方式　

 

try {
    openAndReadAFile("foo")
} catch {
    case t: Throwable => t.printStackTrace()
}

 

try {
    val i = s.toInt
} catch {
    case _: Throwable => println("exception ignored")
}

 

　　2.17 在try/catch/finally块之前声明变量

 

　　1. 问题描述

 

　　当你需要调用对象的闭包方法，你想要在try块中使用一个对象，并且在finally块中访问它

 

　　2. 解决方案

 

　　在try之前将字段声明为Option类型，之后在try块中创建一个Some对象　

 

import java.io._

object CopyBytes extends App {
    var in = None: Option[FileInputStream]
    var out = None: Option[FileOutputStream]
    try {
        in = Some(new FileInputStream("/tmp/Test.class"))
        out = Some(new FileOutputStream("/tmp/Test.class.copy"))
        var c = 0
        while ({c = in.get.read; c != ?1}) {
            out.get.write(c)
        }
    } catch {
        case e: IOException => e.printStackTrace
    } finally {
        println("entered finally ...")
        if (in.isDefined) in.get.close
        if (out.isDefined) out.get.close
    }
}

 

　　2.18 自定义控制结构

 

　　1. 问题描述

 

　　你想要自定义控制结构来改善Scala语言

 

　　2. 解决方案

 

　　假如有一天你不喜欢while结构，而是如下结构　

 

package foo

import com.alvinalexander.controls.Whilst._

object WhilstDemo extends App {
    var i = 0
    whilst (i < 5) {
        println(i)
        i += 1
    }
}

 

　　此时，whilist就是你自定义的控制结构，你可能采用如下方法定义whilist方法　

 

def whilst(testCondition: => Boolean)(codeBlock: => Unit) {
    while (testCondition) {
        codeBlock
    }
}

 

　　但是更好的方法是不调用while循环　　

 

package com.leesf.controls

import scala.annotation.tailrec

object Whilst {
    // 2nd attempt
    @tailrec
    def whilst(testCondition: => Boolean)(codeBlock: => Unit) {
        if (testCondition) {
            codeBlock
            whilst(testCondition)(codeBlock)
        }
    }
}

 

　　3. 讨论

 

　　在第二版的whilist中，使用了递归调用，假设有一个两个参数的控制结构，当都为真时执行代码块，调用代码如下

 

doubleif(age > 18)(numAccidents == 0) { println("Discount!") }

 

　　doubleif方法则可定义如下　　

 

def doubleif(test1: => Boolean)(test2: => Boolean)(codeBlock: => Unit) {
    if (test1 && test2) {
        codeBlock
    }
}

 

　　其中，codeBlock: => Unit对应的是println("Discount!")

 

三、总结

 

　　本篇博文学习了Scala的控制结构及其细节，同时还学习了自定义控制结构，也谢谢各位园友的观看~

Scala Control Structures


联系
我们