Go語言將函數作為返回值

2020-07-16 10:05:13
本節將通過範例來演示一下如何使用閉包將函數作為返回值,程式碼如下所示:
package main
import "fmt"

func main() {
    // make an Add2 function, give it a name p2, and call it:
    p2 := Add2()
    fmt.Printf("Call Add2 for 3 gives: %vn", p2(3))
    // make a special Adder function, a gets value 3:
    TwoAdder := Adder(2)
    fmt.Printf("The result is: %vn", TwoAdder(3))
}
func Add2() func(b int) int {
    return func(b int) int {
        return b + 2
    }
}
func Adder(a int) func(b int) int {
    return func(b int) int {
        return a + b
    }
}
輸出結果如下所示:

Call Add2 for 3 gives: 5
The result is: 5

在上述程式碼中我們看到了函數 Add2 和 Adder 均會返回簽名為 func(b int) int 的函數:

func Add2() (func(b int) int)
func Adder(a int) (func(b int) int)

函數 Add2 不接受任何引數,但函數 Adder 接受一個 int 型別的整數作為引數,我們也可以將 Adder 返回的函數存到變數中。

下例為一個略微不同的實現:
package main
import "fmt"

func main() {
    var f = Adder()
    fmt.Print(f(1), " - ")
    fmt.Print(f(20), " - ")
    fmt.Print(f(300))
}
func Adder() func(int) int {
    var x int
    return func(delta int) int {
        x += delta
        return x
    }
}
函數 Adder() 現在被賦值到變數 f 中(型別為 func(int) int )。

輸出結果如下所示:

1 - 21 - 321

三次呼叫函數 f 的過程中函數 Adder() 中變數 delta 的值分別為:1、20 和 300。

我們可以看到,在多次呼叫中,變數 x 的值是被保留的,即 0 + 1 = 1 ,然後 1 + 20 = 21,最後 21 + 300 = 321,閉包函數會儲存並積累其中變數的值,不管外部函數退出與否,它都能夠繼續操作外部函數中的區域性變數。

這些區域性變數同樣可以是引數,例如之前例子中的 Adder(as int)。

這些例子清楚地展示了如何在Go語言中使用閉包,在閉包中使用到的變數可以是在閉包函數體內宣告的,也可以是在外部函數宣告的:

var g int
go func(i int) {
    s := 0
    for j := 0; j < i; j++ { s += j }
    g = s
}(1000) // Passes argument 1000 to the function literal.

這樣閉包函數就能夠被應用到整個集合的元素上,並修改它們的值。然後這些變數就可以用於表示或計算全域性或平均值。