Go學習筆記04-函數

目錄

• 函數定義
• 函數示例
• 小結

函數定義

函數定義與變數定義相似，

func function_name(var1, var2, var3, ...) (return_type1, return_type1, ...) {
    //function body
}

函數示例

package main

import (
    "fmt"
    "math"
    "reflect"
    "runtime"
)

//函數可以有多個參數，多個返回值
func eval(a, b int, op string) (int, error) {
    switch op {
    case "+":
        return a + b, nil
    case "-":
        return a - b, nil
    case "*":
        return a * b, nil
    case "/":
        //函數div()有兩個返回值，有一個返回值沒用時可以用"_"代替
        q, _ := div(a, b)
        return q, nil
    default:
        return 0, fmt.Errorf(
            "unsupported operation: %s", op)
    }
}

//可以給返回值起名，但是只建議在很簡單的函數中使用
func div(a, b int) (q, r int) {
    return a / b, a % b
}

//函數可以作為另一個函數的參數
func apply(op func(int, int) int, a, b int) int {
    
    //獲取函數的指針
    p := reflect.ValueOf(op).Pointer()
    //獲取函數名
    opName := runtime.FuncForPC(p).Name()
    fmt.Printf("Calling function %s with args "+
        "(%d, %d)\n", opName, a, b)

    return op(a, b)
}

//定義可變參數
func sum(numbers ...int) int {
    s := 0
    //range獲取可變參數範圍
    for i := range numbers {
        s += numbers[i]
    }
    return s
}

func swap(a, b int) (int, int) {
    return b, a
}

func main() {
    fmt.Println("Error handling")
    if result, err := eval(3, 4, "x"); err != nil {
        fmt.Println("Error:", err)
    } else {
        fmt.Println(result)
    }
    q, r := div(13, 3)
    fmt.Printf("13 div 3 is %d mod %d\n", q, r)
    
    //函數式編程，可以在表達式里直接定義一個匿名函數
    fmt.Println("pow(3, 4) is:", apply(
        func(a int, b int) int {
            return int(math.Pow(
                float64(a), float64(b)))
        }, 3, 4))

    fmt.Println("1+2+...+5 =", sum(1, 2, 3, 4, 5))

    a, b := 3, 4
    a, b = swap(a, b)
    fmt.Println("a, b after swap is:", a, b)
}

小結

• 函數用關鍵字 func 定義。
• 定義函數時，函數名在前，返回類型在後。
• 一個函數可以有多個返回值。
• 返回值可以指定一個名字（僅用於非常簡單的函數 ）。
• 函數可以作為參數。
• Go語言中函數沒有預設參數、可選參數，函數重載