位元詩人 [Golang] 程式設計教學:撰寫函式 (Function)

Golang函式
Facebook Twitter LinkedIn LINE Skype EverNote GMail Yahoo Email

前言

在本教程先前的文章中,大部分的程式碼都是放在主函式 (main function) 中,然而,隨著程式碼規模上升,這樣的方式漸不敷使用,像是程式碼上下捲動不易維護,也無法將程式碼分離在不同檔案中。程式設計者撰寫函式 (function) 以分離程式碼,函式也是程式碼共用的基礎。物件導向的方法 (method) 也是函式為基礎。

使用函式

到目前為止,我們已經使用一些函式,像是在 fmt 套件的 Println 函式,可在終端機中印出文字,math/rand 套件的 Intn 可以產生隨機的整數等。讀者會發現 Go 的公開函式都是以大寫開頭,這是 Go 語言的一項設計,使用大小寫來控管套件中函式和方法的權限,只有字母大寫的函式或方法才會公開給外界使用。

撰寫函式

以下範例撰寫一個簡單的 hello 函式,並呼叫三次:

package main

import (
    "fmt"
)

func hello() {
    fmt.Println("Hello World")
}

func main() {
    hello()
    hello()
    hello()
}

雖然這個函式相對簡單,我們可以從這個範例看出來,函式可以重覆再利用。

函式也可以加入參數 (parameter),透過參數,我們可以改變函式的行為;如以下範例:

package main

import (
    "fmt"
)

func hello(name string) {
    fmt.Println(fmt.Sprintf("Hello %s", name))
}

func main() {
    hello("Michelle")
    hello("Jenny")
    hello("Tommy")
}

函式運作後,也可以有回傳值 (return value),如以下範例:

package main

import (
    "log"
    "math"
)

func add(a float64, b float64) float64 {
    return a + b
}

func main() {
    if !(math.Abs(add(3, 2)-5.0) < (1.0 / 1000000)) {
        log.Fatal("Wrong value")
    }
}

多個回傳值

Go 函式允許多個回傳值,如以下實例:

package main

import (
    "log"
)

func divmod(a int, b int) (int, int) {
    return a / b, a % b
}

func main() {
    m, n := divmod(5, 3)

    if !(m == 1) {
        log.Fatal("Wrong value m")
    }

    if !(n == 2) {
        log.Fatal("Wrong value n")
    }
}

多回傳值時常用於錯誤處理 (error handling) 中,我們將於後續文章中介紹。

改變參數的函式

搭配指標,函式也可以改變參數。在我們這個例子中,我們以函式搭配結構撰寫一個 C 風格的 Point 物件:

package main

import (
    "log"
)

type Point struct {
    x float64
    y float64
}

func Point_new(x float64, y float64) *Point {
    p := new(Point)

    p.x = x
    p.y = y

    return p
}

func Point_get_x(p *Point) float64 {
    return p.x
}

func Point_get_y(p *Point) float64 {
    return p.y
}

func Point_set_x(p *Point, x float64) {
    p.x = x
}

func Point_set_y(p *Point, y float64) {
    p.y = y
}

func main() {
    p := Point_new(0, 0)

    if !(Point_get_x(p) == 0) {
        log.Fatal("Wrong value")
    }

    if !(Point_get_y(p) == 0) {
        log.Fatal("Wrong value")
    }

    Point_set_x(p, 3.0)
    Point_set_y(p, 4.0)

    if !(Point_get_x(p) == 3.0) {
        log.Fatal("Wrong value")
    }

    if !(Point_get_y(p) == 4.0) {
        log.Fatal("Wrong value")
    }
}

由於 Go 支援物件的語法,實務上,我們不會用這種方法撰寫物件,本範例只是展示如何在函式中使用指標。

不定長度參數

