甜点cc

V1

2022/12/07阅读:15主题:蔷薇紫

Go | 闭包的理解与使用

闭包基本介绍

闭包就是 一个函数 和其相关的 引用环境 组合的一个整体

好处: 保存引用的变量,下次继续使用,不会销毁

下面通过闭包的方式,写一个数字累加器,体验一下闭包的妙处👇

闭包实现数字累加

package main

import "fmt"

// 累加器
// 闭包 - 函数柯里化
// 返回值类型: func(int) int
func AddUpper() func(int) int {
 var n int = 100
 return func(i int) int {
  n = n + i
  return n
 }
}

func main() {
 f := AddUpper()
 fmt.Println(f(1)) // 101
 fmt.Println(f(2)) // 103
 fmt.Println(f(3)) // 106
}

代码说明

  1. AddUpper是一个函数,返回的数据类型 func(int) int

  2. 闭包的说明

返回的是一个匿名函数,但是这个匿名函数引用到了函数外的变量 n ,因此这个匿名函数就和 n 形成一个整体,构成闭包

  1. 当反复调用 f 函数时,因为 n 只初始化一次,保存了变量的值,因此每调用一次就相当于进行了累加。

  2. 我们要搞清楚闭包,关键就是要分析返回的函数使用到哪些变量

代码分析

这里我引入了一个字符串变量str,来帮助分析闭包是怎么保存变量的。

package main

import (
 "fmt"
)

// 累加器
// 闭包 - 函数柯里化
// 返回值类型: func(int) int
func AddUpper() func(int) int {
 var n int = 100
 var str = "hello"
 return func(i int) int {
  n = n + i
  fmt.Println("i=", i)
  str += string(36// ascii 36 = '$'
  fmt.Printf("str==%s\n", str)
  return n
 }
}

func main() {
 f := AddUpper()
 // fmt.Println(AddUpper()(1)) // 101
 fmt.Println("f(1)=", f(1)) // 101
 fmt.Println("f(2)=", f(2)) // 103
 fmt.Println("f(3)=", f(3)) // 106
}


i= 1
str==hello$
f(1)= 101
i= 2
str==hello$$
f(2)= 103
i= 3
str==hello$$$
f(3)= 106

从输出可以看出来,闭包引用的变量nstr并没有在调用函数的时候重复声明,而是保留了下次函数调用后更新的值。

闭包案例

需求:

  1. 编写一个函数 makeSuffix(suffix string) ,可以接收一个文件后缀名,并返回一个闭包

  2. 调用闭包,可以传入一个文件名,如果该文件名没有指定后缀,则返回 文件名.jpg ,如果已经有.jpg,则返回原文件名。

strings.HasSuffix,该函数可以判断某个字符串是否有指定的后缀。

上代码

package main

import (
 "fmt"
 "strings"
)

func makeSuffix(suffix string) func(string) string {
 return func(name string) string {
  if !strings.HasSuffix(name, suffix) {
   return name + suffix
  }
  return name
 }
}

// 传统写法
func makeSuffixV2(suffix string, name string) string {
 if !strings.HasSuffix(name, suffix) {
  return name + suffix
 }
 return name
}

func main() {
 // 闭包调用
 f := makeSuffix(".jpg")
 fmt.Println(f("xiao"))         // xiao.jpg
 fmt.Println(f("xiaoxiao.jpg")) // xiaoxiao.jpg
 fmt.Println(f("xiaoxiao.666")) // xiaoxiao.666.jpg

 // 传统写法调用
 fmt.Println("makeSuffixV2=", makeSuffixV2(".jpg""allblue"))  // makeSuffixV2= allblue.jpg
 fmt.Println("makeSuffixV2=", makeSuffixV2(".jpg""all.blue")) // makeSuffixV2= all.blue.jpg
}

代码说明

返回的匿名函数和 makeSuffix(suffix string) 的 suffix 变量组合成一个闭包

传统写法和闭包写法实现效果一样,但是,传统写法需要重复写变量, 比如上面的

makeSuffixV2(".jpg", "all.blue"))

闭包则解决了这个问题,是代码看起来更加的简洁

闭包的好处之一: 参数复用

好处: 保存引用的变量,下次继续使用,不会销毁

函数柯里化(闭包)

参考我的另一篇文章: 甜点cc的语雀知识库

我是 甜点cc

微信公众号:【看见另一种可能】

专注前端开发,也喜欢专研各种跟本职工作关系不大的技术,技术、产品兴趣广泛且浓厚。本号主要致力于分享个人经验总结,希望可以给一小部分人一些微小帮助。

分类:

后端

标签:

Golang

作者介绍

甜点cc
V1