[Go WebSocket] 多房间的聊天室(一)思考篇

8,267 阅读5分钟

大家好,我是公众号「线下聚会游戏」作者,开发了《联机桌游合集》,是个网页,可以很方便的跟朋友联机玩斗地主、五子棋等游戏。其中的核心技术就是WebSocket,快来关注专栏《Go WebSocket》,跟我一起学习吧!

背景

第一篇文章:《为什么我选用Go重构Python版本的WebSocket服务?》,介绍了我的目标。

第二篇文章:《你的第一个Go WebSocket服务: echo server》,介绍了一下怎么写一个WebSocket server。

第三篇文章:《你的第二个Go WebSocket服务: 聊天室》,介绍了如何实现一个单房间的聊天室。

今天我们实现一个多房间的聊天室。如果你没阅读上面的文章,一定要先看一下,因为这篇文章更复杂,如果你不弄懂上面几篇,这篇可能跟不上节奏噢。

以第三篇文章的代码为起点

今天,我们修改上次「聊天室」的代码即可。

因为gorilla/websocket官方并未提供多房间聊天室的demo,只提供了单房间聊天室的demo(就是上篇文章所讲的),这次我们要自己动手开发啦!

思考

还记得上次绘制的架构图吗?

image.png

这次,我们要动手改代码了!先不要直接写代码,我们先画图!理清思路,代码就信手拈来了!

以前,我们全局只有1个Hub goroutine,而广播(broadcast)正是依赖了这个goroutine。客户端把希望广播的消息传到broadcast这个channel里,由Hub goroutine读取broadcast channel,遍历它的clients,把消息写入send channel,再由各个客户端的Write goroutine读取send channel,发送消息。

现在既然要实现多房间,Hub goroutine应该有多个。具体表现应该是这样:

0723Go多房间架构图.png

何时创建房间

上一篇文章,我们只有一个房间,所以在服务启动时,在main函数里已经启动了Hub goroutine,但是这次不行了。我们可能有多个房间,所以有两种方案。

方案一:设置固定的几个房间

如果房间数目是固定的,例如你设置了一共只有10个房间,那么你完全可以参考上篇文章,只不过这次要调用10次go hub.run()了。因为每次调用就启动一个goroutine,10个房间就是10次。

方案二:动态创建房间

如果房间允许动态创建,那么你就需要在客户端连接时,给你传入一个「房间号」参数,你需要判断当前「房间号」是否存在,如果不存在,需要主动创建(调用go hub.run()),如果存在,就用存在的房间。

结论

如果是方案一,没有挑战性,参考上篇文章,你也可以轻松实现。今天我们按照方案二来做。

如何决定客户端连哪个房间

既然我们有多个房间,客户端是一定需要告诉服务端:我要进哪个房间的。有3个常用方案:

方案一:URL里指定

我们可以在连接WebSocket时,在URL里指定(通过路径或参数)。上篇文章,我们对WebSocket服务定义的路有规则是/ws。可以再扩充一下,变为/ws/【房间号】。读取【房间号】即可。

方案二:cookie里指定

在连接WebSocket时,会把当前域名的cookie也带过去,服务端可以读取的到。因此,也可以把房间号信息存到cookie里。

注:「cookie指定」这种方式很常用,尤其是网页里的【用户消息】,如果有人给你点赞、关注,这个消息需要实时推送给用户。而且如果你多开了几个Tab,这些Tab都应该收到。这种情况下,你的每个浏览器Tab就是一个「Client」,你这个账号就是一个「Room」。(这时候Room就是一个抽象概念了,不能把它当作是一个真实的房间,只能当作一个连接池子,广播消息时,这个池子里的所有Client会收到相同的消息)

方案三:连接完成时,客户端主动发一个消息,表明房间

这个跟方案一、二有本质不同。方案一、二在WebSocket连接时,就确定了房间号,效率更高。但是方案三是在连接完成时,客户端主动发一个消息,告知服务端房间号。

虽然效率不如方案一、二,但是更加灵活,扩展性好。因为那个消息里可以附带更多的信息。

这也是很多应用采取的方案,大胆用吧!因为效率影响可以忽略不计,但是可扩展性影响太大了,会影响你未来几年的维护成本!

为什么不放在header信息?

因为WebSocket协议其实不建议把信息放在自定义header里。你看建立WebSocket连接的JS API的语法:

image.png

不像Http请求APIfetchWebSocket不允许用户传入header参数。如果你想传递额外信息,官方建议就是三种:URL、cookie、连接建立后发个消息。本文均已罗列。

好了。以上就是本篇内容了,下一篇,我们进入【实践篇】,写代码,实现一个多房间的聊天室。

写在最后

我是HullQin,独立开发了《联机桌游合集》,是个网页,可以很方便的跟朋友联机玩斗地主、五子棋等游戏,不收费无广告。还独立开发了《合成大西瓜重制版》。还开发了《Dice Crush》参加了某个游戏制作比赛。喜欢可以关注我噢~我有空了会分享做游戏的相关技术,会在这2个专栏里分享:《教你做小游戏》《极致用户体验》

本文正在参加技术专题18期-聊聊Go语言框架