• 【Go语言学习】匿名函数与闭包


    前言

    入坑 Go 语言已经大半年了,却没有写过一篇像样的技术文章,每次写一半就搁笔,然后就烂尾了。

    1.gif

    几经思考,痛定思痛,决定金盆洗手,重新做人,哦不,重新开始写技术博文。

    这段时间在研究Go语言闭包的过程中,发现了很多有意思的东西,也学到了不少内容,于是便以次为契机,重新开始技术文章的输出。

    什么是闭包

    闭包Go 语言中一个重要特性,也是 函数式编程 中必不可少的角色。那么什么是 闭包 呢?

    A closure is a function value that references variables from outside its body.

    这是 A Tour of Go 上的定义,闭包 是一种引用了外部变量的函数。但我觉得这个定义还不够准确,闭包 应该是引用了外部变量的 匿名函数

    看了很多文章,大多把 闭包匿名函数混淆在了一起,也有很多人说,闭包 其实就是匿名函数,但其实两者是不能直接划等号的。

    闭包 是一种特殊的匿名函数,是匿名函数的子集。所以在说 闭包 之前,我们先来看看 匿名函数 吧。

    匿名函数

    匿名函数 顾名思义,就是没有名字的函数。在Go语言中,函数是一等公民,也就是说,函数可以被赋值或者当作返回值和参数进行传递,在很多时候我们并不需要一个有名字的函数(而且命名确实是一项相当费劲的事),所以我们在某些场景下可以选择使用 匿名函数

    举个例子:

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

    这是一个简单的例子,我们声明了一个 匿名函数 ,然后把它赋值给一个叫 hello 的变量,然后我们就能像调用函数那样使用它了。

    这跟下面的代码效果是一样的:

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

    我们还可以把 匿名函数 当作函数参数进行传递:

    func main(){
        doPrint("Hello World", func(s string){
    		fmt.Println(s)
    	})
    }
    
    type Printer func(string)
    
    func doPrint(s string, printer Printer){
        printer(s)
    }
    

    或者当作函数返回值进行返回:

    func main(){
        getPrinter()("Hello World")
    }
    
    type Printer func(string)
    
    func getPrinter()Printer{
        return func(s string){
    		fmt.Println(s)
    	}
    }
    

    匿名函数 跟普通函数在绝大多数场景下没什么区别,普通函数的函数名可以当作是与该函数绑定的函数常量。

    一个函数主要包含两个信息:函数签名和函数体,函数的签名包括参数类型,返回值的类型,函数签名可以看做是函数的类型,函数的函数体即函数的值。所以一个接收匿名函数的变量的类型便是由函数的签名决定的,一个匿名函数被赋值给一个变量后,这个变量便只能接收同样签名的函数。

    func main(){
        hello := func(){
            fmt.Println("Hello World")
        } // 给 hello 变量赋值一个匿名函数
        hello()
        
        hello = func(){
            fmt.Println("Hello World2")
        } // 重新赋值新的匿名函数
        hello()
        
        hello = hi // 将一个普通函数赋值给 hello
        hello()
        
        hello = func(int){
            fmt.Println("Hello World3")
        } // 这里编译器会报错
        hello()
    }
    
    func hi(){
        fmt.Println("Hi")
    }
    

    匿名函数 跟普通函数的微小区别在于 匿名函数 赋值的变量可以重新设置新的 匿名函数,但普通函数的函数名是与特定函数绑定的,无法再将其它函数赋值给它。这就类似于变量与常量之间的区别。

    闭包的特性

    说完了 匿名函数,我们再回过头来看看 闭包

    闭包 是指由一个拥有许多变量和绑定了这些变量的环境的 匿名函数
    闭包 = 函数 + 引用环境

    听起来有点绕,什么是 引用环境呢?

    引用环境 是指在程序执行中的某个点所有处于活跃状态的变量所组成的集合。

    由于闭包把函数和运行时的引用环境打包成为一个新的整体,所以就解决了函数编程中的嵌套所引发的问题。

    当每次调用包含闭包的函数时都将返回一个新的闭包实例,这些实例之间是隔离的,分别包含调用时不同的引用环境现场。不同于函数,闭包在运行时可以有多个实例,不同的引用环境和相同的函数组合可以产生不同的实例。

    简单来说,闭包 就是引用了外部变量的匿名函数。不太明白?没关系,让我们先来看一个栗子:

    func adder() func() int {
    	var i = 0
    	return func() int {
    		i++
    		return i
    	}
    }
    

    这是用闭包实现的简单累加器,这一部分便是闭包,它引用在其作用域范围之外的变量i。

    func() int {
        i++
        return i
    }
    

    可以这样使用:

    func main() {
    	a := adder()
    	fmt.Println(a())
    	fmt.Println(a())
    	fmt.Println(a())
    	fmt.Println(a())
        b := adder()
    	fmt.Println(b())
    	fmt.Println(b())
    }
    

    输出如下:

    1
    2
    3
    4
    1
    2
    

    上述例子中,adder 是一个函数,没有入参,返回值是一个返回 int 类型的无参函数,也就是说调用 adder 函数会返回一个函数,这个函数的返回值是 int 类型,且不接收参数。

    main 方法中:

    a := adder()
    

    这里是将调用后得到的函数赋值给了变量 a ,随后进行了四次函数调用和输出:

    fmt.Println(a())
    fmt.Println(a())
    fmt.Println(a())
    fmt.Println(a())
    

    也许你还是会感到困惑,iadder 函数里的变量,调用完成之后变量的生命周期不久结束了吗?为什么还能不断累加?

    这就涉及到闭包的另一个重要话题了:闭包 会让被引用的局部变量从栈逃逸到堆上,从而使其能在其作用域范围之外存活。闭包 “捕获”了和它在同一作用域的其它常量和变量。这就意味着当闭包被调用的时候,不管在程序什么地方调用,闭包能够使用这些常量或者变量。它不关心这些捕获了的变量和常量是否已经超出了作用域,只要闭包还在使用它们,这些变量就还会存在。

    匿名函数和闭包的使用

    可以利用匿名函数闭包可以实现很多有意思的功能,比如上面的累加器,便是利用了 闭包 的作用域隔离特性,每调用一次 adder 函数,就会生成一个新的累加器,使用新的变量 i,所以在调用 b() 时,仍旧会从1开始输出。

    再来看几个匿名函数闭包应用的例子。

    工厂函数

    工厂函数即生产函数的函数,调用工厂函数可以得到其内嵌函数的引用,每次调用都可以得到一个新的函数引用。

    func getFibGen() func() int {
    	f1 := 0
    	f2 := 1
    	return func() int {
    		f2, f1 = f1 + f2, f2
    		return f1
    	}
    }
    
    func main() {
    	gen := getFibGen()
    	for i := 0; i < 10; i++ {
    		fmt.Println(gen())
    	}
    }
    

    上面是利用闭包实现的函数工厂来求解斐波那契数列问题,调用 getFibGen 函数之后,gen 便获得了内嵌函数的引用,且该函数引用里一直持有 f1f2 的引用,每执行一次 gen(),便会运算一次斐波那契的递推关系式:

    func() int {
        f2, f1 = f1 + f2, f2
        return f1
    }
    

    输出如下:

    1
    1
    2
    3
    5
    8
    13
    21
    34
    55
    

    由于闭包能构造出单独的变量环境,可以很好的实现环境隔离,所以很适合应用于函数工厂,在实现功能时保存某些状态变量。

    装饰器/中间件

    修饰器是指在不改变对象的内部结构情况下,动态地扩展对象的功能。通过创建一个装饰器,来包装真实的对象。使用闭包很容易实现装饰器模式

    在 gin 中的 Middleware 便是使用装饰器模式来实现的。比如我们可以这样实现一个自定义的 Logger:

    func Logger() gin.HandlerFunc {
    	return func(context *gin.Context) {
    		host := context.Request.Host
    		url := context.Request.URL
    		method := context.Request.Method
    		fmt.Printf("%s::%s 	 %s 	 %s 
    ", time.Now().Format("2006-01-02 15:04:05"), host, url, method)
    		context.Next()
            fmt.Println("response status: ", context.Writer.Status())
    	}
    }
    

    这是在 gin 中利用 匿名函数 实现的自定义日志中间件,在 gin 中,类似的用法十分常见。

    defer

    这是匿名函数闭包最常用的地方,我们会经常在 defer 函数中使用匿名函数闭包来做释放锁,关闭连接,处理 panic 等函数善后工作。

    func main() {
        defer func() {
            if ok := recover(); ok != nil {
                fmt.Println("recover from panic")
            }
        }()
    
        panic("error")
    }
    

    gorutine

    匿名函数闭包还有一个十分常用的场景,那便是在启动 gorutine 时使用。

    func main(){
        go func(){
            fmt.Println("Hello World")
        }()
        time.Sleep(1 * time.Second)
    }
    

    重新声明一下,在函数内部引用了外部变量便是闭包,否则就是匿名函数

    func main(){
        hello := "Hello World"
        go func(){
            fmt.Println(hello)
        }()
        time.Sleep(1 * time.Second)
    }
    

    context

    在cancelContext中也使用到了闭包:

    // A CancelFunc tells an operation to abandon its work.
    // A CancelFunc does not wait for the work to stop.
    // A CancelFunc may be called by multiple goroutines simultaneously.
    // After the first call, subsequent calls to a CancelFunc do nothing.
    type CancelFunc func()
    
    // WithCancel returns a copy of parent with a new Done channel. The returned
    // context's Done channel is closed when the returned cancel function is called
    // or when the parent context's Done channel is closed, whichever happens first.
    //
    // Canceling this context releases resources associated with it, so code should
    // call cancel as soon as the operations running in this Context complete.
    func WithCancel(parent Context) (ctx Context, cancel CancelFunc) {
    	c := newCancelCtx(parent)
    	propagateCancel(parent, &c)
    	return &c, func() { c.cancel(true, Canceled) }
    }
    
    // newCancelCtx returns an initialized cancelCtx.
    func newCancelCtx(parent Context) cancelCtx {
    	return cancelCtx{Context: parent}
    }
    

    闭包的陷阱

    闭包很好用,但在某些场景下,也十分具有欺骗性,稍有不慎,就会掉入其陷阱里。

    不如先来看一个例子:

    for j := 0; j < 2; j++ {
    	defer func() {
    		fmt.Println(j)
    	}()
    }
    

    你猜会输出什么?

    2
    2
    

    这是因为在 defer 中使用的闭包引用了外部变量 j

    闭包 中持有的是外部变量的引用

    这是很容易犯的错误,在循环体中使用 defer,来关闭连接,释放资源,但由于闭包内持有的是外部变量的引用,在这里持有的是变量 j 的引用,defer 会在函数执行完成前调用闭包,在开始执行闭包时,j 的值已经是2了。

    那么这个问题应该如何修复呢?有两种方式,一种是重新定义变量:

    for j := 0; j < 2; j++ {
        k := j
    	defer func() {
    		fmt.Println(k)
    	}()
    }
    

    在循环体里,每次循环都定义了一个新的变量 k 来获取原变量 j 的值,因此每次调用闭包时,引用的是不同的变量 k,从而达到变量隔离的效果。

    另一种方式是把变量当成参数传入:

    for j := 0; j < 2; j++ {
    	defer func(k int) {
    		fmt.Println(k)
    	}(j)
    }
    

    这里每次调用闭包时,传入的都是变量 j 的值,虽然 defer 仍会在函数执行完成前调用,但传入闭包的参数值却是先计算好的,因而能够正确输出。

    闭包返回的包装对象是一个复合结构,里面包含匿名函数的地址,以及环境变量的地址。

    为了更好的理解这一点,我们再来看一个例子:

    package main
    
    import "fmt"
    
    func main() {
        x, y := 1, 2
    
        defer func(a int) { 
            fmt.Printf("x:%d,y:%d
    ", a, y)  
        }(x)     
    
        x += 1
        y += 1
        fmt.Println(x, y)
    }
    

    输出如下:

    2 3
    x:1,y:3
    

    另外,由于闭包会使得其持有的外部变量逃逸出原有的作用域,所以使用不当可能会造成内存泄漏,这一点由于相当具有隐蔽性,所以也需要谨慎对待。

    总结

    闭包是一种特殊的匿名函数,是由函数体和引用的外部变量一起组成,可以看成类似如下结构:

    type FF struct {
    	F unitptr
    	A *int
    	B *int
    	X *int // 如果X是string/[]int,那么这里应该为*string,*[]int
    }
    

    在Go语言中,闭包的应用十分广泛,掌握了闭包的使用可以让你在写代码时能更加游刃有余,也可以避免很多不必要的麻烦。所以是必须要掌握的一个知识点。

    至此,关于闭包的内容就完结了,希望能对你有帮助。

  • 相关阅读:
    线段树合并
    bzoj 3675 [Apio2014]序列分割
    模版总结【长期更新】
    动态规划的题目总结(长期更新)
    搜索(另类状态BFS):NOIP 华容道
    贪心(模拟费用流):NOIP2011 观光公交
    基础算法(二分,贪心):NOIP 2012 疫情控制
    模拟(堆):USACO Jan11 瓶颈
    搜索(DLX重复覆盖模板):HDU 2295 Radar
    数学:lucas定理的总结
  • 原文地址:https://www.cnblogs.com/mfrank/p/13383467.html
Copyright © 2020-2023  润新知