Golang 筆記 — Type Assertion

在 golang 當中，對於 interface{} 的型別，你可以做型別斷定，來假設這個 interface 是某個型別，這樣一來你就可以操作這個型別中的方法。

var i interface{} = "hello"
s := i.(string)
fmt.Println(s)

這段程式碼可以將原本的 interface{} 斷定為字串。

不過要特別注意的是，如果這個型別並不是字串的話，這段程式碼會直接 panic。

args := []interface{}{
		1,
		"1",
		"2",
	}
for _, arg := args {
  val := arg.(int) // interface conversion: interface {} is string, not int
  fmt.Println(val)
}

golang 提供了一個機制，型別斷定會回傳兩個數字，一個是轉換後的數字，另外一個是這個斷定是否正確。

args := []interface{}{
		1,
		"1",
		"2",
	}
for _, arg := args {
  if val, ok := arg.(int); ok {
    fmt.Println(val)    
  } else {
    fmt.Println("not int.")
  }
}

到這邊，有兩個地方需要注意：

如果沒有第二個回傳值，除非你很確定型別，不然程式有可能因為斷定失敗而導致 panic
加入第二個回傳值後，不會造成 panic

雖然麻煩一些，但還是總是做型別轉換比較保守一點。

另外要注意的一點是，斷定失敗後的 val 會是你斷定型別的 zero-value。

斷定失敗後的 val 會是你斷定型別的 zero-value。

for _, arg := range args {
  if val, ok := arg.(int); ok {
    fmt.Println(val, ok) // 1, true
  } else {
    fmt.Println(val, ok) // 0 , false
    // 0, false
  }
}

一般來講斷定失敗後，是不常用到這個 value 啦，但一不小心可能還是會落入陷阱。