首页 > 文章列表 > 如何理解 Golang 中函数类型的高阶函数?

如何理解 Golang 中函数类型的高阶函数?

高阶函数 函数类型
257 2024-04-23

Golang 高阶函数可接受和返回函数。它们分两类:接收函数作为参数:处理其他函数或执行动态程序。返回函数作为返回值:创建和返回可存储和后期执行的函数。

如何理解 Golang 中函数类型的高阶函数?

理解 Golang 高阶函数

Golang 中的高阶函数是一种可以接受和返回函数作为参数或返回值的函数。这为代码的重用性、抽象性和可维护性提供了强大的潜力。

高阶函数的种类

Golang 中有两种主要类型的高阶函数:

  • 接受函数作为参数:这些函数将其他函数作为输入处理,允许动态执行。
  • 返回函数作为返回值:这些函数创建并返回可以存储并用于后期执行的函数。

实战案例

接受函数作为参数

func mapFunc(fn func(int) int, nums []int) []int {
    result := make([]int, len(nums))
    for i, num := range nums {
        result[i] = fn(num)
    }
    return result
}

func main() {
    nums := []int{1, 2, 3, 4, 5}
    squaredNums := mapFunc(func(num int) int {
        return num * num
    }, nums)
    fmt.Println(squaredNums) // 输出: [1 4 9 16 25]
}

在这里,mapFunc 接受一个函数 fn 作为参数,并将它应用于每个元素。fn 将每个元素平方并返回一个新数组。

返回函数作为返回值

func genAdder(x int) func(int) int {
    return func(y int) int {
        return x + y
    }
}

func main() {
    add5 := genAdder(5)
    result := add5(10)
    fmt.Println(result) // 输出:15
}

在此示例中,genAdder 返回一个闭包函数,该闭包函数能够捕获并使用外部变量 xadd5 函数被调用,它将 10 作为参数并返回 15。