乐趣区

关于后端:Go-WebSocket-多房间的聊天室二代码实现

jiezi

我是 HullQin，公众号 线下团聚游戏 的作者（欢送关注公众号，发送加微信，交个敌人），转发本文前需取得作者 HullQin 受权。我独立开发了《联机桌游合集》，是个网页，能够很不便的跟敌人联机玩斗地主、五子棋等游戏，不免费没广告。还开发了《Dice Crush》加入 Game Jam 2022。喜爱能够关注我 HullQin 噢～我有空了会分享做游戏的相干技术。

背景

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

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

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

第四篇文章：《多房间的聊天室（一）思考篇》，介绍了实现一个多房间的聊天室的思路。

明天咱们实现一个多房间的聊天室。如果你没浏览下面的文章，肯定要先看一下，因为这篇文章更简单，如果你不弄懂下面几篇，这篇可能跟不上节奏噢。

计划回顾

上篇文章，有 2 个决策点：

  1. 何时创立房间？
  2. 如何决定客户端连哪个房间？

上篇文章都提到了多种解决方案，都是能够抉择的。然而本文要开始写代码实现了，必须作出一个抉择，抉择如下：

决策点 1 抉择「计划二：动态创建房间」。决策点 2 抉择「计划一：URL 里指定」。

这也是我的「联机桌游合集」所采纳的计划。

间接看源码

多房间聊天室案例代码的地址：https://github.com/HullQin/go-websocket-examples

在 chat-multi-rooms 文件夹中，文章可配套 commit 记录浏览：

  • ws handler logic 对应「URL 指定房间号：路由参数」、「动态创建房间逻辑」。
  • http handler logic 对应「批改 http 服务」。

初始代码

咱们用《单房间的聊天室》的代码为根底，在它下面革新。

URL 指定房间号：路由参数

因为咱们要在 URL 里指定房间号，怎么实现呢？咱们能够借助 gorilla/mux 这个库，它能够实现弱小的路由能力。Github 介绍如下：Package gorilla/mux implements a request router and dispatcher for matching incoming requests to their respective handler.

装置依赖：

go get github.com/gorilla/mux

咱们把之前的代码改为：

import ("github.com/gorilla/mux")
// ...
r := mux.NewRouter()
r.HandleFunc("/", serveHome)
r.HandleFunc("/ws/{room}", func(w http.ResponseWriter, r *http.Request) {vars := mux.Vars(r)
    roomId := vars["room"]
    // ...
}
err := http.ListenAndServe(*addr, r)

能够看到，咱们在路有中减少了 {room}，这是个动静参数，能够匹配字符串，赋值给 room。具体参数的值如何取值呢？通过vars := mux.Vars(r) 即可取得一个 map，这个 map 的 key 是参数名，value 是参数的值，均为字符串类型。

所以到当初，咱们从 URL 中取得了 roomId。

动态创建房间逻辑

不像单房间聊天室，咱们须要在全局开启一个 hub goroutine。这次，咱们必须动静新增 hub goroutine。

怎么办呢？咱们须要有一个全局变量，保留所有的 hub，能够用 map。每当连贯来的时候，就查看该 roomId 是否存在，如果存在，就取对应的 hub，如果不存在，就须要新建 hub。

在 addr 定义下方，新建 house 定义。

var addr = flag.String("addr", "localhost:8080", "http service address")
var house = make(map[string]*Hub)

欠缺处理器逻辑。

r.HandleFunc("/ws/{room}", func(w http.ResponseWriter, r *http.Request) {vars := mux.Vars(r)
   roomId := vars["room"]
   room, ok := house[roomId]
   var hub *Hub
   if ok {hub = room} else {hub = newHub()
      house[roomId] = hub
      go hub.run()}
   serveWs(hub, w, r)
})

到当初，咱们曾经把后端逻辑改完了！你敢信？

得益于上上篇文章杰出的设计（把 hub 设计为可扩大为多房间的），从单房间革新到多房间，居然如此轻松！

然而并没有完，为了不便测试，咱们还须要批改一下home.html。

批改 http 服务

home.html

32 行代码革新如下：

conn = new WebSocket("ws://" + document.location.host + "/ws" + document.location.pathname);

也就是说，你前端拜访的是 localhost:8080/ha，就会进入ha 房间。当然你能够批改 ha 进入其它房间。

main.go

这几行代码先删掉：

//if r.URL.Path != "/" {// http.Error(w, "Not found", http.StatusNotFound)
// return
//}

因为咱们之后要拜访的是localhost:8080/ha，而非localhost:8080/，所以 r.URL.Path 肯定不能是/，不然就不晓得房间号了。

此外，路由逻辑也适配一下房间号参数：

r.HandleFunc("/{room}", serveHome)

至此，功败垂成！快去测试一下！

浏览器关上 3 个 Tab，别离拜访http://localhost:8080/123、http://localhost:8080/123、http://localhost:8080/444。而后去发消息，你会发现 123 房间是音讯相互播送的，而且不会发送到 444 房间；而且 444 房间的音讯也不会发送到 123 房间！

多房间的聊天室，咱们实现啦！

待优化项

当初房间数只会源源不断的增多，house 这个 map 会越来越大，终将造成内存不足，这不是一个好事件。

所以咱们后续须要加一个优化：当最初一个客户端断开连接时，回收（删除）这个房间。

写在最初

我是 HullQin，公众号 线下团聚游戏 的作者（欢送关注公众号，发送加微信，交个敌人），转发本文前需取得作者 HullQin 受权。我独立开发了《联机桌游合集》，是个网页，能够很不便的跟敌人联机玩斗地主、五子棋等游戏，不免费没广告。还开发了《Dice Crush》加入 Game Jam 2022。喜爱能够关注我 HullQin 噢～我有空了会分享做游戏的相干技术。

退出移动版