Go 語言的函數(shù)經(jīng)常使用兩個返回值來表示執(zhí)行是否成功:返回某個值以及 true 表示成功;返回零值(或 nil)和 false 表示失?。ǖ?4.4 節(jié))。當不使用 true 或 false 的時候,也可以使用一個 error 類型的變量來代替作為第二個返回值:成功執(zhí)行的話,error 的值為 nil,否則就會包含相應(yīng)的錯誤信息(Go 語言中的錯誤類型為 error: var err error,我們將會在第 13 章進行更多地討論)。這樣一來,就很明顯需要用一個 if 語句來測試執(zhí)行結(jié)果;由于其符號的原因,這樣的形式又稱之為 comma,ok 模式(pattern)。
在第 4.7 節(jié)的程序 string_conversion.go 中,函數(shù) strconv.Atoi 的作用是將一個字符串轉(zhuǎn)換為一個整數(shù)。之前我們忽略了相關(guān)的錯誤檢查:
anInt, _ = strconv.Atoi(origStr)
如果 origStr 不能被轉(zhuǎn)換為整數(shù),anInt 的值會變成 0 而 _ 無視了錯誤,程序會繼續(xù)運行。
這樣做是非常不好的:程序應(yīng)該在最接近的位置檢查所有相關(guān)的錯誤,至少需要暗示用戶有錯誤發(fā)生并對函數(shù)進行返回,甚至中斷程序。
我們在第二個版本中對代碼進行了改進:
示例 1:
示例 5.3 string_conversion2.go
package main
import (
"fmt"
"strconv"
)
func main() {
var orig string = "ABC"
// var an int
var newS string
// var err error
fmt.Printf("The size of ints is: %d\n", strconv.IntSize)
// anInt, err = strconv.Atoi(origStr)
an, err := strconv.Atoi(orig)
if err != nil {
fmt.Printf("orig %s is not an integer - exiting with error\n", orig)
return
}
fmt.Printf("The integer is %d\n", an)
an = an + 5
newS = strconv.Itoa(an)
fmt.Printf("The new string is: %s\n", newS)
}
這是測試 err 變量是否包含一個真正的錯誤(if err != nil)的習(xí)慣用法。如果確實存在錯誤,則會打印相應(yīng)的錯誤信息然后通過 return 提前結(jié)束函數(shù)的執(zhí)行。我們還可以使用攜帶返回值的 return 形式,例如 return err。這樣一來,函數(shù)的調(diào)用者就可以檢查函數(shù)執(zhí)行過程中是否存在錯誤了。
習(xí)慣用法
value, err := pack1.Function1(param1)
if err != nil {
fmt.Printf("An error occured in pack1.Function1 with parameter %v", param1)
return err
}
// 未發(fā)生錯誤,繼續(xù)執(zhí)行:
由于本例的函數(shù)調(diào)用者屬于 main 函數(shù),所以程序會直接停止運行。
如果我們想要在錯誤發(fā)生的同時終止程序的運行,我們可以使用 os 包的 Exit 函數(shù):
習(xí)慣用法
if err != nil {
fmt.Printf("Program stopping with error %v", err)
os.Exit(1)
}
(此處的退出代碼 1 可以使用外部腳本獲取到)
有時候,你會發(fā)現(xiàn)這種習(xí)慣用法被連續(xù)重復(fù)地使用在某段代碼中。
當沒有錯誤發(fā)生時,代碼繼續(xù)運行就是唯一要做的事情,所以 if 語句塊后面不需要使用 else 分支。
示例 2:我們嘗試通過 os.Open 方法打開一個名為 name 的只讀文件:
f, err := os.Open(name)
if err != nil {
return err
}
doSomething(f) // 當沒有錯誤發(fā)生時,文件對象被傳入到某個函數(shù)中
doSomething
練習(xí) 5.1 嘗試改寫 string_conversion2.go 中的代碼,要求使用 := 方法來對 err 進行賦值,哪些地方可以被修改?
示例 3:可以將錯誤的獲取放置在 if 語句的初始化部分:
習(xí)慣用法
if err := file.Chmod(0664); err != nil {
fmt.Println(err)
return err
}
示例 4:或者將 ok-pattern 的獲取放置在 if 語句的初始化部分,然后進行判斷:
習(xí)慣用法
if value, ok := readData(); ok {
…
}
注意事項
如果您像下面一樣,沒有為多返回值的函數(shù)準備足夠的變量來存放結(jié)果:
func mySqrt(f float64) (v float64, ok bool) {
if f < 0 { return } // error case
return math.Sqrt(f),true
}
func main() {
t := mySqrt(25.0)
fmt.Println(t)
}
您會得到一個編譯錯誤:multiple-value mySqrt() in single-value context。
正確的做法是:
t, ok := mySqrt(25.0)
if ok { fmt.Println(t) }
注意事項 2
當您將字符串轉(zhuǎn)換為整數(shù)時,且確定轉(zhuǎn)換一定能夠成功時,可以將 Atoi 函數(shù)進行一層忽略錯誤的封裝:
func atoi (s string) (n int) {
n, _ = strconv.Atoi(s)
return
}
實際上,fmt 包(第 4.4.3 節(jié))最簡單的打印函數(shù)也有 2 個返回值:
count, err := fmt.Println(x) // number of bytes printed, nil or 0, error
當打印到控制臺時,可以將該函數(shù)返回的錯誤忽略;但當輸出到文件流、網(wǎng)絡(luò)流等具有不確定因素的輸出對象時,應(yīng)該始終檢查是否有錯誤發(fā)生(另見練習(xí) 6.1b)。