博客 / 詳情

返回

Go 專欄|函數那些事

原文鏈接: Go 專欄|函數那些事

曾經很長一段時間,我都為自己是互聯網科技公司的一員而感到自豪,我覺得我們與眾不同。

我們的管理更扁平化,沒有那麼多官僚主義,充滿活力,朝氣蓬勃。而且我們的產品正在改變大家的衣食住行,我們正在改變世界。

但近幾年發生的一系列事件,都讓我的信心產生動搖,不停在捶打我:醒醒吧,兄弟,事實不是你想象的那樣。

我能做些什麼呢?不知道。

還是努力更文吧,爭取早日不做打工人。

函數定義

函數包括以下幾個部分:關鍵詞 func,函數名,參數列表,返回列表和函數體。

func name(param-list) ret-list {
    body
}

函數可以沒有參數,也可以沒有返回值。

func funcA() {
    fmt.Println("i am funcA") // i am funcA
}

函數的類型稱作函數簽名,當兩個函數的參數列表和返回列表相同時,則兩個函數的類型或簽名就相同。

func add(x int, y int) int {
    return x + y
}

func sub(x int, y int) (z int) {
    z = x - y
    return
}

fmt.Printf("%T\n", add) // func(int, int) int
fmt.Printf("%T\n", sub) // func(int, int) int

參數

多個相鄰類型的參數可以使用簡寫模式,所以剛才的 addsub 函數還可以這樣寫:

func add(x, y int) int {
    return x + y
}

func sub(x, y int) (z int) {
    z = x - y
    return
}

支持不定參數,使用 ...type 語法。注意不定參數必須是函數的最後一個參數。

func funcSum(args ...int) (ret int) {
    for _, arg := range args {
        ret += arg
    }
    return
}

// 不定參數
fmt.Println(funcSum(1, 2))    // 3
fmt.Println(funcSum(1, 2, 3)) // 6

也可以使用 slice 作為實參傳入,需要使用 ... 將 slice 展開:

// slice 參數
s := []int{1, 2, 3, 4}
fmt.Println(funcSum(s...)) // 10

其實,使用 slice 作為形參同樣可以達到相同的效果,但區別就是傳參的時候,必須要構造出來一個 slice 才行,沒有不定參數使用起來方便。

func funcSum1(args []int) (ret int) {
    for _, arg := range args {
        ret += arg
    }
    return
}

fmt.Println(funcSum1(s))   // 10

返回值

函數可以返回一個值,也可以返回多個值。

// 多返回值
func swap(x, y int) (int, int) {
    return y, x
}

// 多返回值
fmt.Println(swap(1, 2)) // 2 1

如果有不需要的返回值,使用 _ 將其忽略:

x, _ := swap(1, 2)
fmt.Println(x) // 2

支持命名返回值。使用命名返回值的話,直接使用 return 即可,後面不用跟返回值名。

前面不定參數的例子就是通過這種方式來寫的:

func funcSum(args ...int) (ret int) {
    for _, arg := range args {
        ret += arg
    }
    return
}

再來對比一下,如果不是採用命名返回值,應該怎麼寫:

func funcSum(args ...int) int {
    ret := 0
    for _, arg := range args {
        ret += arg
    }
    return ret
}

匿名函數

匿名函數是指不需要定義函數名的一種函數實現方式。可以直接賦值給函數變量,可以當作實參,也可以作為返回值,還可以直接調用。

// 匿名函數
sum := func(a, b int) int { return a + b }
fmt.Println(sum(1, 2)) // 3

作為參數:

// 匿名函數作為參數
func funcSum2(f func(int, int) int, x, y int) int {
    return f(x, y)
}

fmt.Println(funcSum2(sum, 3, 5)) // 8

作為返回值:

// 匿名函數作為返回值
func wrap(op string) func(int, int) int {
    switch op {
    case "add":
        return func(a, b int) int {
            return a + b
        }
    case "sub":
        return func(a, b int) int {
            return a + b
        }

    default:
        return nil
    }
}

f := wrap("add")
fmt.Println(f(2, 4)) // 6

直接調用:

// 直接調用
fmt.Println(func(a, b int) int { return a + b }(4, 5)) // 9

總結

函數在之前的文章中已經使用過了,這篇再系統全面總結一下都有哪些需要注意的點。

包括函數定義,參數,返回和匿名函數。其實還有一個閉包,通過匿名函數來實現。但我感覺閉包使用的並不是很多,就沒有寫,感興趣的同學可以自己搜搜看。

函數可以把複雜的程序分成更小的模塊,使程序可讀性更強,複用性更高,維護性更好。在開發過程中一定要具備將特定功能抽象成函數的能力,而不是將所有代碼都寫在一起,代碼堆成一坨。這樣的代碼除了不好維護,重點是時間長了自己都不想看。


文章中的腦圖和源碼都上傳到了 GitHub,有需要的同學可自行下載。

地址: https://github.com/yongxinz/g...

關注公眾號 AlwaysBeta,回覆「goebook」領取 Go 編程經典書籍。

Go 專欄文章列表:

  1. Go 專欄|開發環境搭建以及開發工具 VS Code 配置
  2. Go 專欄|變量和常量的聲明與賦值
  3. Go 專欄|基礎數據類型:整數、浮點數、複數、布爾值和字符串
  4. Go 專欄|複合數據類型:數組和切片 slice
  5. Go 專欄|複合數據類型:字典 map 和 結構體 struct
  6. Go 專欄|流程控制,一網打盡
  7. Go 專欄|函數那些事
  8. Go 專欄|錯誤處理:defer,panic 和 recover
  9. Go 專欄|説説方法
  10. Go 專欄|接口 interface
user avatar
0 位用戶收藏了這個故事!

發佈 評論

Some HTML is okay.