Go 語言的函數與指針
函數
函數就是一段基本的代碼塊,一般用來對需要重復執行的代碼進行復用。在 go 中,函數是『一等公民』,這與 js 類似,也就是可以將函數當做一個變量進行傳遞。
函數聲明
由于是強類型語言,與 js 不同,在函數聲明的過程中,需要指定參數與返回值的類型。
- func max (n1, n2 int) int {
- var result int
- if n1 >= n2 {
- result = n1
- }
- if n1 < n2 {
- result = n2
- }
- return result
- }
在聲明函數參數和類型的時候,與聲明變量類似,可以一次性指定多個參數的類型,也可以分別指定多個參數為不同類型。
- func max (n1 int, n2 int) int {
- ……
- }
如果函數返回值有多個,在指定返回類型的時候,需要指定每個返回值的類型。
- func max (n1 int, n2 int) (error, int) {
- ……
- return errors.New(""), result
- }
上面的代碼,表示返回的時候需要返回兩個值,第一個值為 error 對象,用來表示執行期間是否出現異常。這種方式也是 Node.js 中常見的 error-first callback 的寫法。
特殊函數
在 go 中,有兩個特殊函數:main、init,這兩個函數聲明之后,一般不需要主動調用,會有自動執行的機制。
func main()
main 函數是 go 語言中默認的入口函數,只能應用于 package main 中,如果在其他的 package 中不會執行。main 函數有如下幾點需要注意:
- 不能定義參數;
- 不能定義返回值;
- 必須在 package main 中聲明;
func init()
init 函數所有的包啟動的時候都會執行,執行時機比 main 函數早,與 main 函數一樣,不能定義參數和返回值。
- package main
- import "fmt"
- func init() {
- fmt.Println("執行 init 函數\n")
- }
- func main() {
- fmt.Println("執行 main 函數\n")
- }
函數調用
函數的調用比較簡單,和其他編程語言類似,只需要將函數需要接受的參數傳入其中,在執行結束后,就能得到對應的返回值。
- // 定義 max 函數
- func max (n1, n2 int) int {
- var result int
- if n1 >= n2 {
- result = n1
- }
- if n1 < n2 {
- result = n2
- }
- return result
- }
- func main () {
- var result = max(5, 100)
- fmt.Println("max return", result)
- }
匿名函數
匿名函數就是一個沒有定義函數名的函數,匿名函數可以當成一個值,將其賦值放到某個變量中。這也是之前為什么說函數是『一等公民』,就是可以將函數當成一個變量。
- var max = func (n1, n2 int) int {
- var result int
- if n1 >= n2 {
- result = n1
- }
- if n1 < n2 {
- result = n2
- }
- return result
- }
- var result = max(5, 100)
- fmt.Println("max return", result)
立即執行函數
由于 go 中的函數是 『一等公民』,可以在聲明之后立即執行,就是在函數聲明結束后,直接加上一個括號,表示該函數會立即執行,執行之后的結果可以通過變量進行接收。
- import "math"
- var Pi = func () float64 {
- return math.Pi
- }()
- fmt.Println("PI =",Pi)
閉包
“閉包就是能夠讀取其他函數內部變量的函數。在本質上,閉包是將函數內部和函數外部連接起來的橋梁。 ——百度百科
上面的描述來自百度百科,初次看概念比較難理解,如果站在使用的角度來說,閉包就是在一個函數調用后,返回另一個匿名函數,并保持當前函數內的局部變量,可以給匿名函數引用。
下面我們可以簡單實現一個迭代器函數,該函數接受一個切片,返回一個匿名函數,該匿名函數每次執行,都會取出切片的一個值,直到全部讀取。
- func generate(slice []int) func() (bool, int) {
- i := 0
- length := len(slice)
- return func () (bool, int) {
- if i >= length {
- return true, 0
- }
- var result = slice[i]
- i++
- return false, result
- }
- }
- func main() {
- slice := []int{1, 2, 3, 4, 5}
- nextNum := generate(slice)
- done, result := nextNum()
- // 直到 done 不等于 false,才停止
- for done == false {
- fmt.Println(result, done)
- done, result = nextNum()
- }
- fmt.Println(result, done)
- }
指針
我們前面常說的變量指的一般是一個值,指針是指向該變量存儲在內存的位置。指針也可以存儲在一個變量中,該變量稱為『指針變量』。
指針變量聲明
聲明指針變量時,需要指針指向哪一種類型,因為不同類型的值在內存占用的空間大小不一樣,僅僅知道內存地址還是不夠,還需要知道該變量在內存中占用多大空間。聲明指針變量只需要在類型前,加上 * 即可。
- var point *int // 聲明 int 類型的指針
指針變量賦值
給指針變量賦值,需要在對應類型的變量前加上&符號,表示取出該變量的地址。
- var i = 1
- var point *int
- point = &i
值傳遞與引用傳遞
一般情況下,我們傳入函數的參數僅為變量的值,這樣的傳遞被稱為值傳遞,在函數內對參數修改也不會影響到外部變量。
- func addOne(slice []int, number int) {
- slice = append(slice, number)
- fmt.Println("inner slice =", slice)
- }
- slice := []int{1, 2, 3}
- addOne(slice, 100)
- fmt.Println("outer slice =", slice)
上述代碼中,我們寫了一個函數,會對傳入的切片追加一個值,調用之后,我們會發現外部切片的值并沒有發生變量。
如果需要外部變量的值會跟隨函數調用發生變化,就需要將變量的指針傳入函數中,這樣的傳遞被稱為引用傳遞。這樣在函數中修改參數就會影響到外部的變量了。
- // 此時 slice 為指針變量
- func addOne(slice *[]int, number int) {
- // 通過 *slice 可以取出 slice 指針對應的值
- *slice = append(*slice, number)
- fmt.Println("inner slice =", *slice)
- }
- slice := []int{1, 2, 3}
- addOne(&slice, 100)
- fmt.Println("outer slice =", slice)