[Go WebSocket] 多房间的聊天室（一）思考篇

我是HullQin，公众号线下聚会游戏的作者（欢迎关注公众号，发送加微信，交个朋友），转发本文前需获得作者HullQin授权。我独立开发了《联机桌游合集》，是个网页，可以很方便的跟朋友联机玩斗地主、五子棋等游戏，不收费没广告。还开发了《Dice Crush》参加Game Jam 2022。喜欢可以关注我 HullQin 噢～我有空了会分享做游戏的相关技术。

背景

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

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

第三篇文章：《单房间的聊天室》，介绍了如何实现一个单房间的聊天室。

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

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

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

因为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 噢～我有空了会分享做游戏的相关技术。