【Scala】使用Option、Either和Try處理資料互動
Scala資料互動
Scala使用一種函式式的方式來處理資料互動,包括入參及返回值。
- Option: 解決null(空指標)問題
- Either: 解決返回值不確定(返回兩個值的其中一個)問題
- Try: 解決函式可能會丟擲異常問題
Option/Some/None的使用
Option實際上有3個型別:Option、Some和None,Some和None都是Option的子型別,Some和None。Option表示可選的值,它的返回型別是scala.Some
或scala.None
。Some代表返回有效資料,None代表返回空值。
返回Option物件的函式
該函式以String物件作為輸入,如果String物件被正確轉換為Int物件,返回Sone[Int];否則返回None:
def toInt(s: String): Option[Int] = {
try {
Some(Integer.parseInt(s.trim))
} catch {
case e : Exception => None
}
}
在Scala的集合類中使用Option
假設有一個字串列表,我們希望得到該列表中所有的整數,通過將toInt方法傳入List物件的map方法中,將列表元素轉換成Some或None值:
scala> val bag = List("1", "2", "foo", "4", "bar")
bag: List[String] = List(1, 2, foo, 4, bar)
//通過flatten將原來的Option物件列表轉換為整數列表
//由於Option是一個含有一個元素或0個元素(None)的集合,故能做出該轉換
scala> bag.map(toInt).flatten
res1: List[Int] = List(1, 2, 4)
//通過flatMap實現相同的轉換
scala> bag.flatMap(toInt)
res2: List[Int] = List(1, 2, 4)
//通過collect方法實現同樣的功能
scala> bag.map(toInt).collect{case Some(i) => i}
res12: List[Int] = List(1, 2, 4)
Option的高階函式
下面的函式是將一個Option[String]物件中字串的長度打印出來,其中用到了Option的map方法和foreach方法:
def printContentLength(x: Option[String]): Unit = {
x.map("length: " + _.length).foreach(println)
}
val value1 = Some("value1")
val value2 = None
printContentLength(value1) //length: 6
printContentLength(value2) //無列印
下面是將Option[String]物件中的字串進行修剪並轉換為大寫字母:
def trimUpper(x: Option[String]): Option[String] = {
x map (_.trim) filter (!_.isEmpty) map (_.toUpperCase)
}
val name1 = Some(" name ")
val name2 = None
println(trimUpper(name1) ) //Some(NAME)
println(trimUpper(name2) ) //None
Try/Success/Failue的使用
在readfile類似的方法裡,我們會使用了try catch語法。Scala2.10提供了Try來更優雅的實現這一功能。對於有可能丟擲異常的操作。我們可以使用Try來包裹它,得到Try的子類Success或者Failure,如果計算成功,返回Success的例項,如果丟擲異常,返回Failure並攜帶相關資訊。
import scala.util.{Try, Success, Failure}
def divideBy(x: Int, y: Int): Try[Int] = {
Try(x / y)
}
println(divideBy(1, 1).getOrElse(0)) // 1
println(divideBy(1, 0).getOrElse(0)) //0
divideBy(1, 1).foreach(println) // 1
divideBy(1, 0).foreach(println) // no print
divideBy(1, 0) match {
case Success(i) => println(s"Success, value is: $i")
case Failure(s) => println(s"Failed, message is: $s")
}
//Failed, message is: java.lang.ArithmeticException: / by zero
readTextFile例子
如果該方法返回成功,將列印/etc/passwd
檔案的內容;如果出現異常,將列印錯誤資訊,java.io.FileNotFoundException: Foo.bar (No such file or directory)
def readTextFile(filename: String): Try[List[String]] = {
Try(Source.fromFile(filename).getLines.toList)
}
val filename = "/etc/passwd"
readTextFile(filename) match {
case Success(lines) => lines.foreach(println)
case Failure(f) => println(f)
}
}
另外
- Try有類似集合的操作 filter, flatMap, flatten, foreach, map
- get, getOrElse, orElse方法
- toOption可以轉化為Option
- recover,recoverWith,transform可以讓你優雅地處理Success和Failure的結果
Either/Left/Right的使用
程式設計中經常會有這樣的需求,一個函式(或方法)在傳入不同引數時會返回不同的值。返回值是兩個不相關的型別,分別為: Left 和 Right 。慣例中我們一般認為 Left 包含錯誤或無效值, Right包含正確或有效值。
在Scala 2.10之前,Either/Right/Left類和Try/Success/Failure類是相似的效果。
def divideBy2(x: Int, y: Int): Either[String, Int] = {
if(y == 0) Left("Dude, can't divide by 0")
else Right(x / y)
}
divideBy2(1, 0) match {
case Left(s) => println("Answer: " + s)
case Right(i) => println("Answer: " + i)
}
//print "Answer: Dude, can't divide by 0"
除了使用match case方式來獲取資料,我們還可以分別使用 .right.get 和 .left.get 方法,當然你需要使用 .isRight 或 .isLeft 先判斷一下。Left或Right型別也有 filter, flatMap, foreach, get, getOrElse, map 方法,它們還有toOption, toSeq 方法,分別返回一個Option或Seq 。