大家好,我是公众号「线下聚会游戏」作者,开发了《联机桌游合集》,是个网页,可以很方便的跟朋友联机玩斗地主、五子棋等游戏。其中的核心技术就是WebSocket,快来关注我 HullQin 跟我一起学习吧!
背景
上篇文章:《[Go WebSocket] 单房间的聊天室》,介绍了如何实现一个单房间的聊天室。
今天我们实现一个多房间的聊天室。如果你没阅读上面的文章,一定要先看一下,因为这篇文章更复杂,如果你不弄懂上面几篇,这篇可能跟不上节奏噢。
以第上篇文章的代码为起点
今天,我们修改上次「聊天室」的代码即可。
因为gorilla/websocket
官方并未提供多房间聊天室的demo,只提供了单房间聊天室的demo(就是上篇文章所讲的),这次我们要自己动手开发啦!
思考
还记得上次绘制的架构图吗?
这次,我们要动手改代码了!先不要直接写代码,我们先画图!理清思路,代码就信手拈来了!
以前,我们全局只有1个Hub goroutine
,而广播(broadcast)正是依赖了这个goroutine。客户端把希望广播的消息传到broadcast这个channel里,由Hub goroutine
读取broadcast channel,遍历它的clients,把消息写入send channel,再由各个客户端的Write goroutine
读取send channel,发送消息。
现在既然要实现多房间,Hub goroutine
应该有多个。具体表现应该是这样:
何时创建房间
上一篇文章,我们只有一个房间,所以在服务启动时,在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的语法:
不像Http请求APIfetch
,WebSocket
不允许用户传入header参数。如果你想传递额外信息,官方建议就是三种:URL、cookie、连接建立后发个消息。本文均已罗列。
好了。以上就是本篇内容了,下一篇,我们进入【实践篇】,写代码,实现一个多房间的聊天室。
写在最后
我是HullQin,公众号线下聚会游戏的作者(欢迎关注公众号,联系我,交个朋友),转发本文前需获得作者HullQin授权。我独立开发了《联机桌游合集》,是个网页,可以很方便的跟朋友联机玩斗地主、五子棋等游戏,不收费无广告。还独立开发了《合成大西瓜重制版》。还开发了《Dice Crush》参加Game Jam 2022。喜欢可以关注我 HullQin 噢~我有空了会分享做游戏的相关技术。