手记

Go中的函数是一等公民

6月2日Go语言中文网在杭州举办了线下的MeetUp活动,这次活动办很成功,感谢站长polaris在杭州举办活动的提议,感谢Seekload的筹备与主持,感谢Aaron提供场地,感谢所有到场者的技术经验分享,没有你们就没有这次精彩的活动。

在活动上,我做了个主题分享,今天把分享整理成文章,分享给学习Go语言的各位朋友。

参加本次活动的朋友,大多是刚接触Go,少数几个朋友把玩Go 2~3年了,所以我把主题定位到能让所有人听懂的主题。另外,大家所处行业各有不同,这就要求专注介绍Go本身的特性,这才是大家通用的地方。

最后选题为First class function in Go,这次没有做中文翻译,避免翻译后有误解。这个特性浅显易懂,但掌握Go语言的思维,才能把它用好。

线下分享后,证明选题选对了,大家都能听懂,所以现在不了解First class function的朋友不用着急,后面我会层层推进的方式介绍,相信你一定能理解,那就进入正文吧。

概念介绍

某个编程语言拥有First class function特性指可以把函数作为变量对待。也就说,函数与变量没有差别,它们是一样的,变量出现的地方都可以替换成函数,并且编译也是可以通过的,没有任何语法问题。

在Go里,变量可以存在于哪些地方?

变量可以被声明、定义,可以使用type创建变量的类型,可以作为函数的入参和返回值,可以存在slice, array, map等数据结构里,可以被动态的创建。

在Go中,函数也可以被声明、定义,可以使用type创建一个函数类型,可以作为其他函数的入参和返回值,可以保存在其他类型的数据结构里,最后,函数是可以被动态创建的。

简要归类一下就是下图的样子,除了上面提到的内容,还有匿名函数和闭包,将按下图顺序介绍每一个小特性。

定义函数类型

使用type定义一个函数类型,type后是类型名称,本例中是Operation,再后面是类型的定义,对于函数而言,被称为signature,即学历函数签名,这个函数签名表示:Operation类型的函数,它以2个int类型为入参,以1个int为返回值。所有满足该函数签名的函数,都是Operation类型的函数。

函数Add和Sub都符合Operation的签名,所以Add和Sub都是Operation类型。

声明函数类型的变量和为变量赋值

变量op是Operation类型的,可以把Add作为值赋值给变量op,执行op等价于执行Add。

高阶函数

高阶函数分为函数作为入参和函数作为返回值2部分。

函数作为其他函数入参

定义一个Calculator结构体,它始终保持计算后的结果。

它有一个方法Do,入参为一个Operation类型的函数op和1个int类型的变量a,使用计算器的值c.v和a作为op的入参,进行指定运算,并把结果保存会c.v。

main中,声明了一个变量calc,calc.v初始值为0,然后运行了加1和减2的操作,加减法的完成使用的我们之前定义的函数Add和Sub。操作等价于:

复制

1

2

calc.v=Add(calc.v, 1)

calc.v=Sub(calc.v, 2)

函数作为返回值+动态创建

这次,改变Operation的定义,修改为接收1个int类型的入参,返回1个int类型的返回值。

同时修改函数Add和Sub,它们接收1个int类型的入参,返回1个Operation类型的函数,这个函数是动态创建出来的。

以Add为例介绍,在Add里动态创建了一个函数,

复制

1

2

3

func(a int) int {

return a + b

}

该函数实现了在变量a基础上加b的操作,并返回结果,我们把这个函数赋值给变量addB,把addB作为返回值返回。

所以本例实现了以函数作为返回值和动态创建函数。

Operation,Add和Sub修改后,Calculator也要同步修改,方法Do修改为只接收Operation类型的函数。

main函数里,注意Do的入参:Add(1),它实现的效果是,创建了1个函数,该函数接收1个值,然后把这个值+1返回,如果用数学表示就是这样:

复制

1

2

// Add(1)

