1. 程式人生 > >【Scala】使用Option、Either和Try處理資料互動

【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.Somescala.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 。