首页 > 编程笔记 > Go语言笔记

Go语言将函数作为返回值

本节将通过示例来演示一下如何使用闭包将函数作为返回值,代码如下所示:
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: %v\n", p2(3))
    // make a special Adder function, a gets value 3:
    TwoAdder := Adder(2)
    fmt.Printf("The result is: %v\n", 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.

这样闭包函数就能够被应用到整个集合的元素上,并修改它们的值。然后这些变量就可以用于表示或计算全局或平均值。

所有教程

优秀文章