add1(x)=x + 1

同理,Sub(2)的数学表示如下:

复制

1

2

// Sub(2)

sub2(x)=x - 2

所以2次Do操作等价于:

复制

1

2

3

4

// calc.Do(Add(1))

calc.v=add1(calc.v)

// calc.Do(Sub(2))

calc.v=sub2(calc.v)

匿名函数

上图左边是普通函数,func后为函数名,然后为函数签名。右边只有func和函数签名,缺少函数名,右边的情况为匿名函数。

以Add函数其中定义的函数为例:

复制

1

2

3

func(a int) int {

return a + b

}

这就是1个匿名函数,它没有名字。addB并不是函数的名字,只是1个变量名而已,只不过这个变量名的类型是没有显示定义出来的。

Add通常简写为右边的形式。

闭包

很多人搞不清什么是匿名函数,什么是闭包,所以这里分开介绍这2个概念。

闭包指有权访问另一个函数作用域中的变量的函数。大白话就是,可以创建1个函数,它可以访问其他函数遍历,但不需要传值。

仍然是Add函数为例,比如匿名函数里直接使用了变量b,该匿名函数也是闭包函数。

闭包的特性注定了,闭包函数要定义在一个函数里面,定义在一个函数里面又只能是匿名函数。

那,匿名函数和闭包是不是就等价了?

No,一个函数可以是匿名函数,但不是闭包函数,因为闭包有时是有副作用的。

我们想并发的把sl中的值打印出来,结果为何会是右边这样?

因为并发的匿名函数,使用的是test1中的i,v,即这是闭包函数,所有的goroutine都共享这2个值,并且启动1个goroutine后,这2个值变为下一个位置的值。你运行的结果也许不是9 9 9….,因为这个goroutine的调度有关。

如何才能符合预期的打印?只使用匿名函数进行传值,不使用闭包。

接下来以一个实际的场景,和3种实现版本看如何用Go的思维去解决问题。

场景介绍

做Go语言工作,尤其是跟网络打交道的工作,连接管理是逃不开的。我做区块链相关的技术工作,区块链中也有网络管理,所以我就以区块链的网络管理为场景进行介绍,但不涉及具体的技术细节,大家莫慌,只需要理解2个概念就行。

区块链是构建在P2P网络之上,在P2P网络中:

一个节点即可以是服务器也可以是客户端,被称为Host,和本节点连接的所有节点都被称为Peer。

具体的场景是:Host需要保存所有建立连接的Peer,并对这些Peer进行维护:增加和删除Peer,并且提供Peer的查询和向所有Peer广播消息的接口。

针对这个问题场景,我写了3个版本的Demo,我们依次来介绍,再看的时候,可以思考其中的不同。

版本1

先看Peer定义,Peer中保存了ID,我们可以通过ID来表示全网中所有的节点,Peer中还有其他字段,比如网络连接、地址、协议版本等信息,此处已经省略掉。

Peer有一个WriteMsg的方法,实现向该Peer发送消息的功能,例子中使用打印替代。

Peer的定义在3个版本中都不会发生变化,所以后面就不再展示。

Host通过peers保存了所有连接的Peer,可以通过Peer.ID对Peer进行索引。Peer的管理是并发场景,比如,我们可能同时接收到多个Peer的连接,又同时需要向所有Peer广播消息,需要对peers加锁保护。最后,我们省略了Host的其他字段。

NewHost()用来创建一个Host对象,用来代表当前节点。

友情提醒:Host在每一个版本都会不同。

Host有4个方法,分别是:

AddPeer: 增加1个Peer。RemovePeer: 删除1个Peer。GetPeer: 通过Peer.ID查询1个Peer。BroadcastMsg: 向所有Peer发送消息。

每一个方法都需要获取lock,然后访问peers,如果只读取peers则使用读锁。

第1个版本已经介绍完了,大家可以思考一下版本1的缺点。

