Golang函数这些神操作你知道哪些
作者:TtrOps 时间:2024-04-26 17:22:15
「1. 用type关键字可以定义函数类型,函数类型变量可以作为函数的参数或返回值。」
package main
import "fmt"
func add(a, b int) int {
return a + b
}
func sub(a, b int) int {
return a - b
}
type Task func(int, int) int
func exec(t Task, a, b int) int {
return t(a, b)
}
func main() {
a := exec(add, 10, 20)
fmt.Println(a)
b := exec(sub, 100, 95)
fmt.Println(b)
}
解析:type Task func(int, int) int,这句是说,使用type关键字定义1个类型为func的Task,这个func有2个int形参、1个int返回值。再看exec这个函数,它有3个形参,形参t的类型是刚定义的函数类型Task,另外两个你懂的,我就不说了。
「2. 匿名函数的玩法是真的骚,看看骚在哪里」
栗子1:匿名函数可以直接赋给变量
func main() {
var aaa = func(a, b int) int {
return a + b
}
ret := aaa(89, 78)
fmt.Println(ret)
}
输出:
167
栗子2:匿名函数作为函数入参
这个例子感觉跟刚才使用type关键字定义函数类型的例子有点相同的赶脚,仔细一看,好像也只是在使用的方式上是一样,定义函数的套路又是不同的。
func work(f func(int, int) int, a, b int) int {
return f(a, b)
}
func add(a, b int) int {
return a + b
}
func main() {
a := work(add, 100, 200)
fmt.Println(a)
}
输出:
300
解析:形参f的类型就是匿名函数,继续看func(int, int) int,这个匿名函数接收两个int形参,返回值也是int类型。另外两个形参a和b,也是int,想必你懂了,我就不废话了。
栗子3:匿名函数作为函数出参
匿名函数作为函数出参(作为函数返回值),经不断调测,有3种骚包玩法
骚包玩法1
// 先来个简单的
func work() func() int {
return func() int {
return 10 + 20
}
}
func main() {
f := work()
ret := f()
fmt.Println(ret)
}
输出:
30
骚包玩法2
// 再改造下
func work(a, b int) func() int {
return func() int {
return a + b
}
}
func main() {
f := work(500, 20)
ret := f()
fmt.Println(ret)
}
输出:
520
骚包玩法3
// 再次改造
func work() func(int, int) int {
return func(a, b int) int {
return a + b
}
}
func main() {
f := work()
ret := f(600, 50)
fmt.Println(ret)
}
输出:
650
对3种骚包玩法的简单解析:
再这里就解析一下第3种玩法,能搞懂这个玩法,前面2个玩法,你就自然懂了。work函数没有入参,但是有出参(也可以说是返回值),出参是匿名函数func(int, int) int,这个定义在返回值里的匿名函数有2个int类型的形参(入参)和1个int类型的返回值。再看看函数体内部,没干很复杂活儿,而是直接return了匿名函数,函数体里的这个匿名函数是和定义在返回值里的类型保持一致的,也是接收了2个int的形参a和b,返回值也是int,啥也没干,就做了个相加。好郁闷!不知道你看懂了没?再看看是如何使用work函数的,关键点就在这里,调用work()赋给了变量f,这时候,f它就是一个函数了,再调用f(600, 50),想必你已经知道了为啥要传入2个int值,这下搞定!不知道把你绕晕没,这玩法确实很骚。
「3. 匿名函数和延迟调用」
延迟调用的规则是:按照「先进后出」的顺序,也就是说函数返回前是会被执行的,而且是按照先进后出的顺序。如何起到延迟的效果,是需要注册的,可通过defer关键字进行注册。那么什么场景下需要用到延迟调用呢?比如常见的场景:当一个函数被即将执行完后,也就是完成任务的最后一刻,需要回收和释放某些资源。
延迟调用的机制可以配合匿名函数来使用,这样就可以让匿名函数被直接调用,只能说是真的骚。
先看个小栗子:
package main
import "fmt"
func work() bool {
fmt.Println("函数开始工作...")
defer func() {
fmt.Println("回收相关资源工作开始!")
}()
defer func() {
fmt.Println("清理工作开始!")
}()
fmt.Println("函数正在工作...")
fmt.Println("函数工作完毕...")
return true
}
func main() {
status := work()
fmt.Println(status)
}
输出:
函数开始工作...
函数正在工作...
函数工作完毕...
清理工作开始!
回收相关资源工作开始!
true
在上面的栗子中,定义了个work函数,先看看输出的结果,感受下。在work函数中,注册了两个延迟调用,work函数从开始->正在->完毕,结束后才执行defer注册的匿名函数,这里要着重注意延迟调用规则:「先进后出」,也就是先注册后执行。“回收相关资源工作”是先注册的,他的执行顺序排在了“清理工作”的后面。感受到了吗?
最后说明一下:defer关键字让匿名函数实现了可直接调用,那么使用defer注册延迟调用时要注意,defer关键字的后面一定是要函数或方法的调用,不能直接写语句哦。
来源:https://mp.weixin.qq.com/s/DtSWp-APaAPmTG1mjAMRQQ