細心的讀者可能會發現,Go 有些函式的參數長度是不固定的,像是 fmt 套件的 PrintfSprintf 函式,這兩個函式,第一個參數是做為模板的字串,第二個以後的參數就是我們要填入的值,而值的數量不是固定的。這是由於 Go 支援不定長度參數的特性,如以下範例:

package main

import (
    "log"
)

func sum(args ...float64) float64 {
    sum := 0.0

    for _, e := range args {
        sum += e
    }

    return sum
}

func main() {
    s := sum(1, 2, 3, 4, 5)

    if !(s == 15.0) {
        log.Fatal("Wrong value")
    }
}

不定參數傳入函式後,參數本身是一個切片,在函式中透過此切片即可取得個別的參數值。

模擬預設變數

Go 的函式本身不支援預設變數,但我們可以透過傳入結構的方式模擬預設變數,如下例:

package main

import (
    "log"
)

type Color int

const (
    White Color = iota
    Black
    Green
    Yellow
)

type Size int

const (
    Large Size = iota
    Middle
    Small
    ExtraLarge
)

type Clothes struct {
    color Color
    size  Size
}

type Param struct {
    Color Color
    Size  Size
}

func MakeClothes(param Param) *Clothes {
    c := new(Clothes)

    c.color = param.Color
    c.size = param.Size

    return c
}

func main() {
    // Clothes with custom parameters
    c1 := MakeClothes(Param{Color: Black, Size: Middle})

    if !(c1.color == Black) {
        log.Fatal("Wrong color")
    }

    if !(c1.size == Middle) {
        log.Fatal("Wrong size")
    }

    // Clothes with default parameters
    c2 := MakeClothes(Param{})

    if !(c2.color == White) {
        log.Fatal("Wrong color")
    }

    if !(c2.size == Large) {
        log.Fatal("Wrong size")
    }
}

有些讀者可能會想到用不定長度參數模擬預設參設,但筆者認為這不是好的模式,使用不定長度參數的話,需要多寫許多程式去檢查不同長度時的情境,而且參數位置是寫死的,無法像結構般靈活。

函式重載

Go 也不支援函式重載,用上一節的方式也可用來模擬函式重載,或是直接以不同函式來命名即可,如下例:

package main

import (
    "log"
)

func add(a float64, b float64) float64 {
    return a + b
}

func addOne(a float64) float64 {
    return add(a, 1)
}

func main() {
    if !(addOne(3) == 4) {
        log.Fatal("Wrong value")
    }
}

傳值呼叫 vs. 傳址呼叫

基本上,如同 C 語言,所有 Go 的函式都是傳值呼叫 (call by value),而這個值有可能是基本型別或指標或其他型別。當我們傳遞指標時,我們會拷貝指標的位址,但不會拷貝指標所指向的值;當值很大時,傳遞指標比傳整個值有效率。所以,其實沒有傳址呼叫 (call by adress),這只是對初學指標的學習者易於記憶的一種詞語。另外,Go 沒有 C++ 的傳參考呼叫 (call by reference)。

init 函式

init 函式是一個特殊的函式,若程式碼內有 init 會在程式一開始執行的時候呼叫該函式,順序在 main 函式之前。通常 init 函式都是用來初始化一些外部資源。以下是一個摘自 Go 官方網站的例子:

func init() {
    if user == "" {
        log.Fatal("$USER not set")
    }
    if home == "" {
        home = "/home/" + user
    }
    if gopath == "" {
        gopath = home + "/go"
    }
    // gopath may be overridden by --gopath flag on command line.
    flag.StringVar(&gopath, "gopath", gopath, "override default GOPATH")
}
關於作者

身為資訊領域碩士,位元詩人 (ByteBard) 認為開發應用程式的目的是為社會帶來價值。如果在這個過程中該軟體能成為永續經營的項目,那就是開發者和使用者雙贏的局面。

位元詩人喜歡用開源技術來解決各式各樣的問題,但必要時對專有技術也不排斥。閒暇之餘,位元詩人將所學寫成文章,放在這個網站上和大家分享。