第1个版本跟其他语言实现其实没有本质区别,用C++、Java等也能写出上面逻辑的代码,只不过这个是Go语言实现的罢了。

这个版本是一个communicate by sharing memory的体现,具体来讲,每个goroutine都是1个实体,它们同时运行,调用Host的不同方法来访问peers,只有拿到当前lock的goroutine才能访问peers,仿佛当前goroutine在同其他goroutine讲:我现在有访问权,你们等一下。本质上就是,通过共享Host.lock这块内存,各goroutine进行交流(表明自己拥有访问权)。

版本2

很多Go老手都听过这句话了,这是Go的“联合创始人”Rob Pike某个会议上说的。

在Go中,推荐使用CSP实现并发,而不是习惯性的使用Lock,使用channel传递数据,达到多goroutine间共享数据的目的,也就是share memory by communicating。

所以,我们版本2,就使用channel的方式,来实现Peer的管理。

在版本1中,peers是大家都想访问的,并且Host有4个方法,画到了上面的图中,我们看下怎么用CSP实现。

peers需要在单独的goroutine中,其他的4个方法在其他的goroutine中调用,它们之间进行通信。

我对使用CSP有一个好的实践,就是把数据流动画出来,并把要流动的数据标上,然后那些数据流动的线条,就是channel,线条上的数据就是channel要传递的数据,图中也把这些线条和数据标上了。具体的细节,可以识别图片中的二维码,看看这篇老文,还有就是并不是所有的并发场景都适合使用channel,有些用锁更好,这篇文章也有介绍。

重新定义Host,增加了4个channel,从上到下分别用于增加Peer、广播消息、删除Peer和停止Host。

Host增加了2个方法:

Start()用于启动1个goroutine运行loop(),loop保存所有的peers。Stop()用于关闭Host,让loop退出。

左边是loop()的实现,它从4个channel里接收数据,然后做不同的操作。

右边是AddPeer, RemovePeer, BroadcastMsg的实现。

利用1分钟的事件,左右两边对照着看,理解增加1个Peer的全过程。

这就是版本2的全部实现了,思考一下版本2有什么问题,原因是啥?

问题就是我们没有实现GetPeer这个方法,聪明的你一定在Host的定义就发现了,只有增加、删除和广播消息的channel。

没能实现GetPeer的原因下图中进行了介绍,你有没有解决办法?

可能会有很多goroutine调用GetPeer,我们需要向每一个goroutine发送结果,这就需要每一个goroutine都需要对应的1个接收结果的channel。

所以我们可以增加1个query channel,channel里传递Peer.ID和接收结果的channel。

还有没有其他办法?我们今天的主题First class function还有入场,你有办法用这个特性实现吗?

版本3

First class function: 函数可以向变量一样使用。那channel里面是不是可以传递函数呢?当然可以。

我们可以建立一个channel,用这个channel向loop传递操作peers的函数,所以函数的入参是peers map[string]*Peer,无需返回值,因为函数是在loop里面调用的,调用AddPeer等函数的goroutine是接收不到返回值的。我们把这个类型的函数定义为Operation。

Host修改为只有2个channel,stop功能如版本2,opCh用来传递Operation类型的函数。

loop函数可以简化为左边的形式了,右边是AddPeer和RemovePeer,以AddPeer为例进行介绍,创建了一个匿名函数,向peers里增加p,然后把函数发送到opCh。

BroadcastMsg与AddPeer类似。

我们重点看一下GetPeer,创建了retCh用于接收查询的结果,创建了匿名函数进行查询,并把查询结果发送到retCh,然后启动1个goroutine把匿名函数写入到opCh,最后等待从retCh读取查询结果。

这样就实现了向每个调用GetPeer的goroutine发送查询结果。

总结都在上面了,不多说了。

友情提醒:这3种方式本身并无优劣之分,具体要用那种实现,要依赖自身的实际场景进行取舍。

0人推荐
随时随地看视频
慕课网APP